Update app.py

app.py

@@ -9,9 +9,10 @@ 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 Hugging Face secrets."
+        return "❌ API key not found. Please add GOOGLE_API_KEY in your Hugging Face secrets."
 
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate:generate?key={API_KEY}"
+    # Use Gemini's free image generation model
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-preview-image-generation:generate?key={API_KEY}"
     headers = {"Content-Type": "application/json"}
     data = {"prompt": {"text": prompt}}
 
@@ -20,33 +21,33 @@ def image_generation(prompt):
     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"
-
     if res.status_code != 200:
-        return debug_info
+        return f"❌ Error {res.status_code}:\n{res.text}"
 
     try:
         result = res.json()
     except Exception:
-        return f"❌ Invalid JSON\n\n{debug_info}"
+        return f"❌ Invalid JSON response:\n{res.text}"
 
     # Extract image if present
-    if "images" in result and len(result["images"]) > 0:
-        image_base64 = result["images"][0].get("image_base64")
-        if image_base64:
-            image = Image.open(BytesIO(base64.b64decode(image_base64)))
-            return image
-
-    # No image — show what was actually returned
-    return f"⚠️ No image found in response.\n\nFull response:\n{result}"
+    if "candidates" in result:
+        for candidate in result["candidates"]:
+            if "content" in candidate:
+                for part in candidate["content"].get("parts", []):
+                    if "inline_data" in part:
+                        image_base64 = part["inline_data"].get("data")
+                        if image_base64:
+                            image = Image.open(BytesIO(base64.b64decode(image_base64)))
+                            return image
+
+    return f"⚠️ No image data found.\n\nFull response:\n{result}"
 
 iface = gr.Interface(
     fn=image_generation,
-    inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g. A cat astronaut floating in space"),
+    inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g. A futuristic city under sunset"),
     outputs=gr.Image(label="Generated Image"),
-    title="Google AI Studio Image Generator (Debug Mode)",
-    description="Uses Imagen 3.0 API — shows debug info if response is empty."
+    title="Google Gemini Image Generator",
+    description="Uses gemini-2.0-flash-preview-image-generation (free-tier). Educational purpose only."
 )
 
 if __name__ == "__main__":