Update app.py

app.py

@@ -58,17 +58,18 @@ def bot_respond(chat_history, openai_gpt_key, model_choice):
         time.sleep(0.02)
         yield chat_history
 
-def save_chat_history(chat_history):
+def save_chat_history():
     global messages
-    messages = []
-    # You could format this as you like. This will just create a text representation.
-    formatted_chat = "\n".join([f"User: {message[0]}\nBot: {message[1]}" for message in chat_history])
+    # Reset messages after saving the history
+    formatted_chat = "\n".join([f"{message['role']}: {message['content']}" for message in messages])
     
     # Use a timestamp for a unique filename for each conversation
     timestamp = time.strftime("%Y%m%d-%H%M%S")
     with open(f'chat_history_{timestamp}.txt', 'w') as f:
         f.write(formatted_chat)
 
+    # Clear the messages list for a new conversation
+    messages = []
 
 
         
@@ -105,7 +106,7 @@ with gr.Blocks() as demo:
 
     # Clear button click event
     clear_btn.click(
-    lambda: save_chat_history(chat_history.get_value()), 
+    lambda: save_chat_history(), 
     None, 
     chat_history, 
     queue=False