Update app.py

app.py

@@ -1,8 +1,9 @@
-# ===========================================
-# title: daysoff-assistant-API-v2, 
+# ———————————————————————————————————————————————————————————————————————————>
+# Comment: various test-blocks to prevent session-outage with more recent UI;
 # chainlit==1.1.404
-# file: app.py
-# ===========================================
+# new imports: user_session, init_ws_context, chainlit.session/ WebsocketSession
+# ref--'Session is disconnected' rt7E1rI_uhficQm8AAAA ..
+# <———————————————————————————————————————————————————————————————————————————
 
 import os
 import re
@@ -18,7 +19,7 @@ from dotenv import load_dotenv
 import chainlit as cl
 from chainlit import user_session
 from chainlit.context import init_ws_context
-from chainlit.session import WebsocketSession # --'Session is disconnected' rt7E1rI_uhficQm8AAAA 
+from chainlit.session import WebsocketSession 
 from langchain import hub
 from langchain_openai import OpenAI
 from langchain.chains import LLMChain
@@ -32,51 +33,36 @@ from langchain_core.messages import BaseMessage
 from pydantic import BaseModel, Field
 from typing import List
 
-#class InMemoryHistory(BaseChatMessageHistory, BaseModel):
-    #messages: List[BaseMessage] = Field(default_factory=list)
+class InMemoryHistory(BaseChatMessageHistory, BaseModel):
+    messages: List[BaseMessage] = Field(default_factory=list)
 
-    #def add_messages(self, messages: List[BaseMessage]) -> None:
-        #self.messages.extend(messages)
+    def add_messages(self, messages: List[BaseMessage]) -> None:
+        self.messages.extend(messages)
 
-    #def clear(self) -> None:
-        #self.messages = []
+    def clear(self) -> None:
+        self.messages = []
 
-# --storing chat histories
-#chat_histories = {}
-
-#def get_session_history(session_id: str) -> BaseChatMessageHistory:
-    #if session_id not in chat_histories:
-        #chat_histories[session_id] = InMemoryHistory()
-    #return chat_histories[session_id]
-
-# ---------------------------------------------------for backend looks, example file:----------------------------------
-#with open('/home/user/.local/lib/python3.10/site-packages/socketio/async_server.py', 'r') as file:
-    #content = file.read()
-    #print("see line 640:", content)
-# ------------------------------------------------------the end--------------------------------------------------------
+chat_histories = {}
 
+def get_session_history(session_id: str) -> BaseChatMessageHistory:
+    if session_id not in chat_histories:
+        chat_histories[session_id] = InMemoryHistory()
+    return chat_histories[session_id]
 
 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
-#retrieved booking information: {table} in mind, provide a conversational answer.
-#If no booking information is requested, provide a conversational answer.
-#combined_message = f"### Informasjon for Bestillingskode:\n\n{table}" 
-
 daysoff_assistant_template = """
-You are a customer support assistant for Daysoff kundeservice and help users retrieve booking information associated with their booking IDs.
-By default, you respond using Norwegian bokmål.
-Provide a conversational answer.
-This way you directly address the user's question in a manner that reflects the professionalism and warmth
-of a customer support representative (female).
+You are a customer support assistant for Daysoff ('Daysoff Kundeservice AI Support') who helps users retrieve booking information based on their bookingnummer. 
+You should concisely use the term ’bookingnummer’. Maintain a friendly and professional tone, **reflecting the warmth of a female customer support 
+representative archetype.** By default, you answer in **Norwegian**.
 ============================
 Chat History: {chat_history}
 Question: {question}
 ============================
-Answer in Markdown:
+Answer:
 """
 
 daysoff_assistant_prompt = PromptTemplate(
@@ -114,21 +100,16 @@ async def set_starters():
 
 
 
-@cl.on_chat_start # <————————————————— the Chainlit context
+@cl.on_chat_start # <————————————————— Chainlit context |  cl.config.run.log_level = "INFO"
 def setup_multiple_chains():
-    #cl.user_session.set("connection_attempts", 0)
-    #cl.user_session.set("last_connection_time", datetime.now())
-
+    cl.user_session.set("connection_attempts", 0)
+    cl.user_session.set("last_connection_time", datetime.now())
 
-    
-    # --config Chainlit logging
-    cl.config.run.log_level = "INFO"
     print("Session id:", cl.user_session.get("id"))
-    ### Session id: 37223c22-1ea4-48f5-880f-8201ae67034b
-
-    #session_id: str
-    #ws_session = WebsocketSession.get_by_id(session_id=session_id)  # find other methods for WebsocketSession() to prevent oytage
-    #init_ws_context(ws_session)
+    
+    Session id = "37223c22-1ea4-48f5-880f-8201ae67034b" #  session_id: str
+    ws_session = WebsocketSession.get_by_id(session_id=session_id)  # find other methods for WebsocketSession() to prevent oytage
+    init_ws_context(ws_session)
     
     llm = OpenAI(
         model="gpt-3.5-turbo-instruct",
@@ -172,14 +153,12 @@ async def handle_message(message: cl.Message):
         payload = {"booking_id": bestillingskode}
 
         try:
-            # --async POST request
             response = await async_post_request(API_URL, headers, payload)
             response.raise_for_status()
             booking_data = response.json()
 
             if "booking_id" in booking_data:
                 try:
-                    # --markdown_table
                     table = (
                         "| 𝑭𝒊𝒆𝒍𝒅      | 𝗜𝗻𝗳𝗼                |\n"
                         "|:-----------|:---------------------|\n"
@@ -194,7 +173,6 @@ async def handle_message(message: cl.Message):
                         f"| 𝙄𝙣𝙘𝙡𝙪𝙙𝙚𝙙   | {booking_data.get('included', 'N/A')} |"
                     )
 
-                    # --send both as combined_message
                     combined_message = f"### Her er informasjon for {bestillingskode}:\n\n{table}"
                     await cl.Message(content=combined_message).send()
 
@@ -215,8 +193,6 @@ async def handle_message(message: cl.Message):
                 "chat_history": ""
             }, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
-            # Send the LLM response as a message
-            #await cl.Message(content=response.get("text")).send()
             await cl.Message(content=response["text"]).send()
 
         except Exception as e: