Update app.py

app.py

@@ -1,42 +1,81 @@
 import gradio as gr
-from PIL import Image # Needed create blank "proxy image in the functions below
+from gradio_client import Client, handle_file
+from PIL import Image
+import requests
 
+# Initialize the client
+client = Client("not-lain/background-removal")
 
-def dummy_process_image(image):
-    """Заглушка для обработки изображения. Возвращает пустое изображение."""
-    return Image.new("RGB", (256, 256), "white"),  Image.new("RGB", (256, 256), "black")
+def process_image_via_api(image):
+    result = client.predict(
+        image=handle_file(image),
+        api_name="/image"
+    )
+    # Convert the output tuple to PIL images and return
+    if result:
+        return (Image.open(result[0]), Image.open(result[1]))
+    return None, None
 
+def process_url_via_api(url):
+    result = client.predict(
+        image=url,
+        api_name="/text"
+    )
+    # Convert the output tuple to PIL images and return
+    if result:
+        return (Image.open(result[0]), Image.open(result[1]))
+    return None, None
 
-def dummy_process_url(url):
+def process_file_via_api(f):
+    result = client.predict(
+        f=handle_file(f),
+        api_name="/png"
+    )
+    # Return the path to the saved PNG file
+    if result:
+        return result
+    return None
 
-    """Заглушка для обработки URL. Возвращает пустое изображение."""
-    return Image.new("RGB", (256, 256), "white"),  Image.new("RGB", (256, 256), "black")
+# Example images
+chameleon = "butterfly.jpg"
+url_example = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"
 
-def dummy_process_file(filepath):
-    """Заглушка для обработки файла. Возвращает имя фиктивного файла."""
-    return "dummy_output.png"
+image_upload = gr.Image(label="Upload an image")
+url_input = gr.Textbox(label="Paste an image URL")
+image_file_upload = gr.Image(label="Upload an image", type="filepath")
+output_file = gr.File(label="Output PNG File")
+sliders_processed = gr.Image(label="Processed Image")
+slider_origin = gr.Image(label="Original Image")
 
+tab1 = gr.Interface(
+    fn=process_image_via_api, 
+    inputs=image_upload, 
+    outputs=[sliders_processed, slider_origin], 
+    examples=[chameleon], 
+    api_name="/image_api"
+)
 
-with gr.Blocks() as demo:
-    with gr.Tab("Image Upload"):
-        with gr.Row():
-            image_upload = gr.Image(label="Upload an image")
-            output_image = gr.Image(label="Processed Image", type="pil")
-        image_upload.change(dummy_process_image, image_upload, output_image)
+tab2 = gr.Interface(
+    fn=process_url_via_api, 
+    inputs=url_input, 
+    outputs=[sliders_processed, slider_origin], 
+    examples=[url_example], 
+    api_name="/url_api"
+)
 
+tab3 = gr.Interface(
+    fn=process_file_via_api, 
+    inputs=image_file_upload, 
+    outputs=output_file, 
+    examples=[chameleon], 
+    api_name="/png_api"
+)
 
-    with gr.Tab("URL Input"):
-         with gr.Row():
-            url_input = gr.Textbox(label="Paste an image URL")
-            output_url = gr.Image(label="Processed Image from URL", type="pil")
-         url_input.submit(dummy_process_url, url_input, output_url) # Or "submit"  is more appropriate for text inputs and not change.
-
-
-    with gr.Tab("File Output"):
-        with gr.Row():
-            image_file_upload = gr.Image(label="Upload an image",  type ="filepath")
-            output_file = gr.File(label="Output PNG File")
-        image_file_upload.change(dummy_process_file, image_file_upload, output_file)
+demo = gr.TabbedInterface(
+    [tab1, tab2, tab3], 
+    ["Image Upload", "URL Input", "File Output"], 
+    title="Background Removal Tool using API"
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(show_error=True)