Update app.py

app.py

@@ -271,26 +271,26 @@ 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")
-    filename = f"chat_history_{timestamp}.json"
+    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 None  
+    return "Chat saved!"  # Return a message to show the status on the UI
 
 def ask(user_message, chat_history):
     if not user_message:
         return chat_history, chat_history, ""
 
-    response = chatbot(user_message)
+    response = chatbot(user_message)  # Your chatbot function
     chat_history.append((user_message, response))
 
-    # Save chat history to file
-    with open(HISTORY_FILE, "w", encoding="utf-8") as f:
-        json.dump(chat_history, f, ensure_ascii=False, indent=2)
-
     return chat_history, chat_history, ""
 
 initial_message = (None, "Hello, how can I help you with Moodle?")
@@ -301,11 +301,12 @@ 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")  # to save the chat 
+    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
 
     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], None)  # to save the chat 
+    save_button.click(save_chat_to_file, [chat_history], save_status)  # Save the chat and show status message
 
 demo.queue()
 demo.launch(share=False)