Update app.py

app.py

@@ -128,12 +128,21 @@ with gr.Blocks() as demo:
 
     chatbot = gr.Chatbot(height=600)
     msg = gr.Textbox(label="Message")
+    save_history = gr.Checkbox(label="Save Conversation History", value=True)
 
     def on_new_session():
         new_id = session_manager.create_session()
         return new_id, []
 
-    def user(message, history, session_id):
+    def user(message, history, session_id, save_history):
+        if save_history:
+            session = session_manager.load_session(session_id)
+            session["history"].append({
+                "timestamp": datetime.now().isoformat(),
+                "type": "user",
+                "content": message
+            })
+            session_manager.save_session(session_id, session)
         return "", history + [[message, None]]
 
     def bot(history, session_id):
@@ -143,7 +152,7 @@ with gr.Blocks() as demo:
                 history[-1][1] = response
                 yield history
 
-    msg.submit(user, [msg, chatbot, session_id], [msg, chatbot]).then(
+    msg.submit(user, [msg, chatbot, session_id, save_history], [msg, chatbot]).then(
         bot, [chatbot, session_id], [chatbot]
     )
     new_session.click(on_new_session, None, [session_id, chatbot])