Update app.py

app.py

@@ -19,14 +19,14 @@ import chainlit as cl
 from langchain import hub
 from langchain.chains import LLMChain, APIChain
 from langchain_core.prompts import PromptTemplate
-#from langchain_community.llms import HuggingFaceHub
+from langchain_community.llms import HuggingFaceHub
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain.memory.buffer import ConversationBufferMemory
 
 logging.basicConfig(level=logging.DEBUG)
 
 HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
-BOOKING_ID = re.compile(r'\b[A-Z]{6}\d{6}\b')
+#BOOKING_ID = re.compile(r'\b[A-Z]{6}\d{6}\b')
 BOOKING_KEYWORDS = [
     "booking",
     "bestillingsnummer",
@@ -96,7 +96,7 @@ def setup_multiple_chains():
     huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN, 
     #max_new_tokens=512,  
     temperature=0.7,     
-    task="text-generation"  
+    #task="text-generation"  
     )
 
     conversation_memory = ConversationBufferMemory(memory_key="chat_history",
@@ -137,17 +137,18 @@ async def handle_message(message: cl.Message):
     #booked = is_booking_query(user_message)
     #if booked: 
 
-    if re.search(r'\b[A-Z]{6}\d{6}\b', user_message):   # ex., "Hei, har du booking info for EQJLCQ362149?"
-        logging.debug(f"User message: {user_message}")
+    if re.search(r'\b[A-Z]{6}\d{6}\b', user_message):   # ex. "EQJLCQ362149"
+        
         response = await api_chain.acall(user_message, 
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])
 
     else: 
         response = await llm_chain.acall(user_message,
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])
-
-        logging.debug(f"API response: {response}")
+      
+        
 
     response_key = "output" if "output" in response else "text"
+    logging.debug({response}")
     await cl.Message(response.get(response_key, "")).send()
     return message.content