Update session handling with unique session IDs

app.py

@@ -3,6 +3,7 @@ import chainlit as cl
 from agent_graph import agent_node
 from dotenv import load_dotenv  # Import dotenv
 from typing import List, Dict
+import time
 
 # Load environment variables from .env file
 load_dotenv()
@@ -13,13 +14,24 @@ openai_api_key = os.getenv("OPENAI_API_KEY")
 if not openai_api_key:
     raise ValueError("OpenAI API key is missing in the .env file")
 
-# Store chat history
+# Store chat history with unique session IDs
 chat_histories: Dict[str, List[Dict[str, str]]] = {}
 
+def get_unique_session_id():
+    """Generate a unique session ID combining user session ID and timestamp."""
+    user_id = cl.user_session.get("id", "unknown")
+    timestamp = int(time.time() * 1000)  # Current time in milliseconds
+    return f"{user_id}_{timestamp}"
+
 @cl.on_chat_start
 async def start_chat():
+    # Generate a unique session ID
+    session_id = get_unique_session_id()
+    # Store the session ID in user session for later use
+    cl.user_session.set("session_id", session_id)
+    
     # Initialize empty chat history for this session
-    chat_histories[cl.user_session.get("id")] = []
+    chat_histories[session_id] = []
     
     welcome_message = """👋 Welcome to the Stock Price Calculator!
 
@@ -38,8 +50,15 @@ What would you like to know?"""
 @cl.on_message
 async def handle_message(message: cl.Message):
     try:
+        # Get the unique session ID
+        session_id = cl.user_session.get("session_id")
+        if not session_id:
+            # If session ID is missing, generate a new one
+            session_id = get_unique_session_id()
+            cl.user_session.set("session_id", session_id)
+            chat_histories[session_id] = []
+        
         # Get chat history for this session
-        session_id = cl.user_session.get("id")
         history = chat_histories.get(session_id, [])
         
         # Add current message to history
@@ -72,6 +91,6 @@ async def handle_message(message: cl.Message):
 @cl.on_chat_end
 async def end_chat():
     # Clean up chat history when session ends
-    session_id = cl.user_session.get("id")
-    if session_id in chat_histories:
+    session_id = cl.user_session.get("session_id")
+    if session_id and session_id in chat_histories:
         del chat_histories[session_id]

push_to_hf.bat

@@ -0,0 +1,4 @@
+@echo off
+git add .
+git commit -m "Update session handling with unique session IDs"
+git push origin main