Update app.py

app.py

@@ -1,142 +1,47 @@
 import os
 import gradio as gr
-import requests
-import base64
 from io import BytesIO
-from PIL import Image
-import hashlib
-import concurrent.futures
 import zipfile
 import tempfile
+import random  
+import json
+from gradio_client import Client, handle_file
+import base64
+from PIL import Image
 
-# Constants
-FREE_URL = os.environ.get("SERVER_URL_FREE")
-PREMIUM_URL = os.environ.get("SERVER_URL_PREMIUM")
-PREMIUM_URL2 = os.environ.get("SERVER_URL_PREMIUM2")
-TOKEN_SERVER_URL = os.environ.get("TOKEN_URL")
-API_KEY = os.environ.get("API_KEY")
+BACKEND = os.getenv("BACKEND")
+TOKEN = os.getenv("TOKEN")
+BUY_PREMIUM = "🥇 Get Premium Token to Unlock"
 
 STATUS_MESSAGES = {
-    301: "Too many faces in the photo.",
-    302: "Face is not clear enough.",
-    303: "No matches found. Try Deep Search with premium token.",
-    304: "No face in the photo.",
-    305: "Search blocked for privacy issue.",
-    401: "Invalid image format.",
-    402: "Wrong request.",
-    403: "Please try again later.",
-    404: "Timeout, try again."
+    201: "Not satisfied with the results? Get a Premium Token for instant, deeper results including social profiles.", 
+    301: "Invalid token! Get Premium Token using link in page"
 }
-FREE_SUFFIX = "*********"
 
-def image_to_base64(image):
-    buffered = BytesIO()
-    image.save(buffered, format="JPEG", quality=90)
-    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+backend = Client(BACKEND, hf_token=TOKEN)
 
 def base64_to_image(base64_str):
     return base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))
 
-def check_db(img_array, suffix):
-    hashes = []
-    out_array = []
-    for item in img_array:
-        try:
-            image_data = base64_to_image(item["image"])
-            image_hash = hashlib.sha256(image_data).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:
-        response = 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 response.json().get('res')]
-    except:
-        raise gr.Error("Token Server Error!")
-
-    return out_array
-
-def verify_token(token):
-    try:
-        response = requests.post(url=TOKEN_SERVER_URL + '/verify-token', json={"token": token})
-        if response.json().get('status') == 'success':
-            return PREMIUM_URL
-        else:
-            raise gr.Error("Invalid token! For free search, use empty string for token")
-    except:
-        raise gr.Error("Invalid token!")
-
-def activate_token(token):
-    try:
-        requests.post(url=TOKEN_SERVER_URL + '/activate-token', json={"token": token})
-    except:
-        raise gr.Error("Invalid token!")
-    
-def get_image_base64(file):
-    try:
-        image = Image.open(file)
-        return image_to_base64(image)
-    except Exception as e:
-        raise gr.Error("Please select image file!")    
-
-def send_requests(url, file):
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        future1 = executor.submit(requests.post, url, headers={"X-RapidAPI-Key": API_KEY}, json={"image": get_image_base64(file)})
-        if url == PREMIUM_URL:
-            future2 = executor.submit(requests.post, PREMIUM_URL2, files={"image": open(file, 'rb')})
-            response1 = future1.result()
-            response2 = future2.result()
-            return response1, response2
-        else:
-            response1 = future1.result()
-            return response1, None
-
-def process_response(response, soc_res, url, token):
-    status_code = response.status_code
-
-    if not soc_res:
-        if status_code in STATUS_MESSAGES:
-            gr.Info(STATUS_MESSAGES[status_code])
-
-        if status_code > 300:
-            return []
-
-    try:
-        res = response.json().get('img_array', []) if status_code in [201, 202] else []
-        if soc_res:
-            res = soc_res[0] + res + soc_res[1]
-        
-        suffix = "" if url == PREMIUM_URL else FREE_SUFFIX
-        out_array = check_db(res, suffix)
-
-        if url == PREMIUM_URL:
-            activate_token(token)
-
-        return out_array
-    except:
-        raise gr.Error("Try again.")
-
-def process_response2(response):
-    if response.status_code == 200:
-        part1 = response.json().get('part1')
-        part2 = response.json().get('part2')
-
-        if not part1 and not part2:
-            return None
-        return (part1, part2)
-    return None
-
 def search_face(file, token=None):
-    url = FREE_URL
-    if token:
-        url = verify_token(token)
-
-    response1, response2 = send_requests(url, file)
-    soc_res = process_response2(response2) if response2 else None
-    return process_response(response1, soc_res, url, token)
+    result_text = backend.predict(
+            file=handle_file(file),
+            token=token, 
+            api_name="/search_face"
+    )
+   
+    result = json.loads(result_text)
+    outarray = []
+    if result['status'] > 300:
+        raise gr.Error(STATUS_MESSAGES[result['status']])
+    
+    for item in result['result']:
+        image = Image.open(BytesIO(base64_to_image(item['image'])))
+        outarray.append((image, item['url']))
+    
+    if result['status'] == 201:
+        gr.Info(STATUS_MESSAGES[result['status']], duration=8)
+    return outarray
 
 def export_images(items):
     if not items:
@@ -145,10 +50,14 @@ def export_images(items):
     zip_buffer = BytesIO()
     with zipfile.ZipFile(zip_buffer, 'w') as zip_file:
         url_text = ""
-        for i, item in enumerate(items):
+        i = 1
+        for item in items:
+            if item[1] == BUY_PREMIUM:
+                continue
             with open(item[0], 'rb') as img_file:
                 zip_file.writestr(f'image_{i}.jpg', img_file.read())
             url_text += f"image_{i}.jpg: {item[1]}\n"
+            i += 1
         zip_file.writestr("urls.txt", url_text)
     zip_buffer.seek(0)
 
@@ -164,10 +73,14 @@ caption.caption {
     cursor: text;
 }
 
-div#component-16 {
+div#component-21 {
     max-height: 63.39px;
 }
 
+div#component-9 {
+    background-color: tomato;
+}
+
 .svelte-snayfm {
     height: auto;
 }
@@ -197,7 +110,6 @@ links.forEach(link => {
             }
         });
         link.setAttribute('href', targetUrl.toString());
-        console.log(`Updated Link: ${targetUrl.toString()}`);
     }
 });
 
@@ -217,7 +129,7 @@ head = """
 </script>
 """
 
-output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto", interactive=False)
+output = gr.Gallery(label="Found Images (The search may take a couple of minutes.)", columns=[3], object_fit="contain", height="480px", interactive=False)
 col2 = gr.Column(scale=2, visible=False)
 
 def init_ui():
@@ -229,23 +141,39 @@ def search_ui():
 def search_face_examples(image):
     return search_face(image), gr.update(visible=False), gr.update(visible=True)
 
-with gr.Blocks(css=custom_css, head=head, delete_cache=(3600, 3600)) as demo:
-    gr.Markdown(
-        """
-    # Free Face Search Online - If this space was helpful, please like ❤️ above!
-    #### [Discover more about our Face Search on our website.](https://faceonlive.com/face-search-online)
+MARKDOWN0 = """
+    # Free Face Search Online
     #### [Check out our Face Search API integration and affiliate program here.](https://portfolio.faceonlive.com/#face_search)
-    <br>
-    """
-    )
+"""
+MARKDOWN1 = """
+    ### [Get Premium Token - Deep Search & Social Media Access](https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb)  
+    ### [Protect Your Privacy – Start Your Takedown Now](https://faceonlive.pocketsflow.com/checkout?productId=676d7e7597f8b3b699f84820)
+"""
+MARKDOWN2 = """
+### [Why Get the Premium Token?](https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb)  
+✅ **Deep Search**  
+✅ **Social Media and Deep Web Scan**  
+✅ **Exclusive Access to Hidden Profiles**  
+✅ **No Subscription Needed**
+"""
+MARKDOWN3 = """
+<div align="right"><a href="https://faceonlive.com/reverse-image-search" target='_blank' style='font-size: 16px;'>Looking For Reverse Image Search</div>
+"""
+with gr.Blocks(css=custom_css, head=head, delete_cache=(3600, 3600)) as demo:
+    gr.Markdown(MARKDOWN0)
     with gr.Row():
         with gr.Column(scale=1) as col1:
-            image = gr.Image(type='filepath', height=480)
-            token = gr.Textbox(placeholder="(Optional) Get Premium Token Below.", label="Premium Token")
-            gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb' target='_blank' style='font-size: 16px;'>Get Premium Token: Deep Search Including Social Media & Full URL Reveal</a>")
-            gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676d7e7597f8b3b699f84820' target='_blank' style='font-size: 16px;'>Protect Your Privacy – Start Your Takedown Now</a>")
-            search_face_button = gr.Button("Search Face")
-            gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, cache_mode='eager', fn=search_face_examples, outputs=[output, col1, col2])
+            image = gr.Image(type='filepath', height=360)
+            with gr.Row():
+              with gr.Column():
+                token = gr.Textbox(placeholder="(Optional) Get Premium Token Below.", label="Premium Token")
+                gr.Markdown(MARKDOWN1)
+              with gr.Column():
+                md_premium1 = gr.Markdown(MARKDOWN2)
+            with gr.Row():
+                search_face_button = gr.Button("Reverse Face Search")
+            gr.HTML(MARKDOWN3)
+            gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, fn=search_face_examples, outputs=[output, col1, col2])
             
         with col2.render():
             gr.Markdown("> ### **⚠️ Reminder:** Export images before refreshing the page by clicking the 'Export Images' button.")
@@ -253,19 +181,21 @@ with gr.Blocks(css=custom_css, head=head, delete_cache=(3600, 3600)) as demo:
             with gr.Row():
                 export_button = gr.Button("Export Images")
                 export_file = gr.File(label="Download")
-            gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb' target='_blank' style='font-size: 16px;'>Get Premium Token: Deep Search Including Social Media & Full URL Reveal</a>")            
-            gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676d7e7597f8b3b699f84820' target='_blank' style='font-size: 16px;'>Protect Your Privacy – Start Your Takedown Now</a>")
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown(MARKDOWN1)
+                with gr.Column():
+                    md_premium2 = gr.Markdown(MARKDOWN2)
             new_search_button = gr.Button("🚀 Try New Search")
-        
+            gr.HTML(MARKDOWN3)
 
     search_face_button.click(search_ui, inputs=[], outputs=[col1, col2], api_name=False)
     search_face_button.click(search_face, inputs=[image, token], outputs=[output], api_name=False)
     export_button.click(export_images, inputs=[output], outputs=export_file, api_name=False)
-
     new_search_button.click(init_ui, inputs=[], outputs=[col1, col2], 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/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online&labelColor=%23ff8a65&countColor=%2337d67a&style=flat&labelStyle=upper" /></a>')
     html = gr.HTML()
     demo.load(None, inputs=None, outputs=html, js=js)
 
-demo.queue(api_open=False, default_concurrency_limit=8).launch(server_name="0.0.0.0", server_port=7860, show_api=False)
+demo.queue(api_open=False, default_concurrency_limit=8).launch(show_api=False)