Update app.py

app.py

@@ -1,11 +1,12 @@
 import re
-import os
 from functools import lru_cache
-
+from urllib.parse import urlparse, parse_qs
 import gradio as gr
 import requests
+import sys
 
-
+TOKEN_ERR=True
+TOKEN_VALUE=os.getenv("TOKEN_VALUE")
 
 LOGIN = os.getenv("LOGIN")
 TOKEN = os.getenv("TOKEN")
@@ -232,6 +233,28 @@ 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,
@@ -270,5 +293,6 @@ with gr.Blocks(css=css, title=title) as demo:
     output = gr.outputs.HTML()
     input = [input_1, input_2, input_3, input_4, input_5, input_6]
     greet_btn.click(fn=generate, inputs=input, outputs=output)
+    demo.load(generate_init, inputs=None, outputs=output)
 
 demo.launch(debug=True, share=False)