Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from gradio_client import Client, handle_file
 from PIL import Image
+import requests
 import tempfile
 import os
 
@@ -12,13 +13,20 @@ def process_image_via_api(image):
         image=handle_file(image),
         api_name="/image"
     )
-    # Convert the output tuple to PIL images and return
+    # Convert the output tuple to PIL images and save them to temporary files
     if result:
         processed_image_path = result[0]
         origin_image_path = result[1]
         processed_image = Image.open(processed_image_path)
         origin_image = Image.open(origin_image_path)
-        return (processed_image, origin_image)
+        
+        # Save images to temporary files
+        processed_temp_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
+        origin_temp_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
+        processed_image.save(processed_temp_path)
+        origin_image.save(origin_temp_path)
+        
+        return (processed_temp_path, origin_temp_path)
     return None, None
 
 def process_url_via_api(url):
@@ -26,13 +34,20 @@ def process_url_via_api(url):
         image=url,
         api_name="/text"
     )
-    # Convert the output tuple to PIL images and return
+    # Convert the output tuple to PIL images and save them to temporary files
     if result:
         processed_image_path = result[0]
         origin_image_path = result[1]
         processed_image = Image.open(processed_image_path)
         origin_image = Image.open(origin_image_path)
-        return (processed_image, origin_image)
+        
+        # Save images to temporary files
+        processed_temp_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
+        origin_temp_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
+        processed_image.save(processed_temp_path)
+        origin_image.save(origin_temp_path)
+        
+        return (processed_temp_path, origin_temp_path)
     return None, None
 
 def process_file_via_api(f):
@@ -50,8 +65,8 @@ chameleon = "butterfly.jpg"
 url_example = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
 
 # Tab 1: Image Upload
-slider1_processed = gr.Image(label="Processed Image", type="pil")
-slider1_origin = gr.Image(label="Original Image", type="pil")
+slider1_processed = gr.Image(label="Processed Image", type="filepath")
+slider1_origin = gr.Image(label="Original Image", type="filepath")
 image_upload = gr.Image(label="Upload an image")
 tab1 = gr.Interface(
     fn=process_image_via_api, 
@@ -62,8 +77,8 @@ tab1 = gr.Interface(
 )
 
 # Tab 2: URL Input
-slider2_processed = gr.Image(label="Processed Image", type="pil")
-slider2_origin = gr.Image(label="Original Image", type="pil")
+slider2_processed = gr.Image(label="Processed Image", type="filepath")
+slider2_origin = gr.Image(label="Original Image", type="filepath")
 url_input = gr.Textbox(label="Paste an image URL")
 tab2 = gr.Interface(
     fn=process_url_via_api,