Update app.py

app.py

@@ -36,7 +36,6 @@
 
 
 
-
 import os
 import gradio as gr
 from google import genai
@@ -55,9 +54,13 @@ def generate_or_edit_image(prompt, uploaded_img):
 
     # If user uploads an image, include it in the request
     if uploaded_img is not None:
-        # Convert uploaded image to bytes (keep original format if possible)
+        # Validate image format
+        if uploaded_img.format not in ["JPEG", "JPG", "PNG"]:
+            return "❌ Please upload a JPG, JPEG, or PNG image."
+
+        # Convert uploaded image to PNG bytes (safe for Gemini)
         buffered = BytesIO()
-        uploaded_img.save(buffered, format="PNG")  # convert to PNG for consistency
+        uploaded_img.save(buffered, format="PNG")
         image_bytes = buffered.getvalue()
 
         contents.append({
@@ -85,8 +88,7 @@ demo = gr.Interface(
         gr.Textbox(label="Enter your prompt"),
         gr.Image(
             label="Upload an image (optional)",
-            type="pil",
-            file_types=[".jpg", ".jpeg", ".png"]  # allow JPG, JPEG, PNG
+            type="pil"   # just keep type=pil, no file_types in your version
         )
     ],
     outputs=gr.Image(label="Generated/Edited Image"),
@@ -96,4 +98,3 @@ demo = gr.Interface(
 
 if __name__ == "__main__":
     demo.launch()
-