Create app.py

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+import cv2
+import numpy as np
+from PIL import Image
+
+def refine_edges(image: Image.Image):
+    # Convert to OpenCV format (RGBA)
+    image_np = np.array(image)
+    if image_np.shape[2] == 4:
+        alpha_channel = image_np[:, :, 3]
+    else:
+        return image  # Return original if no alpha channel
+    
+    # Step 1: Blur the alpha channel
+    blurred = cv2.GaussianBlur(alpha_channel, (5, 5), 0)
+
+    # Step 2: Morphological refinement
+    kernel = np.ones((3,3), np.uint8)
+    refined = cv2.morphologyEx(blurred, cv2.MORPH_OPEN, kernel)
+
+    # Step 3: Recombine channels
+    b, g, r = image_np[:, :, 0], image_np[:, :, 1], image_np[:, :, 2]
+    rgba = cv2.merge((b, g, r, refined))
+
+    return Image.fromarray(rgba)
+
+iface = gr.Interface(
+    fn=refine_edges,
+    inputs=gr.Image(type="pil", image_mode="RGBA"),
+    outputs="image",
+    title="Edge Refiner",
+    description="Upload a background-removed PNG and get smoother edges!"
+)
+
+if __name__ == "__main__":
+    iface.launch()