Update main.py

main.py

@@ -129,6 +129,8 @@ def generate_prompts(prompt_text: str):
 
 from fastapi import Query
 
+from fastapi import HTTPException
+
 @app.get("/send_inputs")
 def send_inputs(
     inputs: str,
@@ -147,12 +149,11 @@ def send_inputs(
             return {"image_base64": image_base64}
         else:
             # Return an error message if the image couldn't be generated
-            return {"error": "Failed to generate image."}
+            raise HTTPException(status_code=500, detail="Failed to generate image.")
     except Exception as e:
         # Log the error and return an error message
         print(f"Error generating image: {e}")
-        return {"error": "Failed to generate image."}
-
+        raise HTTPException(status_code=500, detail="Failed to generate image.")
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")