Add sessions, remove session when closed, refreshed tab

app.py

@@ -7,13 +7,15 @@ API_URL = "https://host.palple.polrambora.com/pmsq"
 API_TOKEN = os.getenv("POLLY") 
 URL_HEADER = os.getenv("KEY")
 headers = {
-    "authorization":"",
+    "authorization": "",
     "Content-Type": 'application/json',
 }
 
 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,
@@ -25,20 +27,19 @@ def authorize(user, api_key):
 
     print("Preparing to send the request...") 
     try:
-        print(f"Request Data: {test_data}")
-        print(f"Request Headers: {test_headers}")
-        print(f'{json.dumps(test_data)}')
         response = requests.post(
             "https://host.palple.polrambora.com/check_key_impv",
             json=test_data,
         )
 
-        print("Request sent!") 
-
         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] = []
+                
             return True
         else:
             print(f"Failed request - Status code: {response.status_code}, Response: {response.text}")
@@ -54,7 +55,9 @@ def authorize(user, api_key):
         return False
 
 
-def respond(message, history, system_message, max_tokens, top_p, temperature):
+def respond(message, api_key, system_message, max_tokens, top_p, temperature):
+    history = sessions.get(api_key, [])
+
     messages = []
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
         if user_message:
@@ -82,28 +85,19 @@ def respond(message, history, system_message, max_tokens, top_p, temperature):
         "conversation_history": messages,
         "input": message
     }
-    print(headers)
+    
     response = requests.post(API_URL, headers=headers, data=json.dumps(data))
     
     if response.status_code == 200:
         response_json = response.json()
         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  
         return history, assistant_reply
     else:
         return history, "Error: " + response.json().get("error", "Unknown error occurred.")
 
-def load_conversation_from_file(json_file):
-    with open(json_file, 'r') as file:
-        data = json.load(file)
-        history = []
-        for msg in data['conversation']:
-            user_message = msg.get('user_message', "")
-            assistant_message = msg.get('assistant_message', "")
-            history.append((user_message, assistant_message, "You", "P-ALPLE", USER_PIC_PATH, ASSISTANT_PIC_PATH))
-        return history
-
-# Message rendering
 def render_message(history):
     messages_html = ""
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
@@ -157,28 +151,19 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
         history_state = gr.State([])  
         last_message_state = gr.State("")  
 
-        load_btn = gr.Button("Load Conversation")
-        json_file_input = gr.Textbox(placeholder="Path to .json file", label="JSON File Path")
-
-        def load_conversation(file_path):
-            history = load_conversation_from_file(file_path)
-            return render_message(history), history
-
-        load_btn.click(load_conversation, inputs=json_file_input, outputs=[chatbot_output, history_state])
-
-        def user_interaction(message, history, system_message, max_tokens, top_p, temperature):
-            history, assistant_reply = respond(message, history, system_message, max_tokens, top_p, temperature)
+        def user_interaction(message, api_key, system_message, max_tokens, top_p, temperature):
+            history, assistant_reply = respond(message, api_key, system_message, max_tokens, top_p, temperature)
             return render_message(history), history, "", message  
 
         def regenerate_response(history, last_message, system_message, max_tokens, top_p, temperature):
             return "", []
 
         msg_input.submit(user_interaction, 
-                         inputs=[msg_input, history_state, system_message, max_tokens, top_p, temperature], 
+                         inputs=[msg_input, api_key_input, system_message, max_tokens, top_p, temperature], 
                          outputs=[chatbot_output, history_state, msg_input, last_message_state])
 
         send_btn.click(user_interaction, 
-                       inputs=[msg_input, history_state, system_message, max_tokens, top_p, temperature], 
+                       inputs=[msg_input, api_key_input, system_message, max_tokens, top_p, temperature], 
                        outputs=[chatbot_output, history_state, msg_input, last_message_state])
 
         regen_btn.click(regenerate_response,