Update app.py

app.py

@@ -51,12 +51,18 @@ api_key = os.getenv("Gemini_api_key")
 # Initialize Gemini client
 client = genai.Client(api_key=api_key)
 
-# Function to generate or edit image
 def generate_or_edit_image(prompt, uploaded_img):
-    # Always start with prompt as text
-    contents = [types.Content(role="user", parts=[types.Part(text=prompt)])]
+    contents = []
 
-    # If an image is uploaded, add it properly as a Blob
+    # Add prompt always
+    contents.append(
+        types.Content(
+            role="user",
+            parts=[types.Part(text=prompt)]
+        )
+    )
+
+    # If an image is uploaded, include it as reference
     if uploaded_img is not None:
         buffered = BytesIO()
         uploaded_img.save(buffered, format="PNG")
@@ -65,12 +71,17 @@ def generate_or_edit_image(prompt, uploaded_img):
         contents.append(
             types.Content(
                 role="user",
-                parts=[types.Part(
-                    inline_data=types.Blob(
-                        mime_type="image/png",
-                        data=image_bytes
+                parts=[
+                    types.Part(
+                        text="Use this image as reference while applying my changes."
+                    ),
+                    types.Part(
+                        inline_data=types.Blob(
+                            mime_type="image/png",
+                            data=image_bytes
+                        )
                     )
-                )]
+                ]
             )
         )
 
@@ -80,8 +91,8 @@ def generate_or_edit_image(prompt, uploaded_img):
             contents=contents,
         )
 
-        # Debug
-        print("Gemini Response:", response)
+        # Debug log
+        print("Gemini Response received.")
 
         for part in response.candidates[0].content.parts:
             if part.inline_data is not None:
@@ -90,16 +101,15 @@ def generate_or_edit_image(prompt, uploaded_img):
                     image_bytes = base64.b64decode(data)
                 else:
                     image_bytes = data
-
                 image = Image.open(BytesIO(image_bytes))
                 return image
 
-        return "⚠️ No image returned. Try another prompt."
+        return "⚠️ No image generated. Try refining your prompt."
 
     except Exception as e:
         return f"❌ Error: {e}"
 
-# Gradio UI
+# Gradio app
 demo = gr.Interface(
     fn=generate_or_edit_image,
     inputs=[
@@ -108,7 +118,10 @@ 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 it."
+    description=(
+        "Enter a text prompt to generate a new image.\n"
+        "If you upload an image, Gemini will use it as a reference and apply your requested changes."
+    )
 )
 
 if __name__ == "__main__":