Update app.py

app.py

@@ -1,53 +1,52 @@
 import os
 import requests
 import gradio as gr
+import base64
+from io import BytesIO
+from PIL import Image
 
-# Get the API key from Hugging Face Secrets
 API_KEY = os.getenv("GOOGLE_API_KEY")
 
 def image_generation(prompt):
     if not API_KEY:
-        return "❌ API key not found. Please add GOOGLE_API_KEY in your Hugging Face secrets."
+        return "❌ API key not found. Please add GOOGLE_API_KEY in Hugging Face secrets."
 
-    # Google Imagen 3.0 endpoint for image generation
     url = f"https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate:generate?key={API_KEY}"
-
     headers = {"Content-Type": "application/json"}
     data = {"prompt": {"text": prompt}}
 
-    # Send POST request
-    res = requests.post(url, headers=headers, json=data)
+    try:
+        res = requests.post(url, headers=headers, json=data)
+    except Exception as e:
+        return f"❌ Request failed: {str(e)}"
+
+    # Always show HTTP status and text
+    debug_info = f"HTTP {res.status_code}\n\nResponse:\n{res.text}\n\n"
 
-    # Check for HTTP errors
     if res.status_code != 200:
-        return f"❌ Error {res.status_code}:\n{res.text}"
+        return debug_info
 
     try:
         result = res.json()
     except Exception:
-        return f"❌ Invalid JSON response:\n{res.text}"
+        return f"❌ Invalid JSON\n\n{debug_info}"
 
-    # If API returned base64 image data
+    # Extract image if present
     if "images" in result and len(result["images"]) > 0:
-        image_base64 = result["images"][0].get("image_base64", None)
+        image_base64 = result["images"][0].get("image_base64")
         if image_base64:
-            import base64
-            from io import BytesIO
-            from PIL import Image
-            image_bytes = base64.b64decode(image_base64)
-            image = Image.open(BytesIO(image_bytes))
+            image = Image.open(BytesIO(base64.b64decode(image_base64)))
             return image
 
-    # Otherwise show the raw response for debugging
-    return f"⚠️ Unexpected response:\n{result}"
+    # No image — show what was actually returned
+    return f"⚠️ No image found in response.\n\nFull response:\n{result}"
 
-# Gradio interface
 iface = gr.Interface(
     fn=image_generation,
-    inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g. A robot painting a sunset over the ocean"),
+    inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g. A cat astronaut floating in space"),
     outputs=gr.Image(label="Generated Image"),
-    title="Google AI Studio Image Generator",
-    description="Generate images using Google AI Studio (Imagen 3.0 API). Educational use only."
+    title="Google AI Studio Image Generator (Debug Mode)",
+    description="Uses Imagen 3.0 API — shows debug info if response is empty."
 )
 
 if __name__ == "__main__":