Create app.py

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+from rembg import remove
+from PIL import Image
+import numpy as np
+import time
+import random
+
+def washing_machine(image):
+    # Convert to PIL Image if it's a numpy array
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    
+    # Simulate washing process with progress updates
+    for i in range(1, 6):
+        time.sleep(0.5)  # Simulate washing cycles
+        yield f"Washing cycle {i}/5...", image
+    
+    # Actual background removal
+    result = remove(image)
+    
+    # Add some "spin cycle" effects
+    for angle in [15, -15, 0]:
+        rotated = result.rotate(angle)
+        yield "Spin cycle complete!", rotated
+    
+    # Final clean image
+    yield "Your image is clean and background-free!", result
+
+# Custom CSS for washing machine theme
+css = """
+#wash-btn {background-color: #00a8ff; color: white; border: none; padding: 10px 20px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 12px;}
+#wash-btn:hover {background-color: #0095e6;}
+.title {font-family: Arial; color: #333; text-align: center;}
+.description {font-family: Arial; color: #666; text-align: center;}
+#output-image {border: 5px dashed #00a8ff; border-radius: 10px;}
+#input-image {border: 5px dashed #ccc; border-radius: 10px;}
+"""
+
+with gr.Blocks(css=css) as demo:
+    gr.Markdown("<h1 class='title'>🚀 Background Washing Machine �</h1>")
+    gr.Markdown("<p class='description'>Upload your image and we'll wash the background right off!</p>")
+    
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(label="Dirty Image (with background)")
+            wash_btn = gr.Button("Start Washing!", elem_id="wash-btn")
+        
+        with gr.Column():
+            status = gr.Textbox(label="Washing Status")
+            output_image = gr.Image(label="Clean Image (no background)")
+    
+    wash_btn.click(
+        washing_machine,
+        inputs=input_image,
+        outputs=[status, output_image]
+    )
+    
+    gr.Markdown("### How it works:")
+    gr.Markdown("1. Upload your image 🖼️\n2. Click the washing button �\n3. Wait while we wash away the background 🫧\n4. Get your clean, background-free image! ✅")
+
+demo.launch()