Update app.py

app.py

@@ -31,6 +31,8 @@ if not openai_api_key or not serper_api_key:
     logger.error("API keys are not set properly.")
     st.error("API keys for OpenAI and SERPER must be set in the .env file.")
     st.stop()
+else:
+    logger.info("API keys loaded successfully.")
 
 # Initialize OpenAI client
 try:
@@ -44,7 +46,7 @@ except Exception as e:
 # Load knowledge base
 def load_knowledge_base():
     try:
-        loader = TextLoader(r"time_to_rethink_trust_book.md")
+        loader = TextLoader("./data_source/time_to_rethink_trust_book.md")
         documents = loader.load()
         text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
         docs = text_splitter.split_documents(documents)
@@ -124,8 +126,8 @@ def google_search_tool(query: str):
     """
     return google_search(query)
 
-tools = [
-    knowledge_base_tool,google_search_tool]
+tools = [knowledge_base_tool, google_search_tool]
+
 # Create the prompt template
 prompt_message = """
 Act as an expert copywriter who specializes in creating compelling marketing copy using AI technologies.
@@ -144,11 +146,12 @@ prompt_template = ChatPromptTemplate.from_messages(
 
 # Create Langchain Agent with specific model and temperature
 try:
-    llm = ChatOpenAI(model="gpt-4o", temperature=0.5)
+    llm = ChatOpenAI(model="gpt-4o", temperature=0.5)  # Set temperature to 0.5
     llm_with_tools = llm.bind_tools(tools)
 except Exception as e:
     logger.error(f"Error creating Langchain Agent: {e}")
-    
+    st.error(f"Error creating Langchain Agent: {e}")
+    st.stop()
 
 # Define the agent pipeline to handle the conversation flow
 try:
@@ -192,6 +195,7 @@ def chatbot_response(message, history):
 
 # Streamlit app
 
+
 # Create input field for user message
 user_input = st.text_input("You:", "")