add chat history

app.py

@@ -2,7 +2,6 @@ import chainlit as cl
 import pandas as pd
 import time
 from typing import Dict, Any
-from chainlit.session import get_session
 import os
 import json
 from datetime import datetime
@@ -22,15 +21,15 @@ HISTORY_DIR = "chat_history"
 os.makedirs(HISTORY_DIR, exist_ok=True)
 
 # Function to get or create a user ID
-def get_user_id(session):
+def get_user_id():
     # Check if user already has an ID
-    user_id = session.user_session.get("user_id")
+    user_id = cl.user_session.get("user_id")
     
     if not user_id:
         # Generate a new user ID if none exists
         # In a real app, you might use authentication or cookies
         user_id = str(uuid.uuid4())
-        session.user_session.set("user_id", user_id)
+        cl.user_session.set("user_id", user_id)
         
         # Create a user directory for this user
         user_dir = os.path.join(HISTORY_DIR, user_id)
@@ -98,15 +97,12 @@ def load_chat_history(user_id, conversation_id=None):
 
 @cl.on_chat_start
 async def start():
-    # Get the current session
-    session = get_session()
-    
     # Get or create user ID
-    user_id = get_user_id(session)
+    user_id = get_user_id()
     
     # Create a new conversation ID for this session
     conversation_id = str(uuid.uuid4())
-    session.user_session.set("conversation_id", conversation_id)
+    cl.user_session.set("conversation_id", conversation_id)
     
     # Load previous conversations for this user
     conversations = load_chat_history(user_id)
@@ -130,7 +126,7 @@ async def start():
         if res and res.get("value") != "new":
             # User selected a previous conversation
             selected_conv_id = res.get("value")
-            session.user_session.set("conversation_id", selected_conv_id)
+            cl.user_session.set("conversation_id", selected_conv_id)
             
             # Load the selected conversation
             conversation = load_chat_history(user_id, selected_conv_id)
@@ -151,10 +147,9 @@ async def start():
 
 @cl.on_message
 async def on_message(message: cl.Message):
-    # Get the current session
-    session = get_session()
-    user_id = session.user_session.get("user_id")
-    conversation_id = session.user_session.get("conversation_id")
+    # Get user and conversation IDs
+    user_id = cl.user_session.get("user_id")
+    conversation_id = cl.user_session.get("conversation_id")
     
     # Process the message and generate a response
     # Replace this with your actual processing logic
@@ -168,9 +163,8 @@ async def on_message(message: cl.Message):
 
 @cl.action_callback("clear_history")
 async def clear_history_callback(action):
-    session = get_session()
-    user_id = session.user_session.get("user_id")
-    conversation_id = session.user_session.get("conversation_id")
+    user_id = cl.user_session.get("user_id")
+    conversation_id = cl.user_session.get("conversation_id")
     
     # Clear specific conversation or all conversations
     user_dir = os.path.join(HISTORY_DIR, user_id)
@@ -182,7 +176,7 @@ async def clear_history_callback(action):
     
     # Create a new conversation ID
     new_conversation_id = str(uuid.uuid4())
-    session.user_session.set("conversation_id", new_conversation_id)
+    cl.user_session.set("conversation_id", new_conversation_id)
 
 @cl.on_message
 async def on_message(message: cl.Message):