v1 of save_identitiy

app.py

@@ -1,12 +1,26 @@
+import gradio as gr
+
 import numpy as np
 from deepface import DeepFace
-import gradio as gr
+from pymongo.mongo_client import MongoClient
+
+credentials = "jamshaid:jamshaid19gh"
 
+uri = f"mongodb+srv://{credentials}@cluster0.uimyui3.mongodb.net/?retryWrites=true&w=majority"
+client = MongoClient(uri)
+db = client["Face_identification"]
+identities_collection = db["face_identities"]
 
 def save_identity(image , name):
   try:
     embeddings = DeepFace.represent(image , model_name="Facenet")
-    return str(embeddings)
+    embeddings = embeddings[0]
+
+    identity = {"embeddings":embeddings["embedding"] , "name" : name }
+
+    result = identities_collection.insert_one(identity)
+
+    return str(result)
   except Exception as error:
     return str(error)
 

requirements.txt

@@ -1,2 +1,3 @@
 numpy
-deepface
+deepface
+pymongo