Update app.py

app.py

@@ -6,18 +6,30 @@ os.system("pip install pillow")
 import gradio as gr
 import cv2
 import numpy as np
+from PIL import Image
 
-def image_to_silhouette(image):
-    # Convertir imagen a escala de grises
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    # Aplicar bordes
-    edges = cv2.Canny(gray, 50, 150)
-    # Crear máscara con forma de silueta negra
-    silhouette = np.zeros_like(image)
-    silhouette[:,:,0] = edges
-    silhouette[:,:,1] = edges
-    silhouette[:,:,2] = edges
-    return silhouette
+def image_to_silhouette_contours(image):
+    try:
+        # Convertir imagen a escala de grises
+        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+        # Aplicar bordes
+        edges = cv2.Canny(gray, 50, 150)
+        # busca los contornos de los objetos en la imagen
+        contours, _ = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+        # Crear una imagen negra para dibujar los contornos
+        silhouette = np.zeros((image.shape[0], image.shape[1], 3), dtype=np.uint8)
+        # Dibuja los contornos en la imagen negra
+        cv2.drawContours(silhouette, contours, -1, (255, 255, 255), 1)
+
+        return Image.fromarray(silhouette)
+    except Exception as e:
+        print("An error occurred:", e)
+        return "Error"
 
 # Crear interfaz de usuario con Gradio
-gr.Interface(image_to_silhouette, gr.inputs.Image(label="Input"), gr.outputs.Image(type="jpeg", label="Output"), css="Footer {visibility: hidden}").launch()
+gr.Interface(
+    image_to_silhouette_contours, 
+    gr.inputs.Image(type="filepath", label="Input"), 
+    gr.outputs.Image(type="pil", format="jpeg", label="Output"),
+    css="Footer {visibility: hidden}"
+    ).launch()