Update app.py

app.py

@@ -130,27 +130,27 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("## Upload or select an additional image")
     
     with gr.Row():
-        uploaded_image = gr.Image(label="Upload Image")
+        uploaded_image = gr.Image(label="Upload Image", type="pil")
         image_url = gr.Textbox(label="Image URL", placeholder="Enter image URL")
         use_generated_image = gr.Button("Use Generated Image")
     
     additional_image_output = gr.Image(label="Selected Image", show_label=False)
 
-    def select_image(uploaded_image, image_url, use_generated):
+    def select_image(uploaded_image, image_url, use_generated=False):
         if use_generated:
-            return result.value  # Return the value of the generated image
+            return result.value
         elif uploaded_image is not None:
             return uploaded_image
         elif image_url:
             try:
-                return gr.Image.load(image_url)
+                img = gr.Image.load(image_url)
+                return img
             except Exception as e:
                 return f"Failed to load image from URL: {e}"
         return None
 
-    # Updated click and change triggers
-    use_generated_image.click(lambda: select_image(None, None, True), inputs=[], outputs=[additional_image_output])
-    uploaded_image.change(select_image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=[additional_image_output])
-    image_url.submit(select_image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=[additional_image_output])
+    use_generated_image.click(fn=lambda: select_image(None, None, True), inputs=[], outputs=additional_image_output)
+    uploaded_image.change(fn=select_image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
+    image_url.submit(fn=select_image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
 
 demo.launch()