Update app.py

app.py

@@ -1,8 +1,6 @@
 import requests
 import pyrebase
 import urllib
-from PIL import Image
-from io import BytesIO
 from retinaface import RetinaFace
 from deepface import DeepFace
 from fastapi import FastAPI
@@ -30,33 +28,26 @@ def verify_image(url1):
     path = "Faces/"
     files = storage.bucket.list_blobs(prefix=path)
     flag = False
-    username = "Not Found"
-    
+    # url1 = "https://api.time.com/wp-content/uploads/2023/04/shah-rukh-khan-time100-2023-1.jpg"
     for file in files:
-        if file.name.endswith((".jpg", ".jpeg")):
-            url = storage.child(file.name).get_url(None)
-            try:
-                # Retrieve the image from URL
-                response = requests.get(url)
-                response.raise_for_status()  # Raise an exception for HTTP errors
-                # Open the image using PIL
-                img = Image.open(BytesIO(response.content))
-
-                # Verify the image
-                result = DeepFace.verify(url1, url, model_name="Facenet", distance_metric='cosine')
-                if result['verified']:
-                    flag = True
-                    # Extract username from the file name
-                    start_index = file.name.rfind('/')
-                    end_index = file.name.rfind('$')
-                    if start_index != -1 and end_index != -1:
-                        username = file.name[start_index + 1:end_index]
-                    break  # No need to continue loop if verified
-            except Exception as e:
-                print(f"Error processing image: {e}")
-    
-    if flag:
-        return {"username": username}
-    else:
-        print("Not Verified")
-        return {"username": "Not Found"}
+      if (file.name.endswith(".jpg") or file.name.endswith(".jpeg")) :
+        # print(file.name)
+        url = storage.child(f"{file.name}").get_url(None)
+        # print(url)
+        with requests.get(url) as response:
+          result = DeepFace.verify(f"{url1}",url, model_name="Facenet", distance_metric='cosine')
+          if result['verified']:
+            flag = True
+            start_index = file.name.rfind('/')
+            end_index = file.name.rfind('$')
+            if start_index != -1 and end_index != -1:
+              name = file.name[start_index + 1:end_index]
+              return {"username": name}
+
+
+    if flag == False:
+      print("Not Verified")
+      return {"username": "Not Found"}
+
+
+