Update app.py

app.py

@@ -214,6 +214,14 @@ def generate_image_from_parameters(prompt, guidance_scale, controlnet_scale, con
         # Handle exceptions and return an error message if something goes wrong
         return str(e)
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # You can replace ["*"] with specific origins if needed
+    allow_credentials=True,
+    allow_methods=["*"],  # Allow all methods
+    allow_headers=["*"],  # Allow all headers
+)
+
 @app.post("/generate_image")
 async def generate_image(
     prompt: str = Form(...),
@@ -248,8 +256,8 @@ async def generate_image(
         return StreamingResponse(content=output_image_io, media_type="image/png")
 
     except Exception as e:
-        # Handle exceptions and return an error message if something goes wrong
-        return str(e)
+        logger.error("Error occurred during image generation: %s", str(e))
+        return "Failed to generate image"
 
 if __name__ == "__main__":
     import uvicorn