Update app.py

app.py

@@ -3,27 +3,107 @@ 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):
-    remover = Remover()
-    if isinstance(image, Image.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()
         output = remover.process(image)
-    elif isinstance(image, np.ndarray):
-        image_pil = Image.fromarray(image)
-        output = remover.process(image_pil)
-    else:
-        raise TypeError("Unsupported image type")
-    return output
+        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;
+}
+"""
 
-iface = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(label="Upload Image"),
-    outputs=gr.Image(label="Output Image"),
-    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"
-)
+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.
+    """)
+    
+    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")
+    
+    # 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
+    )
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()