Update app.py

app.py

@@ -32,7 +32,7 @@ def authorize(user, api_key):
             
             if api_key not in sessions:
                 sessions[api_key] = {
-                    "history": [],  
+                    "history": [], 
                     "headers": {   
                         "authorization": api_key,
                         "Content-Type": 'application/json'
@@ -94,7 +94,7 @@ 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"] = history 
+        sessions[api_key]["history"] = history  
         
         return history, assistant_reply
     else:
@@ -160,6 +160,16 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
         def regenerate_response(history, last_message, system_message, max_tokens, top_p, temperature):
             return "", []
 
+        def clear_history(api_key):
+            if api_key in sessions:
+                sessions[api_key]["history"] = []  
+            return "", []
+
+        def load_conversation(api_key):
+            session = sessions.get(api_key, {})
+            history = session.get("history", [])
+            return render_message(history), history
+
         msg_input.submit(user_interaction, 
                          inputs=[msg_input, api_key_input, system_message, max_tokens, top_p, temperature], 
                          outputs=[chatbot_output, history_state, msg_input, last_message_state])
@@ -168,18 +178,19 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
                        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, 
-                        inputs=[history_state, last_message_state, system_message, max_tokens, top_p, temperature], 
+        regen_btn.click(clear_history, 
+                        inputs=[api_key_input], 
                         outputs=[chatbot_output, history_state])
 
     def authorize_and_proceed(user, api_key):
         if authorize(user, api_key):
-            return gr.update(visible=False), gr.update(visible=True)
+            messages_html, history = load_conversation(api_key)
+            return gr.update(visible=False), gr.update(visible=True), messages_html, history
         else:
             return gr.update(visible=True), gr.update(visible=False), auth_status.update(value="Invalid userid/token")
 
 
-    auth_button.click(authorize_and_proceed, inputs=[api_user_input, api_key_input], outputs=[auth_view, chat_view])
+    auth_button.click(authorize_and_proceed, inputs=[api_user_input, api_key_input], outputs=[auth_view, chat_view, chatbot_output, history_state])
 
 if __name__ == "__main__":
     demo.queue = False