Update app.py

app.py

@@ -11,8 +11,6 @@ import torch
 import logging
 
 from api_docs_mck import api_docs_str 
-#from daysoff import daysoff_str                        ## make daysoff.py, put json info in dict.
-#from personvernpolicy import personvernpolicy_str      ## make personvernpolicy.py, put json info in dict.
 
 import chainlit as cl
 
@@ -28,7 +26,6 @@ from langchain_huggingface import HuggingFacePipeline
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain_core.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
-#logging.basicConfig(level=logging.DEBUG)
 HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 #HF_INFERENCE_ENDPOINT = 
@@ -51,7 +48,6 @@ BOOKING_KEYWORDS = [
     "identyfikacyjny płatności"
 ]
 
-
 daysoff_assistant_template = """
 You are a customer support assistant (’kundeservice AI assistent’) for Daysoff named "Agrippa". 
 Your primary objective is to provide exceptional, empathetic, and efficient customer service.
@@ -177,9 +173,6 @@ def setup_multiple_chains():
 
     cl.user_session.set("api_chain", api_chain)
     
-
-
-
 import logging
 
 @cl.on_message
@@ -190,7 +183,6 @@ async def handle_message(message: cl.Message):
 
     api_keywords = ["firmahytteordning", "personvernspolicy"]
     
-    # --check message for a booking ID
     try:
         # --check message for booking ID
         if re.search(r'\b[A-Z]{6}\d{6}\b', user_message):  
@@ -214,35 +206,13 @@ async def handle_message(message: cl.Message):
     await cl.Message(response.get(response_key, "")).send()
     return message.content
 
-"""
-@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")
 
-    #if any(keyword in user_message for keyword in ["firmahytteordning","personvernspolicy"]):
-    
-       
-    #def is_booking_query(user_message):
+
+#def is_booking_query(user_message):
         #match = re.search(r'\b[A-Z]{6}\d{6}\b', user_message)
         #return match is not None  # --works boolean
     #booked = is_booking_query(user_message)
-    #if booked: 
-
-    if re.search(r'\b[A-Z]{6}\d{6}\b', user_message):   # ex. "EQJLCQ362149"
-        
-        response = await api_chain.ainvoke(user_message, 
-                                         callbacks=[cl.AsyncLangchainCallbackHandler()])
-
-    else: 
-        response = await llm_chain.ainvoke(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
-"""
+    #if booked:
+        #response = await api_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+    # etc..