Update app.py

app.py

@@ -25,6 +25,15 @@ def generate_embedding(image):
     return embedding_model.predict(preprocessed_image)[0]
 
 # Register new user
+def register_user(image, user_id):
+    try:
+        embedding = generate_embedding(image)
+        user_embeddings[user_id] = embedding
+        return f"User {user_id} registered successfully."
+    except Exception as e:
+        return f"Error during registration: {str(e)}"
+
+# Recognize user
 def recognize_user(image):
     new_embedding = generate_embedding(image)
     min_distance = float('inf')
@@ -40,21 +49,7 @@ def recognize_user(image):
         return "User not recognized."
     else:
         return f"Recognized User: {recognized_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
-
-    return f"Recognized User: {recognized_user_id}"
-
+        
 # Gradio interface for registering users
 register_interface = gr.Interface(
     fn=register_user,