Update app.py

app.py

@@ -4,18 +4,43 @@ import requests
 import base64
 from io import BytesIO
 from PIL import Image
+import hashlib
 
-count = 0
 def image_to_base64(image):
     buffered = BytesIO()
     image.save(buffered, format="JPEG", quality=90)
     return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
 def base64_to_image(base64_str):
-    return Image.open(BytesIO(base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))))
+    return base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))
+
+def check_db(img_array, suffix, token_server_url):
+    hashes = []
+    out_array = []
+    for item in img_array:
+        try:
+            image_data = base64_to_image(item["image"])
+            
+            hash_object = hashlib.sha256(image_data)
+            image_hash = hash_object.hexdigest()
+            
+            hashes.append(image_hash)
+            out_array.append((Image.open(BytesIO(image_data)), item["url"] + suffix))
+    
+        except base64.binascii.Error as e:
+            raise ValueError(f"Invalid base64 string: {str(e)}")
+        except Exception as e:
+            raise ValueError(f"Error processing image: {str(e)}")
+
+    try:
+        r = requests.post(url=token_server_url + '/lookup-hashes', json={"hashes": hashes})
+        out_array = [value for i, value in enumerate(out_array) if i not in r.json().get('res')]
+    except:
+        raise gr.Error("Token Server Error!")
+
+    return out_array
 
 def search_face(file, token):
-    global count    
     free_url = os.environ.get("SERVER_URL_FREE")
     premium_url = os.environ.get("SERVER_URL_PREMIUM")
     token_server_url = os.environ.get("TOKEN_URL")
@@ -54,32 +79,28 @@ def search_face(file, token):
     elif status_code == 402:
         gr.Info("Wrong request.")
     elif status_code == 403:
-        gr.Info("Requests all used in your token.")
+        gr.Info("Please try again later.")
     elif status_code == 404:
         gr.Info("Timeout, try again.")
 
     if status_code > 300:
-        return [], count   
+        return []   
 
     try:        
         res = r.json().get('img_array')
-        out_array = []
         suffix = "*********"
         if url == premium_url:
             suffix = ""
-        for item in res:
-            out_array.append((base64_to_image(item["image"]), item["url"] + suffix))
-        count += 1
+        
+        out_array = check_db(res, suffix, token_server_url)
 
         if url == premium_url:
             try:
                 r = requests.post(url=token_server_url + '/activate-token', json={"token": token})
-                #if r.json().get('status') == 'success':
-                #    url = premium_url
             except:
                 raise gr.Error("Invalid token!")
 
-        return out_array, count
+        return out_array
     except:
         raise gr.Error("Try again.")
 
@@ -106,14 +127,14 @@ with gr.Blocks(css=custom_css) as demo:
             image = gr.Image(type='filepath', height=480)
             token = gr.Textbox(placeholder="(Optional) Premium Token via the link below.", label="Premium Token")
             gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb' target='_blank'>Get Premium Token: Perform Deep Search & Full URLs</a>")            
+            gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb' target='_blank'>Send DMCA & GDPR Takedown Notices On Your Behalf, Opt-Out From Search</a>")
             search_face_button = gr.Button("Search Face")
         with gr.Column(scale=2):
             output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto")
-            countwg = gr.Number(label="Count")
     
-    gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, cache_mode='lazy', fn=search_face_free, outputs=[output, countwg])
+    gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, cache_mode='lazy', fn=search_face_free, outputs=[output])
 
-    search_face_button.click(search_face, inputs=[image, token], outputs=[output, countwg], api_name=False)
+    search_face_button.click(search_face, inputs=[image, token], outputs=[output], api_name=False)
 
     gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online&labelColor=%23ff8a65&countColor=%2337d67a&style=flat&labelStyle=upper" /></a>')