Update app.py

app.py

@@ -108,7 +108,6 @@ async def handle_message(message: cl.Message):
     match = re.search(booking_pattern, user_message)
 
     if match:
-
         bestillingskode = match.group()
         headers = {
             "Authorization": auth_token,
@@ -117,14 +116,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"
@@ -139,28 +138,36 @@ async def handle_message(message: cl.Message):
                         f"| 𝙄𝙣𝙘𝙡𝙪𝙙𝙚𝙙   | {booking_data.get('included', 'N/A')} |"
                     )
 
+                    # --invoke LLM w/ table + user_message
                     response = await llm_chain.ainvoke({
                         "table": table,
                         "question": user_message,
                         "chat_history": ""
                     }, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
-                    await cl.Message(content=table).send()
+                    # --send both as combined_message
+                    combined_message = f"### Booking Details Table:\n\n{table}\n\n### LLM Response:\n\n{response.get('text', 'No response available.')}"
+                    await cl.Message(content=combined_message).send()
 
                 except Exception as e:
-                    except requests.exceptions.RequestException as e:
-                await cl.Message(content=f"Request failed: {str(e)}").send()
-
+                    await cl.Message(content=f"Error processing booking data: {str(e)}").send()
 
             else:
-                await cl.Message(content="Booking information 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()
+
     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: