Update app.py

app.py

@@ -24,30 +24,31 @@ def search_face(file):
         r = requests.post(url=url, headers={"X-RapidAPI-Key": os.environ.get("API_KEY")}, json={"image": image_base64})
     except:
         raise gr.Error("Please select image file!")
-    
-    if r.status_code == 301:
-        gr.Info("Too many faces in the photo.")
-    elif r.status_code == 302:
-        gr.Info("Face is not clear enough.")
-    elif r.status_code == 303:
-        gr.Info("No matches found.")
-    elif r.status_code == 304:
-        gr.Info("No face in the photo.")
-    elif r.status_code == 305:
-        gr.Info("Search blocked for privacy issue.")
-    elif r.status_code == 401:
-        gr.Info("Invalid image format.")
-    elif r.status_code == 402:
-        gr.Info("Wrong request.")
-    elif r.status_code == 403:
-        gr.Info("Requests all used in your token.")
-    elif r.status_code == 404:
-        gr.Info("Timeout, try again.")
 
-    if r.status_code > 300:
-        return [], count
-    
     try:
+        status_code = r.json().get('AC')
+        if status_code == 301:
+            gr.Info("Too many faces in the photo.")
+        elif status_code == 302:
+            gr.Info("Face is not clear enough.")
+        elif status_code == 303:
+            gr.Info("No matches found.")
+        elif status_code == 304:
+            gr.Info("No face in the photo.")
+        elif status_code == 305:
+            gr.Info("Search blocked for privacy issue.")
+        elif status_code == 401:
+            gr.Info("Invalid image format.")
+        elif status_code == 402:
+            gr.Info("Wrong request.")
+        elif status_code == 403:
+            gr.Info("Requests all used in your token.")
+        elif status_code == 404:
+            gr.Info("Timeout, try again.")
+    
+        if status_code > 300:
+            return [], count   
+    
         res = r.json().get('img_array')
         out_array = []
         for item in res: