Update app.py

app.py

@@ -19,7 +19,7 @@ from langchain.memory.buffer import ConversationBufferMemory
 
 from api_docs_mck import api_docs_str 
 from faq_data import ansatte_faq_data, utleiere_faq_data
-from personvernspolicy.py import personvernspolicy_data
+from personvernspolicy import personvernspolicy_data
 
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 
@@ -93,7 +93,8 @@ With the API Documentation for Daysoff's official API: {api_docs} in mind,
 and the specific user question: {question},
 and given this API URL: {api_url} for querying,
 here is the response from Daysoff's API: {api_response}.
-If the response includes a booking ID (bestillingskode), present this information directly to the user without summarizing it.
+If the response includes a booking ID (bestillingskode), consistently present this information 
+directly to the user without summarizing it.
 For all other responses, please provide a clear and concise summary (in Norwegian) that directly addresses the user's question,
 delivered in a manner that reflects the professionalism and warmth of a human customer service agent.
 Summary:
@@ -148,6 +149,14 @@ async def handle_message(message: cl.Message):
     booking_pattern = r'\b[A-Z]{6}\d{6}\b' 
     base_url = "https://670dccd0073307b4ee447f2f.mockapi.io/daysoff/api/V1/booking"
 
+    faq_keywords = (
+    [item["question"] for item in ansatte_faq_data] + 
+    [item["answer"] for item in ansatte_faq_data] +   
+    [item["question"] for item in utleiere_faq_data] + 
+    [item["answer"] for item in utleiere_faq_data] +   
+    [key for key in personvernspolicy_data.keys()]    
+    )
+
     if re.search(booking_pattern, user_message):  
         bestillingskode = re.search(booking_pattern, user_message).group(0)  
         question = f"Retrieve information for booking ID {base_url}?search={bestillingskode}"
@@ -160,11 +169,10 @@ async def handle_message(message: cl.Message):
             },
             callbacks=[cl.AsyncLangchainCallbackHandler()])
 
-    elif any(keyword in user_message for keyword in (["personvernspolicy_data", "ansatte_faq_data", "utleiere_faq_data"])): 
-        
-        response = await api_chain.acall(user_message,
-                                         callbacks=[cl.AsyncLangchainCallbackHandler()])
-    
+    elif any(keyword in user_message for keyword in faq_keywords): 
+
+        response = await api_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+
     else:    
         response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])