From e15e015db162f7236a7d38bbc7992a8076ac568a Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sat, 24 Feb 2024 15:48:14 -0800
Subject: [PATCH] fix: images
---
backend/apps/images/main.py | 8 ++++----
src/lib/components/chat/Settings/Images.svelte | 6 ++++--
2 files changed, 8 insertions(+), 6 deletions(-)
diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py
index 39d3f96aa..3ea8b1711 100644
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -49,7 +49,7 @@ async def toggle_enabled(request: Request, user=Depends(get_admin_user)):
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))
+ raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
class UrlUpdateForm(BaseModel):
@@ -109,7 +109,7 @@ def get_models(user=Depends(get_current_user)):
models = r.json()
return models
except Exception as e:
- raise HTTPException(status_code=r.status_code, detail=ERROR_MESSAGES.DEFAULT(e))
+ raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
@app.get("/models/default")
@@ -120,7 +120,7 @@ async def get_default_model(user=Depends(get_admin_user)):
return {"model": options["sd_model_checkpoint"]}
except Exception as e:
- raise HTTPException(status_code=r.status_code, detail=ERROR_MESSAGES.DEFAULT(e))
+ raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
class UpdateModelForm(BaseModel):
@@ -190,4 +190,4 @@ def generate_image(
return r.json()
except Exception as e:
print(e)
- raise HTTPException(status_code=r.status_code, detail=ERROR_MESSAGES.DEFAULT(e))
+ raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
diff --git a/src/lib/components/chat/Settings/Images.svelte b/src/lib/components/chat/Settings/Images.svelte
index 94cb3213b..cc8274546 100644
--- a/src/lib/components/chat/Settings/Images.svelte
+++ b/src/lib/components/chat/Settings/Images.svelte
@@ -32,9 +32,11 @@
const getModels = async () => {
models = await getDiffusionModels(localStorage.token).catch((error) => {
toast.error(error);
- return null;
+ return [];
+ });
+ selectedModel = await getDefaultDiffusionModel(localStorage.token).catch((error) => {
+ return '';
});
- selectedModel = await getDefaultDiffusionModel(localStorage.token);
};
const updateAUTOMATIC1111UrlHandler = async () => {