Update app.py

app.py

@@ -102,6 +102,20 @@ def setup_multiple_chains():
     cl.user_session.set("api_chain", api_chain)
 
 
+# --regex for alphanum. "LLLLLLxxxxxx", i.e. booking_id  |==> 308.9 trillion unique possibilities
+BOOKING_ID = r'\b[A-Z]{6}\d{6}\b'
+
+# TODO: add all booking-related keywords from JSON data
+BOOKING_KEYWORDS = [
+    "booking", "bestilling", "bestillingsnummer", 
+    "booking_id", "booking ID","booking nummer",
+    "navn", "etternavn", "kunde", "customer",
+    "utestående beløp", "faktura", "cost", "price",
+    "date", "dato", "checkin", "checkout",
+    "address", "adresse", "user id", "bruker_id", "bruker id",  
+]
+
+
 # --wrapper function around the @cl.on_message decorator; chain trigger(s)
 @cl.on_message
 async def handle_message(message: cl.Message):
@@ -109,18 +123,20 @@ async def handle_message(message: cl.Message):
     llm_chain = cl.user_session.get("llm_chain")
     api_chain = cl.user_session.get("api_chain")
 
-    if any(keyword in user_message for keyword in ["booking_id", "full_name", "amount",  # + "bestillingsnummer", "checkin", "checkout" for api_docs
-                                                   "date", "address", "amount", "user_id"]):
-        # --if any keywords in user_message, use api_chain
+    is_booking_query = any(
+        re.search(keyword, user_message, re.IGNORECASE) 
+        for keyword in BOOKING_KEYWORDS + [BOOKING_ID]
+    )
+
+    if is_booking_query:
         response = await api_chain.acall(user_message,
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])
     else:
-        # --defaults to llm_chain4general queries
         response = await llm_chain.acall(user_message,
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])
+    
     response_key = "output" if "output" in response else "text"
     await cl.Message(response.get(response_key, "")).send()
     return message.content
 
 
-