Update app.py

app.py

@@ -3,14 +3,20 @@
 # file: app.py
 # ===========================================
 
-import json
-import asyncio
 import os
 import re
+import uuid
+import time
+import json
+import asyncio
 import requests
+from pathlib import Path
+from datetime import datetime
 from dotenv import load_dotenv
+
 import chainlit as cl
 from chainlit import user_session
+
 from langchain import hub
 from langchain_openai import OpenAI
 from langchain.chains import LLMChain
@@ -26,7 +32,6 @@ from langchain.memory.buffer import ConversationBufferMemory
 load_dotenv()
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 auth_token = os.environ.get("DAYSOFF_API_TOKEN")
-
 API_URL = "https://aivisions.no/data/daysoff/api/v1/booking/"
 
 #If booking information is requested, and with
@@ -80,14 +85,31 @@ async def set_starters():
             )
         ]
 
-@cl.on_chat_start
-#def setup_multiple_chains():
+def create_output_path(user=None):
+    this_uuid = str(uuid.uuid4())
+    ts = round(time.time())
+    dt_object = datetime.fromtimestamp(ts)
+    date = str(dt_object).replace(" ","_").replace(":","-")
+    
+    base_path = os.environ.get('HOME', '/home/user')
+    logs_path = Path(f"{base_path}/logs")
+    
+    path_out = logs_path / f"{date}--{user or 'default'}--{this_uuid}"
+    path_out.mkdir(exist_ok=True, parents=True)
+    
+    return path_out
+    
+@cl.on_chat_start # <————————————————— the Chainlit context
 def init_session():
     user_session.set('chat_history', [])
     print("chat_history: ", user_session.get('chat_history'))
     user_session.set('path_out', create_output_path())
     print("path_out: ", user_session.get('path_out'))
 
+    #cl.user_session.set("session_created_at", datetime.now())
+    #cl.user_session.set("connection_attempts", 0)
+    #cl.user_session.set("connection_attempts", connection_attempts + 1)
+
     llm = OpenAI(
         model="gpt-3.5-turbo-instruct",
         temperature=0.7,
@@ -111,7 +133,7 @@ def init_session():
     
 @cl.on_message
 async def handle_message(message: cl.Message):
-    
+    #connection_attempts = cl.user_session.get("connection_attempts", 0)
     user_message = message.content
     llm_chain = cl.user_session.get("llm_chain")
     memory = cl.user_session.get("memory")
@@ -145,7 +167,7 @@ async def handle_message(message: cl.Message):
                     await cl.Message(content=f"Error processing booking data: {str(e)}").send()
 
             else:
-                await cl.Message(content="Booking not found or invalid response.").send()
+                await cl.Message(content="Booking not found.").send()
 
         except requests.exceptions.RequestException as e:
             await cl.Message(content=f"Request failed: {str(e)}").send()