Update app.py

app.py

@@ -1,12 +1,9 @@
 import re
 from functools import lru_cache
-from urllib.parse import urlparse, parse_qs
 import gradio as gr
 import requests
-import os
 
 
-TOKEN_ERR=True
 TOKEN_VALUE=os.getenv("TOKEN_VALUE")
 LOGIN = os.getenv("LOGIN")
 TOKEN = os.getenv("TOKEN")
@@ -233,40 +230,20 @@ def check_record(
 
     return result
 
-#Проверяем перменную v в GET параметре на наличие правильного токена, без него форма не должна работать
-def generate_init(request: gr.Request):
-    global TOKEN_ERR
-    try:
-      headers =request.headers
-      url = headers.get('referer')
-      parsed_url = urlparse(url)
-      query_params = parse_qs(parsed_url.query)
-      my_dict = query_params
-
-    except KeyError as e:
-        print("Что то случилось в header", e)
-    try:
-      my_v = my_dict['v'][0]
-      if (my_v==TOKEN_VALUE): 
-          TOKEN_ERR = False
-          my_v = "Token accept"
-      else : my_v = "Token error"
-    except KeyError:
-      my_v = "Token error"
-      TOKEN_ERR = True
-    return my_v
+
 
 def generate(
         company_code, domain, company_linkedin,
-        email, phone, contact_linkedin):
-    if (TOKEN_ERR): return "Token error"
-    return check_record(
+        email, phone, contact_linkedin, token):
+    if  (TOKEN_VALUE == token):     return check_record(
         company_code=company_code,
         domain=domain,
         company_linkedin=company_linkedin,
         email=email,
         phone=phone,
         contact_linkedin=contact_linkedin)
+    else: return "Token error"
+
 
 
 title = "NetHunt search"
@@ -283,17 +260,23 @@ with gr.Blocks(css=css, title=title) as demo:
     )
     with gr.Row():
         with gr.Column():
+            input_7 = gr.Text(label="Token")
             input_1 = gr.Text(label="Сompany Code")
             input_2 = gr.Textbox(label="Corp Domain")
-            input_3 = gr.Textbox(label="Company Linkedin")
+            gr.Markdown("""
+
+            
+            """)
+            greet_btn = gr.Button("Search")
         with gr.Column():
             input_4 = gr.Textbox(label="Email")
             input_5 = gr.Textbox(label="Phone")
             input_6 = gr.Textbox(label="Contact LinkedIn")
-    greet_btn = gr.Button("Search")
+            input_3 = gr.Textbox(label="Company LinkedIn")
+            
     output = gr.outputs.HTML()
-    input = [input_1, input_2, input_3, input_4, input_5, input_6]
+    input = [input_1, input_2, input_3, input_4, input_5, input_6, input_7]
     greet_btn.click(fn=generate, inputs=input, outputs=output)
-    demo.load(generate_init, inputs=None, outputs=output)
+
 
 demo.launch(debug=True, share=False)