Update app.py

app.py

@@ -6,33 +6,17 @@ import gradio as gr
 import cv2
 import numpy as np
 
-def image_to_silhouette_contours(image):
-    try:
-        print("Original image type:", type(image))
-        print("Original image shape:", image.shape)
-        # Convertir imagen a escala de grises
-        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-        print("Gray image shape:", gray.shape)
-        # Aplicar bordes
-        edges = cv2.Canny(gray, 50, 150)
-        print("Edges image shape:", edges.shape)
-        # 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)
-    
-        print("Silhouette image shape:", silhouette.shape)
-        return silhouette
-    except Exception as e:
-        print("An error occurred:", e)
-        return "Error"
+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
 
 # Crear interfaz de usuario con Gradio
-gr.Interface(
-    image_to_silhouette_contours, 
-    gr.inputs.Image(type="filepath", label="Input"), 
-    gr.outputs.Image(type="pil", label="Output"),
-    css="Footer {visibility: hidden}"
-    ).launch()
+gr.Interface(image_to_silhouette, gr.inputs.Image(label="Input"), gr.outputs.Image(label="Output"), css="Footer {visibility: hidden}").launch()