Update app.py

app.py

@@ -35,20 +35,25 @@ def register_user(image, user_id):
 
 # Recognize user
 def recognize_user(image):
-    new_embedding = generate_embedding(image)
-    min_distance = float('inf')
-    recognized_user_id = "Unknown"
-    
-    for user_id, embedding in user_embeddings.items():
-        distance = cosine(new_embedding, embedding)
-        if distance < min_distance:
-            min_distance = distance
-            recognized_user_id = user_id
-    
-    if min_distance > RECOGNITION_THRESHOLD:
-        return "User not recognized."
-    else:
-        return f"Recognized User: {recognized_user_id}"
+    try:
+        new_embedding = generate_embedding(image)
+        min_distance = float('inf')
+        recognized_user_id = "Unknown"
+        
+        for user_id, embedding in user_embeddings.items():
+            distance = cosine(new_embedding, embedding)
+            print(f"Distance for {user_id}: {distance}")  # Debug: Print distances
+            if distance < min_distance:
+                min_distance = distance
+                recognized_user_id = user_id
+        
+        print(f"Min distance: {min_distance}")  # Debug: Print minimum distance
+        if min_distance > RECOGNITION_THRESHOLD:
+            return "User not recognized."
+        else:
+            return f"Recognized User: {recognized_user_id}"
+    except Exception as e:
+        return f"Error during recognition: {str(e)}"
         
 # Gradio interface for registering users
 register_interface = gr.Interface(
@@ -67,5 +72,5 @@ recognize_interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    register_interface.launch(share=True)
+    #register_interface.launch(share=True)
     recognize_interface.launch(share=True)