Update app.py

app.py

@@ -28,10 +28,13 @@ def generate_image(prompt, image=None, output_filename="generated_image"):
         img_byte_arr = io.BytesIO()
         image.save(img_byte_arr, format="PNG")
         img_bytes = img_byte_arr.getvalue()
-        parts.append(types.Part.from_inline_data(
-            mime_type="image/png",
-            data=img_bytes
-        ))
+        # Add the image as a Part with inline_data
+        parts.append({
+            "inline_data": {
+                "mime_type": "image/png",
+                "data": img_bytes
+            }
+        })
 
     contents = [
         types.Content(