Update FastAPI.py

FastAPI.py

@@ -37,8 +37,13 @@ class ImgInput(BaseModel):
     image_url: HttpUrl
 
 class ImgOutput(BaseModel):
+    label: list
+
+
+class UserSaved(BaseModel):
     label: str
 
+
 def recognize_face(image_url: HttpUrl) -> ImgOutput:
 
     storage.child().download("Faces/pkl/face_encodings.pkl","face_encodings.pkl")
@@ -53,28 +58,38 @@ def recognize_face(image_url: HttpUrl) -> ImgOutput:
         face_encodings = data["encodings"]
         labels = data["labels"]
 
-    # Load a new image you want to recognize
-    new_image = cv2.imread("examp.jpg")
 
-    new_face_encoding = face_recognition.face_encodings(new_image)
+        # Load the new image you want to recognize
+        new_image = cv2.imread("examp.jpg")
+
+        # Find face encodings in the new image
+        new_face_encodings = face_recognition.face_encodings(new_image)
+
+        if len(new_face_encodings) == 0:
+            print("No faces found in the new image.")
+            return ImgOutput(label=["unable to detect"])
+        else:
+            output_labels = []
 
-    if len(new_face_encoding) == 0:
-        print("No faces found in the new image.")
-    else:
-        # Compare the new face encoding to the stored encodings
-        results = face_recognition.compare_faces(face_encodings, new_face_encoding[0])
+            for new_face_encoding in new_face_encodings:
+                # Compare the new face encoding to the stored encodings
+                results = face_recognition.compare_faces(face_encodings, new_face_encoding)
 
-        os.remove("examp.jpg")
+                for i, result in enumerate(results):
+                    if result:
+                        output_labels.append(labels[i])
 
-        for i, result in enumerate(results):
-            if result:
-                return ImgOutput(label=labels[i])
+            os.remove("examp.jpg")
 
-    return ImgOutput(label="unable to detect")
+            if output_labels:
+                return ImgOutput(label=output_labels)
+            else:
+                out = ["unable to detect"]
+                return ImgOutput(label=out)
 
 
 
-def add_face(image_url: HttpUrl,user_name : str) -> ImgOutput:
+def add_face(image_url: HttpUrl,user_name : str):
     # Downloading image
     response = requests.get(image_url)
     with open("examp.jpg", 'wb') as file:
@@ -118,4 +133,5 @@ async def scoring_endpoint(item:ImgInput):
 @app.post('/user/')
 async def scoring_endpoint(item:ImgSave):
     add_face(item.image_url, item.user_name)
-    return ImgOutput(label="User Saved")
+    results = ["User Saved"]
+    return ImgOutput(label=results)