Update app.py

app.py

@@ -1,6 +1,6 @@
 
 # ===========================================
-# ver2.01-----app.py
+# ver01.01-5.workload-----app.py
 # ===========================================
 
 import asyncio
@@ -8,61 +8,35 @@ import os
 import re
 import time
 import json
-import torch
-
-import logging
-
-from api_docs_mck import api_docs_str 
 
 import chainlit as cl
 
 from langchain import hub
+from langchain_openai import OpenAI
 from langchain.chains import LLMChain, APIChain
 from langchain_core.prompts import PromptTemplate
 from langchain.memory.buffer import ConversationBufferMemory
 
-from langchain_openai import OpenAI
-
-from langchain_community.llms import HuggingFaceHub
-from langchain_huggingface import HuggingFacePipeline
-from langchain_huggingface import HuggingFaceEndpoint
-from langchain_core.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
+from api_docs_mck import api_docs_str 
+from faq_data import help, ansatte_faq_data, utleiere_faq_data
+from personvernspolicy import help2, personvernspolicy_data
 
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
-#HF_INFERENCE_ENDPOINT = 
-#BOOKING_ID = re.compile(r'\b[A-Z]{6}\d{6}\b')
-#HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
-
-BOOKING_KEYWORDS = [
-    "booking",
-    "bestillingsnummer",
-    "bookingen",
-    "ordrenummer",
-    "reservation",
-    "rezerwacji",
-    "bookingreferanse",
-    "rezerwacja",
-    "booket",
-    "reservation number",
-    "bestilling",
-    "order number",
-    "booking ID",
-    "identyfikacyjny płatności"
-]
 
 daysoff_assistant_template = """
 You are a customer support assistant (’kundeservice AI assistent’) for Daysoff.no
 By default, you respond in Norwegian language, using a warm, direct and professional tone. 
-Your expertise is exclusively in in providing information related to a given booking ID (’bestillingsnummer’)  
-and booking-related queries such as firmahytteordning and personvernspolicy. 
-You do not provide information outside of this scope. If a question is not about booking or booking-related queries,
-respond with, "Ønsker du annen informasjon, må du kontakte oss her på [email protected]"
+You can provide information associated with a given booking ID, and with {help} in mind, you
+can also answer frequently asked questions (FAQ) about DaysOff firmahytteordning for 
+employees: {ansatte_faq_data} and for employers:{utleiere_faq_data}.
+To understand how best to answer queries about privacy policy, 
+refer to {help2} and {personvernspolicy_data}
 Chat History: {chat_history}
 Question: {question}
 Answer:
 """
 
-daysoff_assistant_prompt= PromptTemplate(
+daysoff_assistant_prompt = PromptTemplate(
     input_variables=["chat_history", "question"],
     template=daysoff_assistant_template
 )
@@ -79,7 +53,6 @@ API URL:
 api_url_prompt = PromptTemplate(input_variables=['api_docs', 'question'],
                                 template=api_url_template)
 
-# (..) If {question} contains an alphanumeric identifier consisting of 6 letters followed by 6 digits (e.g., DAGHNS116478)
 api_response_template = """
 With the API Documentation for Daysoff's official API: {api_docs} in mind, 
 and  the specific user question: {question} in mind,
@@ -152,7 +125,7 @@ async def handle_message(message: cl.Message):
               
             },
             callbacks=[cl.AsyncLangchainCallbackHandler()])
-
+        
     else:  
         response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])