Update app.py

app.py

@@ -3,107 +3,32 @@ import gradio as gr
 from transparent_background import Remover
 from PIL import Image
 import numpy as np
-import io
-import time
-
-# Initialize the remover once to improve performance
-remover = Remover()
 
 @spaces.GPU
 def remove_background(image):
-    try:
-        # Convert input to PIL Image if needed
-        if isinstance(image, np.ndarray):
-            image = Image.fromarray(image)
-        
-        # Process the image
-        start_time = time.time()
+    remover = Remover()
+    if isinstance(image, Image.Image):
         output = remover.process(image)
-        processing_time = time.time() - start_time
-        
-        # Convert to PNG with transparency
-        img_byte_arr = io.BytesIO()
-        output.save(img_byte_arr, format='PNG')
-        img_byte_arr.seek(0)
-        
-        return img_byte_arr, f"Processing time: {processing_time:.2f}s"
-    
-    except Exception as e:
-        return None, f"Error: {str(e)}"
-
-# Custom CSS for a more professional look
-custom_css = """
-#header {
-    text-align: center;
-}
-#processing-info {
-    text-align: center;
-    color: #666;
-    font-size: 0.9em;
-}
-.output-image {
-    border: 1px solid #e0e0e0;
-    border-radius: 8px;
-}
-.download-btn {
-    background-color: #f36d3a !important;
-    color: white !important;
-    border: none !important;
-}
-.download-btn:hover {
-    background-color: #e55a2b !important;
-}
-"""
-
-with gr.Blocks(title="Professional Background Remover", css=custom_css, theme="Yntec/HaleyCH_Theme_Orange") as demo:
-    gr.Markdown("""
-    # ✂️ Professional Image Background Remover
-    Remove backgrounds from your images with high precision. Perfect for product photos, portraits, and more.
-    """)
+    elif isinstance(image, np.ndarray):
+        image_pil = Image.fromarray(image)
+        output = remover.process(image_pil)
+    else:
+        raise TypeError("Unsupported image type")
     
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(label="Upload Image", type="filepath")
-            process_btn = gr.Button("Remove Background", variant="primary")
-            
-        with gr.Column():
-            output_image = gr.Image(label="Result with Transparency", type="filepath", elem_classes="output-image")
-            processing_info = gr.Markdown(elem_id="processing-info")
-            
-            with gr.Row():
-                download_btn = gr.DownloadButton("Download PNG", variant="primary", elem_classes="download-btn")
+    # Convert to PNG to preserve transparency
+    if output.mode != 'RGBA':
+        output = output.convert('RGBA')
     
-    # Examples
-    gr.Examples(
-        examples=[
-            ["https://upload.wikimedia.org/wikipedia/commons/thumb/4/41/Sunflower_from_Silesia2.jpg/800px-Sunflower_from_Silesia2.jpg"],
-            ["https://upload.wikimedia.org/wikipedia/commons/thumb/b/b6/Image_created_with_a_mobile_phone.png/800px-Image_created_with_a_mobile_phone.png"]
-        ],
-        inputs=input_image,
-        outputs=[output_image, processing_info],
-        fn=remove_background,
-        cache_examples=True,
-        label="Click on any example below to start"
-    )
-    
-    # Set up processing
-    process_btn.click(
-        fn=remove_background,
-        inputs=input_image,
-        outputs=[output_image, processing_info]
-    )
-    
-    # Set up download
-    def get_download_link(output_data):
-        if output_data:
-            return output_data
-        return None
-    
-    output_image.change(
-        fn=get_download_link,
-        inputs=output_image,
-        outputs=download_btn
-    )
+    return output
+
+iface = gr.Interface(
+    fn=remove_background,
+    inputs=gr.Image(label="Upload Image"),
+    outputs=gr.Image(label="Output Image", type="pil", format="png"),
+    title="✂️ Image Background Remover ✂️",
+    description="⚠️ Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
+    theme="Yntec/HaleyCH_Theme_Orange"
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()