Merge commit from fork

SSRF Fix

Merge commit from fork
SSRF Fix
8d257ed5 · Timothy Jaeryang Baek · GitHub · 99d10d11 · 590fd129 · 8d257ed5
Unverified Commit 8d257ed5 authored Aug 08, 2024 by Timothy Jaeryang Baek Committed by GitHub Aug 08, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -150,10 +150,10 @@ async def update_engine_url(
    else:
        url = form_data.AUTOMATIC1111_BASE_URL.strip("/")
        try:
-            r = requests.head(url)
+            r = requests.head(url) 
            app.state.config.AUTOMATIC1111_BASE_URL = url
        except Exception as e:
-            raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
+            raise HTTPException(status_code=400, detail="Invalid URL provided.")

    if form_data.COMFYUI_BASE_URL == None:
        app.state.config.COMFYUI_BASE_URL = COMFYUI_BASE_URL