metadata v9

app.py

@@ -51,36 +51,10 @@ def build_database():
 
         for j, item in enumerate(batch):
             try:
-                img_data = item["image"]
-                print(f"Debug - Image data type: {type(img_data)}")
-                print(f"Debug - Image data content: {img_data}")
-
-                # Handle different image data formats
-                if hasattr(img_data, 'numpy'):
-                    # If it's a tensor/array, convert to PIL
-                    img = Image.fromarray(img_data.numpy())
-                elif isinstance(img_data, dict):
-                    # If it's a dictionary, try to get the image data
-                    if "bytes" in img_data:
-                        img = Image.open(BytesIO(img_data["bytes"]))
-                    elif "path" in img_data:
-                        img = Image.open(img_data["path"])
-                    else:
-                        print(f"❌ Formato de diccionario no soportado: {img_data.keys()}")
-                        continue
-                elif isinstance(img_data, str):
-                    if img_data.startswith("http"):
-                        response = requests.get(img_data, timeout=10)
-                        response.raise_for_status()
-                        img = Image.open(BytesIO(response.content))
-                    else:
-                        img = Image.open(img_data)
-                elif isinstance(img_data, Image.Image):
-                    img = img_data
-                else:
-                    print(f"❌ Formato desconocido en imagen {i+j}: {type(img_data)}")
-                    continue
-
+                # The image is already a PIL Image object
+                img = item["image"]
+                
+                # Ensure image is in RGB mode
                 img = img.convert("RGB")
                 img_processed = preprocess_image(img)
                 embedding = DeepFace.represent(
@@ -95,9 +69,6 @@ def build_database():
                 del img_processed
                 gc.collect()
 
-            except (requests.RequestException, UnidentifiedImageError) as e:
-                print(f"❌ Falló descarga/lectura de imagen {i+j}: {e}")
-                continue
             except Exception as e:
                 print(f"❌ No se pudo procesar imagen {i+j}: {str(e)}")
                 print(f"Error details: {type(e).__name__}")