Update app.py

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 from gradio_client import Client, handle_file
 from PIL import Image
 import requests
-import io
+import tempfile
+import os
 
-# Initialize the client
+# Инициализируем клиент
 client = Client("not-lain/background-removal")
 
 def process_image_via_api(image):
@@ -14,7 +15,11 @@ def process_image_via_api(image):
     )
     # Convert the output tuple to PIL images and return
     if result:
-        return (Image.open(result[0]), Image.open(result[1]))
+        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)
     return None, None
 
 def process_url_via_api(url):
@@ -24,7 +29,11 @@ def process_url_via_api(url):
     )
     # Convert the output tuple to PIL images and return
     if result:
-        return (Image.open(result[0]), Image.open(result[1]))
+        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)
     return None, None
 
 def process_file_via_api(f):
@@ -32,61 +41,55 @@ def process_file_via_api(f):
         f=handle_file(f),
         api_name="/png"
     )
-    # Ensure the result is a valid file path
-    if isinstance(result, str):
+    # Return the path to the saved PNG file
+    if result:
         return result
-    elif isinstance(result, bytes):
-        # If the result is bytes, convert to image and save to a file
-        image = Image.open(io.BytesIO(result))
-        output_path = "output.png"
-        image.save(output_path)
-        return output_path
     return None
 
-# Example images
+# Пример изображений
 chameleon = "butterfly.jpg"
 url_example = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
 
 # Tab 1: Image Upload
-sliders_processed_tab1 = gr.Image(label="Processed Image")
-sliders_origin_tab1 = gr.Image(label="Original Image")
-image_upload_tab1 = gr.Image(label="Upload an image")
+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, 
-    inputs=image_upload_tab1, 
-    outputs=[sliders_processed_tab1, sliders_origin_tab1], 
+    inputs=image_upload, 
+    outputs=[slider1_processed, slider1_origin], 
     examples=[chameleon], 
     api_name="/image_api"
 )
 
 # Tab 2: URL Input
-sliders_processed_tab2 = gr.Image(label="Processed Image")
-sliders_origin_tab2 = gr.Image(label="Original Image")
-url_input_tab2 = gr.Textbox(label="Paste an image URL")
+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, 
-    inputs=url_input_tab2, 
-    outputs=[sliders_processed_tab2, sliders_origin_tab2], 
+    inputs=url_input, 
+    outputs=[slider2_processed, slider2_origin], 
     examples=[url_example], 
     api_name="/url_api"
 )
 
 # Tab 3: File Output
-output_file_tab3 = gr.File(label="Output PNG File")
-image_file_upload_tab3 = gr.Image(label="Upload an image", type="filepath")
+output_file = gr.File(label="Output PNG File")
+image_file_upload = gr.Image(label="Upload an image", type="filepath")
 tab3 = gr.Interface(
     fn=process_file_via_api, 
-    inputs=image_file_upload_tab3, 
-    outputs=output_file_tab3, 
-    examples=[chameleon], 
+    inputs=image_file_upload, 
+    outputs=output_file, 
+    examples=["butterfly.jpg"], 
     api_name="/png_api"
 )
 
-# Create the Tabbed Interface
+# Создаем интерфейс с вкладками
 demo = gr.TabbedInterface(
     [tab1, tab2, tab3], 
     ["Image Upload", "URL Input", "File Output"], 
-    title="Background Removal Tool using API"
+    title="Background Removal Tool"
 )
 
 if __name__ == "__main__":