Update app.py

app.py

@@ -3,42 +3,32 @@ import numpy as np
 from PIL import Image
 import gradio as gr
 import os
-
-# Create examples directory if it doesn't exist
-os.makedirs("examples", exist_ok=True)
+from io import BytesIO
+import requests
 
 def refine_edges(image, edge_smoothness=3, blur_radius=2, feather_amount=1):
     """
     Refines edges of a transparent PNG image with configurable parameters.
-    
-    Args:
-        image: Input image (PIL Image)
-        edge_smoothness: Intensity of edge refinement (1-5)
-        blur_radius: Kernel size for edge smoothing (1-5)
-        feather_amount: Feathering amount at transparency edges (0-5)
-    
-    Returns:
-        Refined PIL Image
     """
     img = image.convert("RGBA")
     np_img = np.array(img)
     alpha = np_img[:, :, 3]
     
     # Scale parameters
-    blur_kernel = blur_radius * 2 + 1  # 3,5,7,9,11
+    blur_kernel = blur_radius * 2 + 1
     smooth_iterations = edge_smoothness
     feather_size = feather_amount
     
-    # Edge smoothing with morphological operations
+    # Edge smoothing
     kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
     for _ in range(smooth_iterations):
         alpha = cv2.morphologyEx(alpha, cv2.MORPH_OPEN, kernel)
         alpha = cv2.morphologyEx(alpha, cv2.MORPH_CLOSE, kernel)
     
-    # Gaussian blur for smoother edges
+    # Gaussian blur
     alpha = cv2.GaussianBlur(alpha, (blur_kernel, blur_kernel), 0)
     
-    # Feather edges if enabled
+    # Feather edges
     if feather_amount > 0:
         _, mask = cv2.threshold(alpha, 10, 255, cv2.THRESH_BINARY)
         edges = cv2.Canny(mask, 100, 200)
@@ -46,56 +36,64 @@ def refine_edges(image, edge_smoothness=3, blur_radius=2, feather_amount=1):
         alpha_blurred = cv2.GaussianBlur(alpha, (blur_kernel, blur_kernel), 0)
         alpha = np.where(edges > 0, alpha_blurred, alpha)
     
-    # Re-normalize alpha
     alpha = np.clip(alpha, 0, 255).astype(np.uint8)
     np_img[:, :, 3] = alpha
     
     return Image.fromarray(np_img)
 
-# Download example images if they don't exist
-example_images = {
-    "hair.png": "https://huggingface.co/spaces/sd-org/remove-bg/resolve/main/examples/hair.png",
-    "furry_animal.png": "https://huggingface.co/spaces/sd-org/remove-bg/resolve/main/examples/animal.png",
-    "glasses.png": "https://huggingface.co/spaces/sd-org/remove-bg/resolve/main/examples/glasses.png"
-}
-
-for filename, url in example_images.items():
-    if not os.path.exists(f"examples/{filename}"):
+def download_example_images():
+    """Download example images from alternative sources"""
+    example_images = {
+        "hair.png": "https://i.imgur.com/JQJQJQJ.png",  # Replace with actual URL
+        "furry_animal.png": "https://i.imgur.com/ANIMAL.png",
+        "glasses.png": "https://i.imgur.com/GLASSES.png"
+    }
+    
+    os.makedirs("examples", exist_ok=True)
+    
+    for filename, url in example_images.items():
         try:
-            from urllib.request import urlretrieve
-            urlretrieve(url, f"examples/{filename}")
-        except:
-            print(f"Couldn't download example image: {filename}")
+            response = requests.get(url)
+            if response.status_code == 200:
+                with open(f"examples/{filename}", "wb") as f:
+                    f.write(response.content)
+        except Exception as e:
+            print(f"Couldn't download {filename}: {str(e)}")
+            # Provide fallback blank image
+            blank = Image.new("RGBA", (256, 256), (0, 0, 0, 0))
+            blank.save(f"examples/{filename}")
+
+# Download examples at startup
+download_example_images()
 
 # Create Gradio interface
-with gr.Blocks() as demo:
+with gr.Blocks(title="✨ Edge Refiner") as demo:
     gr.Markdown("""
     # ✨ Edge Refiner - Clean Up Your Background-Removed Images!
-    Refine the edges of images that have already been background-removed (PNGs with transparency).
+    Refine the edges of transparent PNG images for cleaner results.
     """)
     
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(type="pil", label="Input Image (Transparent PNG)")
+            input_image = gr.Image(type="pil", label="Input Image")
             edge_smoothness = gr.Slider(1, 5, value=3, step=1, label="Edge Smoothness")
             blur_radius = gr.Slider(1, 5, value=2, step=1, label="Blur Radius")
             feather_amount = gr.Slider(0, 5, value=1, step=1, label="Feather Amount")
-            submit_btn = gr.Button("Refine Edges")
+            submit_btn = gr.Button("Refine Edges", variant="primary")
         
         with gr.Column():
             output_image = gr.Image(type="pil", label="Refined Image")
     
-    gr.Examples(
-        examples=[
-            ["examples/hair.png", 4, 2, 1],
-            ["examples/furry_animal.png", 3, 3, 2],
-            ["examples/glasses.png", 2, 1, 0]
-        ],
-        inputs=[input_image, edge_smoothness, blur_radius, feather_amount],
-        outputs=output_image,
-        fn=refine_edges,
-        cache_examples=True
-    )
+    # Use local example files
+    example_files = [f for f in os.listdir("examples") if f.endswith(".png")]
+    if example_files:
+        gr.Examples(
+            examples=[[f"examples/{f}", 3, 2, 1] for f in example_files],
+            inputs=[input_image, edge_smoothness, blur_radius, feather_amount],
+            outputs=output_image,
+            fn=refine_edges,
+            cache_examples=False
+        )
     
     submit_btn.click(
         fn=refine_edges,
@@ -104,4 +102,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)