Update app.py

app.py

@@ -157,13 +157,48 @@ async def handle_message(message: cl.Message):
     llm_chain = cl.user_session.get("llm_chain")
     post_tool = cl.user_session.get("post_tool")
 
+    # Match booking pattern
+    booking_pattern = r'\b[A-Z]{6}\d{6}\b'
+    endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
+
+    match = re.search(booking_pattern, user_message)
+    if match:
+        bestillingskode = match.group(0)
+        post_data = {
+            "url": endpoint_url,
+            "body": {
+                "booking_id": bestillingskode
+            }
+        }
+        try:
+            response = await post_tool.acall(
+                json.dumps(post_data),
+                callbacks=[cl.AsyncLangchainCallbackHandler()]
+            )
+            response_key = "output" if "output" in response else "text"
+            await cl.Message(content=response.get(response_key, "")).send()
+        except Exception as e:
+            await cl.Message(content=f"Error: {str(e)}").send()
+    else:
+        response = await llm_chain.ainvoke(
+            {"chat_history": "", "question": user_message},
+            callbacks=[cl.AsyncLangchainCallbackHandler()]
+        )
+        await cl.Message(content=response).send()
+
+"""
+async def handle_message(message: cl.Message):
+    user_message = message.content
+    llm_chain = cl.user_session.get("llm_chain")
+    cl.user_session.set("post_tool", post_tool)
+
     # --------------------------------AD VAL.II.7--------------------------------
     booking_pattern = r'\b[A-Z]{6}\d{6}\b'
     endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
-    
-    if re.search(booking_pattern, user_message):
-        bestillingskode = re.search(booking_pattern, user_message).group(0)
-        
+
+    match = re.search(booking_pattern, user_message)
+    if match:
+        bestillingskode = match.group(0)   
         
         post_data = {
             "url": endpoint_url,
@@ -185,6 +220,10 @@ async def handle_message(message: cl.Message):
         )
 
     response_key = "output" if "output" in response else "text"
+
+    
     await cl.Message(response.get(response_key, "")).send()
     return message.content
 
+
+"""