verification flow fix

.gitignore

@@ -1 +1,2 @@
-docker-compose.yml
+docker-compose.yml
+env

app.py

@@ -2,6 +2,10 @@ from fastapi import FastAPI
 from deepface import DeepFace
 from pydantic import BaseModel
 from fastapi.responses import JSONResponse
+import base64
+import cv2
+import numpy as np
+import requests
 
 
 class Verify(BaseModel):
@@ -10,6 +14,36 @@ class Verify(BaseModel):
 
 app = FastAPI()
 
+def load_image(source: str):
+    """
+    Load an image from a URL, base64 data URI, or local file path into a numpy array (BGR).
+    """
+    if isinstance(source, str):
+        if source.startswith("http://") or source.startswith("https://"):
+            # Download and decode
+            resp = requests.get(source, headers={"User-Agent": "Mozilla/5.0"}, timeout=20)
+            resp.raise_for_status()
+            data = np.frombuffer(resp.content, dtype=np.uint8)
+            img = cv2.imdecode(data, cv2.IMREAD_COLOR)
+            if img is None:
+                raise ValueError(f"Failed to decode image from URL: {source}")
+            return img
+        if source.startswith("data:image"):
+            # data URI: data:image/<type>;base64,<payload>
+            b64_payload = source.split(",", 1)[1] if "," in source else source
+            binary = base64.b64decode(b64_payload)
+            data = np.frombuffer(binary, dtype=np.uint8)
+            img = cv2.imdecode(data, cv2.IMREAD_COLOR)
+            if img is None:
+                raise ValueError("Failed to decode image from base64 data")
+            return img
+        # Fallback: treat as local path
+        img = cv2.imread(source)
+        if img is None:
+            raise ValueError(f"Failed to load local image: {source}")
+        return img
+    raise TypeError("Unsupported image source type")
+
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
@@ -20,19 +54,48 @@ def verify(v: Verify):
     selfie = data["image"]
     gallery = data["images"]
 
-    import requests
-    res = requests.get(gallery[0], stream=True)
-    with open("x.jpg", "wb") as out_file:
-        out_file.write(res.content)
-
     true_count = 0
+    print(selfie)
+    try:
+        selfie_img = load_image(selfie)
+    except Exception as e:
+        print(f"Failed to load selfie image: {e}")
+        return JSONResponse(content={"verified": False, "image": None, "error": "failed_to_load_selfie"})
+
     for image in gallery:
+        print(image)
+        try:
+            gallery_img = load_image(image)
+        except Exception as e:
+            print(f"Failed to load gallery image {image}: {e}")
+            continue
         try:
-            result = DeepFace.verify(img1_path=selfie, img2_path=image)
+            # Use a more robust detector; fall back if detection fails
+            result = DeepFace.verify(
+                img1_path=selfie_img,
+                img2_path=gallery_img,
+                detector_backend="retinaface"
+            )
             if result.get("verified", False):
                 true_count += 1
                 if true_count >= 2:
                     return JSONResponse(content={"verified": True, "image": image})
         except Exception as e:
-            print(e)
+            msg = str(e)
+            print(f"DeepFace verification error for {image}: {msg}")
+            # Optional fallback without strict detection to avoid generic img2_path errors
+            if "Face could not be detected" in msg or "No face" in msg:
+                try:
+                    result = DeepFace.verify(
+                        img1_path=selfie_img,
+                        img2_path=gallery_img,
+                        detector_backend="retinaface",
+                        enforce_detection=False
+                    )
+                    if result.get("verified", False):
+                        true_count += 1
+                        if true_count >= 2:
+                            return JSONResponse(content={"verified": True, "image": image})
+                except Exception as e2:
+                    print(f"DeepFace fallback error for {image}: {e2}")
     return JSONResponse(content={"verified": False, "image": None})