Update app.py

app.py

@@ -37,7 +37,11 @@ def generate_image(prompt, negative_prompt, sampler, seed, guidance_scale, infer
 
     response.raise_for_status()
     response_body = response.json()
+    # Print the API response for debugging
+    print("API Response:", response_body)
 
+    if response_body.get("output") is None:
+        return "Error: API response does not contain 'output' key."
     # Extract image URL from response
     image_url = response_body.get("output").get("image_url")
     return image_url