Quellcode durchsuchen

feat: allow enabling after url is provided

Timothy J. Baek vor 1 Jahr
Ursprung
Commit
797ed9ac7f
2 geänderte Dateien mit 13 neuen und 4 gelöschten Zeilen
  1. 6 2
      backend/apps/images/main.py
  2. 7 2
      src/lib/components/chat/Settings/Images.svelte

+ 6 - 2
backend/apps/images/main.py

@@ -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):

+ 7 - 2
src/lib/components/chat/Settings/Images.svelte

@@ -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 () => {