Update app.py

app.py

@@ -1,6 +1,6 @@
 
 # ===========================================
-# ver01.01-5.workload-----app.py
+# ver02.02-Zeta.workload-----app.py
 # ===========================================
 
 import asyncio
@@ -22,10 +22,8 @@ from langchain.memory.buffer import ConversationBufferMemory
 from langchain.memory import ConversationTokenBufferMemory
 from langchain.memory import ConversationSummaryMemory
 
-
 from api_docs_mck import api_docs_str 
-#from faq_data import ansatte_faq_data, utleiere_faq_data
-#from personvernspolicy import personvernspolicy_data
+
 
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 
@@ -58,7 +56,6 @@ API URL:
 api_url_prompt = PromptTemplate(input_variables=['api_docs', 'question'],
                                 template=api_url_template)
 
-# If the response includes booking information, provide the information verbatim (do not summarize it.)
     
 api_response_template = """
 With the API Documentation for Daysoff's official API: {api_docs} in mind,
@@ -138,11 +135,6 @@ 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()
@@ -166,28 +158,38 @@ async def handle_message(message: cl.Message):
 
     
     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}"
+    bestillingskode = re.search(booking_pattern, user_message).group(0)  
+    question = f"Retrieve information for booking ID {endpoint_url}?search={bestillingskode}"
    
-        response = await api_chain.acall(
-            {
-                "bestillingskode": bestillingskode,
-                "question": question
-              
-            },
-            callbacks=[cl.AsyncLangchainCallbackHandler()])
+    response = await api_chain.acall(
+        {
+            "bestillingskode": bestillingskode,
+            "question": question
+        },
+        callbacks=[cl.AsyncLangchainCallbackHandler()]
+    )
 
     elif any(keyword.lower() in user_message.lower() for keyword in all_keywords):
-
+        # --match in keywords/iterate
+        matched_entry = None
         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()])
-
+                matched_entry = qa_entry
+                break  # Exit loop once a match is found
+    
+        if matched_entry:
+            # --question@matched entry
+            question = f"User Question: {user_message}\nMatched FAQ:\n{matched_entry}"
+            response = await llm_chain.acall(
+                {
+                    "question": question
+                },
+                callbacks=[cl.AsyncLangchainCallbackHandler()]
+            )
+        else:
+            
+            response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+    
     else:    
         response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])