tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

feat: allow enabling after url is provided

Browse source
This commit is contained in:
Timothy J. Baek 2024-02-21 18:57:53 -08:00
parent d863e7e60b
commit 797ed9ac7f
2 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
backend/apps/images/main.py
View file

@ -43,8 +43,12 @@ async def get_enable_status(request: Request, user=Depends(get_admin_user)):
@app.get("/enabled/toggle", response_model=bool)
async def toggle_enabled(request: Request, user=Depends(get_admin_user)):
app.state.ENABLED = not app.state.ENABLED
return app.state.ENABLED
try:
r = requests.head(app.state.AUTOMATIC1111_BASE_URL)
app.state.ENABLED = not app.state.ENABLED
return app.state.ENABLED
except Exception as e:
raise HTTPException(status_code=r.status_code, detail=ERROR_MESSAGES.DEFAULT(e))
class UrlUpdateForm(BaseModel):

9
src/lib/components/chat/Settings/Images.svelte
View file

@ -45,8 +45,13 @@
};
const toggleImageGeneration = async () => {
enableImageGeneration = await toggleImageGenerationEnabledStatus(localStorage.token);
config.set(await getBackendConfig(localStorage.token));
if (AUTOMATIC1111_BASE_URL) {
enableImageGeneration = await toggleImageGenerationEnabledStatus(localStorage.token);
config.set(await getBackendConfig(localStorage.token));
} else {
enableImageGeneration = false;
toast.error('AUTOMATIC1111_BASE_URL not provided');
}
};
onMount(async () => {