saving chat in file

app.py

@@ -271,24 +271,21 @@ def chatbot(query):
 #demo.launch(share=False)
 # Initialize chat history with a welcome message
 
-import gradio as gr
-import json
-import time
-
-# This function will save chat history to a file only when the button is pressed
 def save_chat_to_file(chat_history):
-    timestamp = time.strftime("%Y%m%d-%H%M%S")  # Create a timestamp
-    filename = f"chat_history_{timestamp}.json"  # Filename with timestamp
-    with open(filename, "w", encoding="utf-8") as f:
-        json.dump(chat_history, f, ensure_ascii=False, indent=2)
-    print(f"Chat history saved to {filename}")
-    return "Chat saved!"  # Return a message to show the status on the UI
+    timestamp = time.strftime("%Y%m%d-%H%M%S")
+    filename = f"chat_history_{timestamp}.json"
+
+    # Prepare the file content
+    file_content = json.dumps(chat_history, ensure_ascii=False, indent=2)
+
+    # Return the file as a download link
+    return (filename, file_content)
 
 def ask(user_message, chat_history):
     if not user_message:
         return chat_history, chat_history, ""
 
-    response = chatbot(user_message)  # Your chatbot function
+    response = chatbot(user_message)
     chat_history.append((user_message, response))
 
     return chat_history, chat_history, ""
@@ -301,12 +298,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     chatbot_ui = gr.Chatbot(value=[initial_message])
     question = gr.Textbox(placeholder="Ask me anything about Moodle...", show_label=False)
     clear_button = gr.Button("Clear")
-    save_button = gr.Button("Save Chat")  # Save the chat button
-    save_status = gr.Textbox(placeholder="Status will appear here", show_label=False, interactive=False)  # Display save status
+    save_button = gr.Button("Save Chat")  # to save the chat 
 
     question.submit(ask, [question, chat_history], [chatbot_ui, chat_history, question])
     clear_button.click(lambda: ([initial_message], [initial_message], ""), None, [chatbot_ui, chat_history, question], queue=False)
-    save_button.click(save_chat_to_file, [chat_history], save_status)  # Save the chat and show status message
+    
+    # Clicking the save button will trigger file download
+    save_button.click(save_chat_to_file, [chat_history], gr.File())  
 
 demo.queue()
 demo.launch(share=False)