Update app.py

app.py

@@ -10,67 +10,65 @@ from google.genai import types
 # 1. Gemini Client
 # ==============================
 client = genai.Client(
-    api_key=os.environ["GEMINI_API_KEY"]
+    api_key=os.environ.get("GEMINI_API_KEY")
 )
 
 
 # ==============================
-# 2. Safe Multimodal Function
+# 2. Multimodal Function (SAFE)
 # ==============================
 def analyze_image(image, prompt):
 
-    # ---- SAFETY CHECKS ----
-    if image is None:
-        return "❌ Please upload an image."
+    try:
+        # ---- Checks ----
+        if image is None:
+            return "❌ No image uploaded"
 
-    if prompt is None or prompt.strip() == "":
-        return "❌ Please enter a text prompt."
+        if not prompt or prompt.strip() == "":
+            return "❌ Prompt is empty"
 
-    # Convert image to bytes
-    buffer = io.BytesIO()
-    image.save(buffer, format="PNG")
-    image_bytes = buffer.getvalue()
+        # Convert image to bytes
+        buffer = io.BytesIO()
+        image.save(buffer, format="PNG")
+        image_bytes = buffer.getvalue()
 
-    # Create Gemini image part
-    image_part = types.Part.from_bytes(
-        data=image_bytes,
-        mime_type="image/png"
-    )
+        image_part = types.Part.from_bytes(
+            data=image_bytes,
+            mime_type="image/png"
+        )
+
+        # ✅ USE FLASH MODEL (IMPORTANT)
+        response = client.models.generate_content(
+            model="gemini-1.5-flash",
+            contents=[
+                prompt,
+                image_part
+            ]
+        )
 
-    # Generate response
-    response = client.models.generate_content(
-        model="gemini-1.5-pro",
-        contents=[
-            prompt,
-            image_part
-        ]
-    )
+        return response.text
 
-    return response.text
+    except Exception as e:
+        # 🔍 Show real Gemini error
+        return f"❌ Gemini Error:\n{str(e)}"
 
 
 # ==============================
-# 3. Gradio Interface
+# 3. Gradio UI
 # ==============================
 interface = gr.Interface(
     fn=analyze_image,
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
-        gr.Textbox(
-            label="Prompt",
-            placeholder="Describe the image"
-        )
+        gr.Textbox(label="Prompt", value="Describe the image")
     ],
-    outputs=gr.Textbox(label="Gemini Response"),
-    title="Multimodal AI App (Gemini)",
-    description="Upload an image and ask a simple question."
+    outputs=gr.Textbox(label="Response"),
+    title="Gemini Multimodal Test App",
 )
 
 
 # ==============================
 # 4. Launch
 # ==============================
-interface.launch()
-
-
+interface.launch(ssr_mode=False)