Update app.py

app.py

@@ -1,5 +1,5 @@
 # ===========================================
-# title: daysoff-assistant-API-v2
+# title: daysoff-assistant-API-v2, table
 # file: app.py
 # ===========================================
 
@@ -29,16 +29,26 @@ 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.
+
+
+
+
 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).
+============================
 Chat History: {chat_history}
 Question: {question}
-Answer:
+============================
+Answer in Markdown:
 """
+
 daysoff_assistant_prompt = PromptTemplate(
     input_variables=["chat_history", "question"],
     template=daysoff_assistant_template,
@@ -48,6 +58,30 @@ 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():
@@ -100,29 +134,37 @@ async def handle_message(message: cl.Message):
             if "booking_id" in booking_data:
                 try:
                     # --markdown_table
-                    table = (
+                    #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')} |"
+                        #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')} |"
                     )
 
+                    # --invoke LLM w/ table + user_message
+                    response = await llm_chain.ainvoke({
+                        #"table": table,
+                        "booking_data": booking_data,
+                        "question": user_message,
+                        "chat_history": ""
+                    }, callbacks=[cl.AsyncLangchainCallbackHandler()])
+
                     # --send both as combined_message
-                    combined_message = f"### Informasjon for Bestillingskode:\n\n{table}"
+                    #combined_message = f"### Informasjon for Bestillingskode:\n\n{table}" 
                     await cl.Message(content=combined_message).send()
 
                 except Exception as e:
                     await cl.Message(content=f"Error processing booking data: {str(e)}").send()
 
             else:
-                await cl.Message(content="Booking not found").send()
+                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()