Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 import re
 import time
 import json
+import pandas as pd
 
 import chainlit as cl
 
@@ -136,15 +137,34 @@ 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 #.lower()
     llm_chain = cl.user_session.get("llm_chain")
     api_chain = cl.user_session.get("api_chain")
     
+# --regex
     booking_pattern = r'\b[A-Z]{6}\d{6}\b' 
+# --endpoint
     endpoint_url = "https://670dccd0073307b4ee447f2f.mockapi.io/daysoff/api/V1/booking"
+# --dataframe
+    personvernspolicy_df = pd.read_csv('personvernspolicy.csv')
+    faq_df = pd.read_csv('faq.csv')
+# --dictionaries
+    personvernspolicy_qa = personvernspolicy_df.to_dict(orient='records')
+    faq_qa = faq_df.to_dict(orient='records')
+# --keywords
+    personvernspolicy_keywords = personvernspolicy_df['question'].tolist()
+    faq_keywords = faq_df['question'].tolist()
+    all_keywords = personvernspolicy_keywords + faq_keywords
 
+    
     if re.search(booking_pattern, user_message):  
         bestillingskode = re.search(booking_pattern, user_message).group(0)  
         question = f"Retrieve information for booking ID {endpoint_url}?search={bestillingskode}"
@@ -157,6 +177,17 @@ async def handle_message(message: cl.Message):
             },
             callbacks=[cl.AsyncLangchainCallbackHandler()])
 
+    elif any(keyword.lower() in user_message.lower() for keyword in all_keywords):
+
+        for qa_entry in personvernspolicy_qa + faq_qa:
+            if qa_entry['question'].lower() in user_message.lower():
+                
+                response = llm_chain.acall(f"User Question: {user_message}\nMatched FAQ:\n{qa_entry}")
+                return response
+
+            else:
+                response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+
     else:    
         response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])