Update app.py

app.py

@@ -1,5 +1,6 @@
 # ===========================================
-# title: daysoff-assistant-API-v2, table
+# title: daysoff-assistant-API-v2, 
+# chainlit==1.1.404
 # file: app.py
 # ===========================================
 
@@ -83,7 +84,32 @@ 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 # <————————————————— the Chainlit context
 def setup_multiple_chains():
     llm = OpenAI(
@@ -115,6 +141,7 @@ 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'
     match = re.search(booking_pattern, user_message)
 
@@ -127,12 +154,14 @@ 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"
@@ -148,25 +177,28 @@ async def handle_message(message: cl.Message):
                     )
 
                     # --send both as combined_message
-                    combined_message = f"### Informasjon for Bestillingskode:\n\n{table}"
+                    combined_message = f"### Her er 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").send()
 
         except requests.exceptions.RequestException as e:
             await cl.Message(content=f"Request failed: {str(e)}").send()
 
     else:
         try:
+            # --invoke LLM w/ user_message
             response = await llm_chain.ainvoke({
                 "question": user_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: