Update app.py

app.py

@@ -6,27 +6,29 @@ import gradio as gr
 import cv2
 import numpy as np
 
-def image_to_silhouette(image):
+def image_to_silhouette_contours(image):
     try:
-        print("Image shape:", image.shape)
+        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 shape:", gray.shape)
+        print("Gray image shape:", gray.shape)
         # Aplicar bordes
         edges = cv2.Canny(gray, 50, 150)
-        print("Edges shape:", edges.shape)
-        # Crear máscara con forma de silueta negra
-        silhouette = np.zeros_like(image)
-        silhouette[:,:,0] = edges
-        silhouette[:,:,1] = edges
-        silhouette[:,:,2] = edges
-
+        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"
 
-
 # Crear interfaz de usuario con Gradio
 gr.Interface(
     image_to_silhouette,