Update app.py

app.py

@@ -151,9 +151,39 @@ def setup_multiple_chains():
     )
 
     cl.user_session.set("api_chain", api_chain)
+
+@cl.on_message
+async def handle_message(message: cl.Message):
+    user_message = message.content
+    llm_chain = cl.user_session.get("llm_chain")
+    api_chain = cl.user_session.get("api_chain")
     
-import logging
+    booking_pattern = r'\b[A-Z]{6}\d{6}\b' 
+    base_url = "https://670dccd0073307b4ee447f2f.mockapi.io/daysoff/api/V1/booking"
 
+    try:
+        if re.search(booking_pattern, user_message):  
+            booking_id = re.search(booking_pattern, user_message).group(0)  
+            logging.debug(f"Booking ID detected: {booking_id}")
+            
+            url = f"{base_url}?search={booking_id}"
+            logging.debug(f"Constructed API URL: {url}")
+
+            response = await api_chain.acall({"booking_id": booking_id}, callbacks=[cl.AsyncLangchainCallbackHandler()])
+        else: 
+            logging.debug("Triggered LLMChain for <booking-related>")
+            response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+
+    except Exception as e:
+        logging.error(f"Error in processing message: {str(e)}")
+        response = {"output": "Jeg får desverre ikke fram din informasjon akkurat nå."}
+
+    response_key = "output" if "output" in response else "text"
+    await cl.Message(response.get(response_key, "")).send()
+    return message.content
+
+    
+"""
 @cl.on_message
 async def handle_message(message: cl.Message):
     user_message = message.content
@@ -161,6 +191,9 @@ async def handle_message(message: cl.Message):
     api_chain = cl.user_session.get("api_chain")
 
     # api_keywords = ["firmahytteordning", "personvernspolicy"]
+
+    #base_url = "https://670dccd0073307b4ee447f2f.mockapi.io/daysoff/api/V1/booking"
+    #url = f"{base_url}?search={booking_id}"
     
     try:
         # --check message for booking ID
@@ -168,13 +201,13 @@ async def handle_message(message: cl.Message):
             logging.debug(f"Booking ID detected in message: {user_message}")
             response = await api_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
-        # --check message for API keywords
-        elif any(keyword in user_message for keyword in ["firmahytteordning", "personvernspolicy"]):   # any(keyword in user_message for keyword in api_keywords):
-            logging.debug(f"API keyword detected in message: {user_message}")
-            response = await api_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+        # --check message for  keywords
+        #elif any(keyword in user_message for keyword in ["firmahytteordning", "personvernspolicy"]):   # any(keyword in user_message for keyword in api_keywords):
+            #logging.debug(f"API keyword detected in message: {user_message}")
+            #response = await api_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
         else: 
-            logging.debug("Triggering LLMChain for everything else.")
+            logging.debug("Triggers LLMChain for everything else.")
             response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
     except Exception as e:
@@ -185,6 +218,9 @@ async def handle_message(message: cl.Message):
     await cl.Message(response.get(response_key, "")).send()
     return message.content
 
+"""
+
+
 
 
 #def is_booking_query(user_message):