Update app.py

app.py

@@ -31,7 +31,6 @@ BOOKING_KEYWORDS = [
     "rezerwacji",
     "bookingreferanse",
     "rezerwacja",
-    "logg inn",
     "booket",
     "reservation number",
     "bestilling",
@@ -41,19 +40,19 @@ BOOKING_KEYWORDS = [
 ]
 
 
-daysoff_assistant_booking_template = """
-You are a Norwegian customer support AI assistant for Daysoff.no. By default, 
-you respond in Norwegian to user question: {question}. In all other cases,
+daysoff_assistant_system_template = """
+You are an AI customer support assistant for Daysoff. By default, 
+you respond in Norwegian to {question}. In all other cases,
 adapt to user's language and respond accordingly. 
-You can retrieving booking information for a given booking ID and provide 
-general helpful info about Daysoff's services."
+You can retrieving booking information for a given booking ID. In addition,
+you can inform on Daysoff's personvernspolicy and verticals."
 Chat History: {chat_history}
 Question: {question}
 Answer:
 """
-daysoff_assistant_booking_prompt= PromptTemplate(
+daysoff_assistant_system_prompt= PromptTemplate(
     input_variables=["chat_history", "question"],
-    template=daysoff_assistant_booking_template
+    template=daysoff_assistant_system_template
 )
 
 api_url_template = """
@@ -69,15 +68,15 @@ api_url_prompt = PromptTemplate(input_variables=['api_docs', 'question'],
                                 template=api_url_template)
 
 api_response_template = """
-With the API Documentation for Daysoff's official API: {api_docs}
-in mind, and IF {question} contains keywords from:{BOOKING_KEYWORDS},
-and given this API URL: {api_url} for querying, here is the
+IF and only IF {question} contains alphanumeric identifier:{BOOKING_ID},
+with the API Documentation for Daysoff's official API: {api_docs}
+in mind, and given this API URL: {api_url} for querying, here is the
 response from Daysoff's API: {api_response}.
-Please provide a summary that directly addresses the user's question, 
+Please provide only information that directly addresses the user's question, 
 omitting technical details like response format, and 
 focusing on delivering the answer with clarity and conciseness, 
-as if Daysoff's kundeservice is providing this information themselves.
-Booking Info:
+as if Daysoff's human customer service agent is providing this information themselves.
+Her er informasjon om bestilligen:
 """
 
 api_response_prompt = PromptTemplate(
@@ -119,40 +118,24 @@ def setup_multiple_chains():
 
     cl.user_session.set("api_chain", api_chain)
     
-import re
 
-def extract_booking_id(user_message):
-# Extract booking ID from user message
-    match = re.search(r'\b[A-Z]{6}\d{6}\b', user_message) 
-    return match.group(0) if match else None
-    
+
 @cl.on_message
 async def handle_message(message: cl.Message):
     user_message = message.content.lower()
     llm_chain = cl.user_session.get("llm_chain")
     api_chain = cl.user_session.get("api_chain")
 
-    # booking_id = extract_booking_id(user_message)
+    def is_booking_query(user_message):
+        match = re.search(r'\b[A-Z]{6}\d{6}\b', user_message)
 
-    
-    #is_booking_query = any(
-        #re.search(keyword, user_message, re.IGNORECASE) # re.search(keyword, r'\b[A-Z]{6}\d{6}\b', user_message, re.IGNORECASE)
-        #for keyword in BOOKING_KEYWORDS + [BOOKING_ID]
-    #)
-    if any(keyword in user_message for keyword in ["BOOKING_KEYWORDS"]): # could be(?): if any(keyword in user_message for keyword in BOOKING_KEYWORDS + [BOOKING_ID]
-            
-        # <f any of the keywords are in user_message>,  use api_chain
-        response = await api_chain.acall(user_message, 
-                                         callbacks=[cl.AsyncLangchainCallbackHandler()])
+        return match
 
-    elif: 
-        # <or if the booking_id regex format is discovered in user message>, use api_chain
-        re.search(keyword, r'\b[A-Z]{6}\d{6}\b', user_message):  
-            response = await api_chain.acall(user_message, 
+    if match: 
+        response = await api_chain.acall(user_message, 
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])
 
-    else:
-        
+    else: 
         response = await llm_chain.acall(user_message,
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])