Update app.py

app.py

@@ -4,18 +4,12 @@ import json
 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',
-}
+
+sessions = {}
 
 ASSISTANT_PIC_PATH = "https://huggingface.co/spaces/PLRMB/P-MSQ-API-PREVIEW/resolve/main/API.png"  
 USER_PIC_PATH = "https://huggingface.co/spaces/PLRMB/P-MSQ-API-PREVIEW/resolve/main/usr.png"
 
-sessions = {}
-
 def authorize(user, api_key):
     test_data = {
         "user": user,
@@ -34,11 +28,16 @@ def authorize(user, api_key):
 
         if response.status_code == 200:
             response_json = response.json()
-            headers['authorization'] = api_key
             print(f"Response: {response_json}")
             
             if api_key not in sessions:
-                sessions[api_key] = []
+                sessions[api_key] = {
+                    "history": [],  
+                    "headers": {   
+                        "authorization": api_key,
+                        "Content-Type": 'application/json'
+                    }
+                }
                 
             return True
         else:
@@ -56,7 +55,9 @@ def authorize(user, api_key):
 
 
 def respond(message, api_key, system_message, max_tokens, top_p, temperature):
-    history = sessions.get(api_key, [])
+    session = sessions.get(api_key, {})
+    history = session.get("history", [])
+    headers = session.get("headers", {})
 
     messages = []
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
@@ -93,7 +94,8 @@ def respond(message, api_key, system_message, max_tokens, top_p, temperature):
         assistant_reply = response_json["msq"]["message"][0]
         
         history.append((message, assistant_reply, "You", "P-ALPLE", USER_PIC_PATH, ASSISTANT_PIC_PATH))
-        sessions[api_key] = history  
+        sessions[api_key]["history"] = history 
+        
         return history, assistant_reply
     else:
         return history, "Error: " + response.json().get("error", "Unknown error occurred.")