Update app.py

app.py

@@ -1,5 +1,6 @@
 # ===========================================
-# !-----app.py
+# title: daysoff-assistant-API-v2
+# file: app.py
 # ===========================================
 
 import json
@@ -15,18 +16,25 @@ from langchain.chains import LLMChain
 from langchain_core.prompts import PromptTemplate
 from langchain.memory.buffer import ConversationBufferMemory
 
+# ---------------------------------------------------for backend looks, example file:----------------------------------
+
+#with open('/usr/local/lib/python3.10/site-packages/transformers/utils/chat_template_utils.py', 'r') as file:
+    #content = file.read()
+    #print("base.py:", content)
+# ------------------------------------------------------the end--------------------------------------------------------
+
 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/"
 
-# You help users retrieve booking information associated with their booking IDs.Provide a conversational answer.
 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 in Norwegian language.
-You should always keep in mind the professionalism and warmth of a real human female customer support representative when you provide users with their
-the reqyested booking information in a conversational answer.
+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).
 Chat History: {chat_history}
 Question: {question}
 Answer:
@@ -40,10 +48,35 @@ daysoff_assistant_prompt = PromptTemplate(
 async def async_post_request(url, headers, data):
     return await asyncio.to_thread(requests.post, url, headers=headers, json=data)
 
+@cl.set_starters
+async def set_starters():
+    return [
+        cl.Starter(
+            label="Booking ID request",
+            message="Kan du gi meg info om en reservasjon?",
+            icon="/public/booking_id.svg",
+            ),
+        cl.Starter(
+            label="Metric Space Self-Identity Framework",
+            message="Explain the Metric Space Self-Identity Framework like I'm five years old.",
+            icon="/public/learn.svg",
+            ),
+        cl.Starter(
+            label="Python script for daily email reports",
+            message="Write a script to automate sending daily email reports in Python, and walk me through how I would set it up.",
+            icon="/public/terminal.svg",
+            ),
+        cl.Starter(
+            label="Morning routine ideation",
+            message="Can you help me create a personalized Yoga/pranayama/meditation morning routine that would help increase my productivity throughout the day? Start by asking me about my current habits and what activities energize me in the morning.",
+            icon="/public/idea.svg",
+            )
+        ]
+
 @cl.on_chat_start
 def setup_multiple_chains():
     llm = OpenAI(
-        model="gpt-3.5-turbo-instruct", # gpt-3.5-turbo-instruct
+        model="gpt-3.5-turbo-instruct",
         temperature=0.7,
         openai_api_key=OPENAI_API_KEY,
         max_tokens=2048,
@@ -54,8 +87,6 @@ def setup_multiple_chains():
 
     conversation_memory = ConversationBufferMemory(
         memory_key="chat_history",
-        #input_key="question",  # ?
-        #output_key="text",     # ?
         max_len=30,
         return_messages=True
     )
@@ -72,12 +103,12 @@ def setup_multiple_chains():
 async def handle_message(message: cl.Message):
     user_message = message.content
     llm_chain = cl.user_session.get("llm_chain")
-    
-    booking_pattern = r'\b[A-Z]{6}\d{6}\b'   
+
+    booking_pattern = r'\b[A-Z]{6}\d{6}\b'
     match = re.search(booking_pattern, user_message)
 
     if match:
-        
+
         bestillingskode = match.group()
         headers = {
             "Authorization": auth_token,
@@ -86,43 +117,52 @@ async def handle_message(message: cl.Message):
         payload = {"booking_id": bestillingskode}
 
         try:
-            
+
             response = await async_post_request(API_URL, headers, payload)
-            response.raise_for_status() 
+            response.raise_for_status()
             booking_data = response.json()
 
             if "booking_id" in booking_data:
-    
-                table = (
-                    "| Field      | Info                |\n"
-                    "|:-----------|:---------------------|\n"
-                    f"| Booking ID | {booking_data.get('booking_id', 'N/A')} |\n"
-                    f"| Full Name  | {booking_data.get('full_name', 'N/A')} |\n"
-                    f"| Amount     | {booking_data.get('amount', 0)} kr |\n"
-                    f"| Check-in   | {booking_data.get('checkin', 'N/A')} |\n"
-                    f"| Check-out  | {booking_data.get('checkout', 'N/A')} |\n"
-                    f"| Address    | {booking_data.get('address', 'N/A')} |\n"
-                    f"| User ID    | {booking_data.get('user_id', 0)} |\n"
-                    f"| Info Text  | {booking_data.get('infotext', 'N/A')} |\n"
-                    f"| Included   | {booking_data.get('included', 'N/A')} |"
-                )
-                
-                await cl.Message(content=table).send()
+
+                try:
+                    table = (
+                        "| 𝑭𝒊𝒆𝒍𝒅      | 𝗜𝗻𝗳𝗼                |\n"
+                        "|:-----------|:---------------------|\n"
+                        f"| 𝙱𝚎𝚜𝚝𝚒𝚕𝚕𝚒𝚗𝚐𝚜𝚔𝚘𝚍𝚎 | {booking_data.get('booking_id', 'N/A')} |\n"
+                        f"| 𝙁𝙪𝙡𝙡 𝙉𝙖𝙢𝙚  | {booking_data.get('full_name', 'N/A')} |\n"
+                        f"| 𝘼𝙢𝙤𝙪𝙣𝙩     | {booking_data.get('amount', 0)} kr |\n"
+                        f"| 𝘾𝙝𝙚𝙘𝙠-𝙞𝙣   | {booking_data.get('checkin', 'N/A')} |\n"
+                        f"| 𝘾𝙝𝙚𝙘𝙠-𝙤𝙪𝙩  | {booking_data.get('checkout', 'N/A')} |\n"
+                        f"| 𝘼𝙙𝙙𝙧𝙚𝙨𝙨    | {booking_data.get('address', 'N/A')} |\n"
+                        f"| 𝙐𝙨𝙚𝙧 𝙄𝘿    | {booking_data.get('user_id', 0)} |\n"
+                        f"| 𝙄𝙣𝙛𝙤 𝙏𝙚𝙭𝙩  | {booking_data.get('infotext', 'N/A')} |\n"
+                        f"| 𝙄𝙣𝙘𝙡𝙪𝙙𝙚𝙙   | {booking_data.get('included', 'N/A')} |"
+                    )
+
+                    response = await llm_chain.ainvoke({
+                        "table": table,
+                        "question": user_message,
+                        "chat_history": ""
+                    }, callbacks=[cl.AsyncLangchainCallbackHandler()])
+
+                    await cl.Message(content=table,["text"]).send()
+
+                except Exception as e:
+                    except requests.exceptions.RequestException as e:
+                await cl.Message(content=f"Request failed: {str(e)}").send()
+
 
             else:
-                await cl.Message(content="Booking not found or invalid response.").send()
-                
-        except requests.exceptions.RequestException as e:
-            await cl.Message(content=f"Request failed: {str(e)}").send()
+                await cl.Message(content="Booking information not found.").send()
     else:
         try:
             response = await llm_chain.ainvoke({
                 "question": user_message,
                 "chat_history": ""
             }, callbacks=[cl.AsyncLangchainCallbackHandler()])
-            
+
             await cl.Message(content=response["text"]).send()
-            
+
         except Exception as e:
             await cl.Message(content=f"Error: {str(e)}").send()