Update app.py

app.py

@@ -34,9 +34,13 @@
 # if __name__ == "__main__":
 #     demo.launch()
 
+
+
+
 import os
 import gradio as gr
-import google.genai as genai   # ✅ correct import
+import google.genai as genai
+from google.genai import types
 from PIL import Image
 from io import BytesIO
 import base64
@@ -49,17 +53,26 @@ client = genai.Client(api_key=api_key)
 
 # Function to generate or edit image
 def generate_or_edit_image(prompt, uploaded_img):
-    contents = [prompt]
+    # Always start with prompt as text
+    contents = [types.Content(role="user", parts=[types.Part(text=prompt)])]
 
-    # If user uploads an image, include it
+    # If an image is uploaded, add it properly as a Blob
     if uploaded_img is not None:
         buffered = BytesIO()
         uploaded_img.save(buffered, format="PNG")
         image_bytes = buffered.getvalue()
-        contents.append({
-            "mime_type": "image/png",
-            "data": image_bytes
-        })
+
+        contents.append(
+            types.Content(
+                role="user",
+                parts=[types.Part(
+                    inline_data=types.Blob(
+                        mime_type="image/png",
+                        data=image_bytes
+                    )
+                )]
+            )
+        )
 
     try:
         response = client.models.generate_content(
@@ -67,12 +80,11 @@ def generate_or_edit_image(prompt, uploaded_img):
             contents=contents,
         )
 
-        # 🔎 Debug: print structure
+        # Debug
         print("Gemini Response:", response)
 
         for part in response.candidates[0].content.parts:
             if part.inline_data is not None:
-                # Sometimes Gemini returns base64 string
                 data = part.inline_data.data
                 if isinstance(data, str):
                     image_bytes = base64.b64decode(data)
@@ -82,7 +94,7 @@ def generate_or_edit_image(prompt, uploaded_img):
                 image = Image.open(BytesIO(image_bytes))
                 return image
 
-        return "⚠️ No image generated. Try a different prompt."
+        return "⚠️ No image returned. Try another prompt."
 
     except Exception as e:
         return f"❌ Error: {e}"
@@ -96,7 +108,7 @@ demo = gr.Interface(
     ],
     outputs=gr.Image(label="Generated/Edited Image"),
     title="Image Generator & Editor",
-    description="Enter a prompt to generate a new image, or upload a JPG/PNG to edit."
+    description="Enter a prompt to generate a new image, or upload a JPG/PNG to edit it."
 )
 
 if __name__ == "__main__":