Update app.py

app.py

@@ -28,26 +28,32 @@ def verify_image(url1):
     path = "Faces/"
     files = storage.bucket.list_blobs(prefix=path)
     flag = False
-    # url1 = "https://api.time.com/wp-content/uploads/2023/04/shah-rukh-khan-time100-2023-1.jpg"
+    username = "Not Found"
+    
     for file in files:
-      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"}
-
-
-
+        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"}