Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
-from gradio_imageslider import ImageSlider
 from gradio_client import Client, handle_file
 from PIL import Image
+import requests
 import tempfile
 import os
 
@@ -13,20 +13,13 @@ def process_image_via_api(image):
         image=handle_file(image),
         api_name="/image"
     )
-    # Convert the output tuple to PIL images and save them to temporary files
+    # Convert the output tuple to PIL images and return
     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)
-        
-        # 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 (processed_image, origin_image)
     return None, None
 
 def process_url_via_api(url):
@@ -34,20 +27,13 @@ def process_url_via_api(url):
         image=url,
         api_name="/text"
     )
-    # Convert the output tuple to PIL images and save them to temporary files
+    # Convert the output tuple to PIL images and return
     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)
-        
-        # 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 (processed_image, origin_image)
     return None, None
 
 def process_file_via_api(f):
@@ -65,8 +51,8 @@ chameleon = "butterfly.jpg"
 url_example = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
 
 # Tab 1: Image Upload
-slider1_processed = ImageSlider(label="Processed Image", type="filepath")
-slider1_origin = ImageSlider(label="Original Image", type="filepath")
+slider1_processed = ImageSlider(label="Processed Image", type="pil")
+slider1_origin = ImageSlider(label="Original Image", type="pil")
 image_upload = gr.Image(label="Upload an image")
 tab1 = gr.Interface(
     fn=process_image_via_api, 
@@ -77,8 +63,8 @@ tab1 = gr.Interface(
 )
 
 # Tab 2: URL Input
-slider2_processed = ImageSlider(label="Processed Image", type="filepath")
-slider2_origin = ImageSlider(label="Original Image", type="filepath")
+slider2_processed = ImageSlider(label="Processed Image", type="pil")
+slider2_origin = ImageSlider(label="Original Image", type="pil")
 url_input = gr.Textbox(label="Paste an image URL")
 tab2 = gr.Interface(
     fn=process_url_via_api, 
@@ -95,7 +81,7 @@ tab3 = gr.Interface(
     fn=process_file_via_api, 
     inputs=image_file_upload, 
     outputs=output_file, 
-    examples=[chameleon], 
+    examples=["butterfly.jpg"], 
     api_name="/png_api"
 )