update

app.py

@@ -28,7 +28,7 @@ def chatbot_response(user_input):  # Open AI GPT-3 and Google T5
 
 def main():
     print_colorful_msg("Starting chatbot main function ...", color='blue')
-    st.title("Simple Chatbot")
+    st.title("ATrad Chatbot")
 
     user_input = st.text_input("Enter your message:")
     if st.button("Send"):
@@ -45,8 +45,8 @@ def main():
         st.markdown("### Source Documents:")
         if source_documents:
             for doc in source_documents:
-                # pass
-                st.write(doc)
+                pass
+                # st.write(doc)
         else:
             st.write("No source documents found.")
         

model.py

@@ -4,6 +4,8 @@ from dotenv import load_dotenv
 # Load environment variables from .env file
 load_dotenv()
 
+
+
 # Access the value of OPENAI_API_KEY
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 
@@ -12,11 +14,18 @@ os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
 from langchain_openai import ChatOpenAI
 
 llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0,)
+
+
+
 ## Create embeddings and splitter
 
 from langchain.embeddings import HuggingFaceBgeEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 
+
+
+
+
 # Create Embeddings
 model_name = "BAAI/bge-large-en"
 
@@ -32,12 +41,18 @@ splitter = RecursiveCharacterTextSplitter(
     chunk_overlap=100,
 )
 
+
+
+
 from langchain_community.vectorstores import FAISS
 
 persits_directory="./faiss_Test02_500_C_BGE_large"
 
 vectorstore= FAISS.load_local(persits_directory, embedding) 
 
+
+
+
 # Define a custom prompt for Unser manual
 from langchain.prompts import PromptTemplate
 
@@ -52,7 +67,7 @@ qa_template = ("""
     if context is not enough to answer the question, ask for more information.
     if context is not related to the question, say I dont know.
     
-    each answer should start with code word ATrad Ai(QA):
+    each answer Must start with code word ATrad Ai(QA):
     
     Question: {question}
     
@@ -69,6 +84,8 @@ qa_template2 = ("""
     
     Question: {question}
     
+    each answer Must start with code word ATrad Ai(QA):
+    
     ATrad Ai(QA): Let me think about it...""")