Update app.py

app.py

@@ -11,6 +11,10 @@ MAIN_JS = os.getenv("MAIN_JS")
 HEAD_HTML = os.getenv("HEAD_HTML")
 CSS = os.getenv("CSS")
 
+STATUS_MESSAGES = {
+    301: "Invalid Premium ID!"
+}
+
 backend = Client(BACKEND)
 
 def base64_to_image(base64_str):
@@ -18,7 +22,7 @@ def base64_to_image(base64_str):
         base64_str = base64_str.split(";base64,")[1].strip()
     return base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))
 
-def search_image(base64_image):
+def search_image(base64_image, token_txt):
     try:
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
             temp_file.write(base64_to_image(base64_image))
@@ -31,20 +35,28 @@ def search_image(base64_image):
 
     result_text = backend.predict(
             file=handle_file(file),
-            token="", 
+            token=token_txt, 
             api_name="/search_face"
     )
     os.remove(file)
+
+    result = json.loads(result_text)
+    outarray = []
+    if result['status'] > 300:
+        gr.Info(STATUS_MESSAGES[result['status']])
+        return '{"result": []}'
+
     return result_text
 
 with gr.Blocks(css=CSS, head=HEAD_HTML, title="DeepSeek? FaceSeek!") as iface:
     html = gr.HTML(MAIN_HTML, max_height=720)
-    js_txt = gr.Textbox(label="Base64-Image", elem_id="base64_image", visible=False)
+    base64_txt = gr.Textbox(label="Base64-Image", elem_id="base64_image", visible=False)
+    token_txt = gr.Textbox(label="Token-Text", elem_id="premium_token", visible=False)
     out_txt = gr.Textbox(label="Result", visible=False)
 
     search_button = gr.Button("🔍 Free Face Search", elem_id="search_btn", visible=False)
 
-    search_button.click(search_image, inputs=js_txt, outputs=out_txt).success(None, inputs=[out_txt], js=JS2)
+    search_button.click(search_image, inputs=[base64_txt, token_txt], outputs=out_txt).success(None, inputs=[out_txt], js=JS2)
     iface.load(None, inputs=None, outputs=html, js=MAIN_JS)
 
 # Launch the interface