Update app.py

app.py

@@ -16,14 +16,22 @@ STATUS_MESSAGES = {
     301: "Invalid token! Get Premium Token using link in page"
 }
 
-backend = Client(os.getenv("BACKEND"))
+BACKEND = os.getenv("BACKEND")
+backend = Client(BACKEND)
+JS_FUNC = os.getenv("JS_FUNC")
 
 def base64_to_image(base64_str):
     return base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))
 
 def search_face(file, token=None):
+    try:
+        file_1 = handle_file(file)
+    except Exception as e:
+        gr.Info("Please upload an image file.")
+        return []
+
     result_text = backend.predict(
-            file=handle_file(file),
+            file=file_1,
             token=token, 
             api_name="/search_face"
     )
@@ -168,7 +176,7 @@ head = """
 </script>
 """
 
-output = gr.Gallery(label="Search may take a few minutes.)", columns=[3], object_fit="contain", height="480px", interactive=False)
+output = gr.Gallery(label="Search may take a few minutes.", columns=[3], object_fit="contain", height="480px", interactive=False)
 col2 = gr.Column(scale=2, visible=False)
 
 def init_ui():
@@ -196,17 +204,16 @@ def update_button(token):
 MARKDOWN0 = """
     # Free Face Search Online - ❤️Like above if this space helps
     #### [Learn more about our Reverse Face Search](https://faceonlive.com/face-search-online)
-    #### [Check out our Face Search API integration and Affiliate Program here.](https://portfolio.faceonlive.com/#face_search)
+    #### [Face Search API and Affiliate Program (50%).](https://portfolio.faceonlive.com/#face_search)
 """
 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**
+### [Why Deep Search with Premium Token?](https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb)  
+✅ **Search Social Media, Deep Web & Uncover Hidden Profiles**  
+✅ **Outperforming PimEyes + FaceCheck.ID Combined!**  
 """
-MARKDOWN3 = """
-<div align="right"><a href="https://faceonlive.pocketsflow.com/checkout?productId=677fe2b5aeced2814bc47dd1" target='_blank' style='font-size: 16px;'>Opt-Out From Search</div><br/>
+
+MARKDOWN3_2 = """
+<div align="right"><a href="https://faceonlive.pocketsflow.com/checkout?productId=677fe2b5aeced2814bc47dd1" target='_blank' style='font-size: 16px;'>Opt-Out From Search</a></div><br/>
 <div align="right"><a href="https://faceonlive.com/reverse-image-search" target='_blank' style='font-size: 16px;'>Reverse Image Search</div><br/>
 <div align="right"><a href="https://faceonlive.com/deepfake-detector" target='_blank' style='font-size: 16px;'>AI Generated Image & Deepfake Detector</div>
 """
@@ -223,16 +230,18 @@ with gr.Blocks(css=custom_css, head=head, delete_cache=(3600, 3600)) as demo:
                 token = gr.Textbox(placeholder="(Optional) Input Premium Token here.", label="Premium Token")
               with gr.Column():
                 md_premium1 = gr.Markdown(MARKDOWN2)
+            gr.HTML("<div id='limit'></div>")
             with gr.Row():
                 with gr.Column():
-                    premium_search_button = gr.Button("🚀 Unlock Deep Search Now!", elem_classes="button-gradient", link=PREMIUM_CHECKOUT)
+                    limit_button = gr.Button("🔍 Free Face Search")
+                    face_search_button = gr.Button("Face Search", visible=False, elem_id="submit_btn")
                 with gr.Column():
-                    face_search_button = gr.Button("🔍 Free Face Search")
+                    premium_search_button = gr.Button("🚀 Unlock Deep Search Now!", elem_classes="button-gradient", link=PREMIUM_CHECKOUT)
             with gr.Row():
                 with gr.Column():
                     gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, fn=search_face_examples, outputs=[output, col1, col2])
                 with gr.Column():
-                    gr.HTML(MARKDOWN3)            
+                    gr.HTML(MARKDOWN3_2)
             
         with col2.render():
             gr.Markdown("> ### **⚠️ Reminder:** Export images before refreshing the page by clicking the 'Export Images' button.")
@@ -243,26 +252,25 @@ with gr.Blocks(css=custom_css, head=head, delete_cache=(3600, 3600)) as demo:
                 with gr.Column():
                     export_button = gr.Button("Export Images")
                     export_file = gr.File(label="Download", elem_id="export_file")
-
             with gr.Row():
                 with gr.Column():
                     premium_link_button = gr.Button("🚀 Unlock Deep Search Now!", elem_classes="button-gradient", link=PREMIUM_CHECKOUT)
                 with gr.Column():
                     new_search_button = gr.Button("🔍 New Search")
-            gr.HTML(MARKDOWN3)
+            gr.HTML(MARKDOWN3_2)
 
+    limit_button.click(None, js=JS_FUNC)
     face_search_button.click(search_ui, inputs=[], outputs=[col1, col2], api_name=False)
     face_search_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)
-    token.change(update_button, inputs=[token], outputs=[premium_search_button, face_search_button])
+    token.change(update_button, inputs=[token], outputs=[premium_search_button, limit_button])
 
     with gr.Row():
         with gr.Column(scale=1):
             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>')
         with gr.Column(scale=5):
             html = gr.HTML()
-
     demo.load(None, inputs=None, outputs=html, js=js)
     demo.load(set_url_token, inputs=None, outputs=[token])