Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 
 API_URL = "https://host.palple.polrambora.com/pmsq"
 API_TOKEN = os.getenv("POLLY") 
-
+URL_HEADER = os.getenv("KEY")
 headers = {
     "authorization":"",
     "Content-Type": 'application/json',
@@ -20,7 +20,7 @@ def authorize(api_key):
         "identificator": api_key
     }
     test_headers = {
-        "authorization": os.getenv("KEY"),  
+        "authorization": f"{URL_HEADER}",  
         "Content-Type": "application/json",
     }
 
@@ -29,7 +29,12 @@ def authorize(api_key):
         print(f"Request Data: {test_data}")
         print(f"Request Headers: {test_headers}")
 
-        response = requests.post("https://polrambora.com/api/v2/checkSDPA", headers=test_headers, data=json.dumps(test_data))
+        response = requests.post(
+            "https://polrambora.com/api/v2/checkSDPA",
+            headers=test_headers,
+            data=json.dumps(test_data),
+            timeout=10 
+        )
 
         print("Request sent!") 
 
@@ -40,10 +45,17 @@ def authorize(api_key):
         else:
             print(f"Failed request - Status code: {response.status_code}, Response: {response.text}")
             return False
+    except requests.exceptions.Timeout:
+        print("Request timed out")
+        return False
+    except requests.exceptions.ConnectionError:
+        print("Network problem occurred (DNS failure, refused connection, etc.)")
+        return False
     except Exception as e:
         print(f"Exception occurred: {str(e)}")
         return False
 
+
 def respond(message, history, system_message, max_tokens, top_p, temperature):
     messages = []
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history: