Update app.py

app.py

@@ -81,32 +81,37 @@ async def compare_faces(req: CompareRequest):
     # Cosine similarity
     similarity = np.dot(emb1, emb2) / (np.linalg.norm(emb1) * np.linalg.norm(emb2))
 
-    matched = similarity > 0.55  # threshold; adjust as needed
+    matched = similarity > 0.55  # threshold
 
     return {
         "similarity": float(similarity),
         "match": matched
     }
 
-# ---------- Gradio UI ----------
-def gradio_ui(img1, img2):
-    import base64
-    import io
-
-    def to_b64(pil_img):
-        buf = io.BytesIO()
-        pil_img.save(buf, format="JPEG")
-        return base64.b64encode(buf.getvalue()).decode()
+# ---------- Gradio UI (TEXT INPUT instead of upload) ----------
+def gradio_ui(img1_text, img2_text):
+    """
+    Accepts:
+      - base64 string
+      - or URL
+    Automatically detects which one.
+    """
+
+    # --- Determine type of input ---
+    def load_any(input_str):
+        if input_str.startswith("http://") or input_str.startswith("https://"):
+            return url_to_img(input_str)
+        else:
+            return b64_to_img(input_str)
+
+    img1 = load_any(img1_text)
+    img2 = load_any(img2_text)
 
     if img1 is None or img2 is None:
-        return "Upload both images."
-
-    # Convert to OpenCV
-    img1_cv = cv2.cvtColor(np.array(img1), cv2.COLOR_RGB2BGR)
-    img2_cv = cv2.cvtColor(np.array(img2), cv2.COLOR_RGB2BGR)
+        return "Invalid image data or URL."
 
-    emb1 = get_embedding(img1_cv)
-    emb2 = get_embedding(img2_cv)
+    emb1 = get_embedding(img1)
+    emb2 = get_embedding(img2)
 
     if emb1 is None or emb2 is None:
         return "Face not detected."
@@ -118,7 +123,10 @@ def gradio_ui(img1, img2):
 
 gr.Interface(
     fn=gradio_ui,
-    inputs=[gr.Image(), gr.Image()],
+    inputs=[
+        gr.Textbox(label="Image 1 (base64 or URL)"),
+        gr.Textbox(label="Image 2 (base64 or URL)")
+    ],
     outputs="text",
-    title="Face Match API"
+    title="Face Match API (Text Input)"
 ).launch()