diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py index 06d0bcce8..f4dfc8a73 100644 --- a/backend/apps/images/main.py +++ b/backend/apps/images/main.py @@ -150,10 +150,11 @@ async def update_engine_url( else: url = form_data.AUTOMATIC1111_BASE_URL.strip("/") try: - r = requests.head(url) + r = requests.head(url) + r.raise_for_status() app.state.config.AUTOMATIC1111_BASE_URL = url except Exception as e: - raise HTTPException(status_code=400, detail="Invalid URL provided.") + raise HTTPException(status_code=400, detail=ERROR_MESSAGES.INVALID_URL) if form_data.COMFYUI_BASE_URL == None: app.state.config.COMFYUI_BASE_URL = COMFYUI_BASE_URL @@ -162,9 +163,10 @@ async def update_engine_url( try: r = requests.head(url) + r.raise_for_status() app.state.config.COMFYUI_BASE_URL = url except Exception as e: - raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e)) + raise HTTPException(status_code=400, detail=ERROR_MESSAGES.INVALID_URL) if form_data.AUTOMATIC1111_API_AUTH == None: app.state.config.AUTOMATIC1111_API_AUTH = AUTOMATIC1111_API_AUTH diff --git a/backend/constants.py b/backend/constants.py index b9c7fc430..08b17e5ce 100644 --- a/backend/constants.py +++ b/backend/constants.py @@ -89,6 +89,7 @@ class ERROR_MESSAGES(str, Enum): OLLAMA_API_DISABLED = ( "The Ollama API is disabled. Please enable it to use this feature." ) + INVALID_URL = "The URL you provided is invalid. Please double-check and try again." class TASKS(str, Enum):