Update app.py

app.py

@@ -95,7 +95,7 @@ def preload_modtran_document():
     text_chunks = get_text_chunks(raw_text)
     vectorstore = get_vectorstore(text_chunks)
     set_global_vectorstore(vectorstore)
-    st.session_state.agent = initialize_chatbot_agent()
+    st.session_state.chat_ready = True
 
 def convert_pdf_to_xml(pdf_file, xml_path):
     os.makedirs("temp", exist_ok=True)
@@ -251,6 +251,7 @@ def handle_user_query(query):
     else:
         context = faiss_search_with_keywords(query)
     return self_reasoning(query, context)
+    
 def main():
     load_environment()
 
@@ -273,7 +274,8 @@ def main():
     user_question = st.text_input("Ask your question:")
 
     if st.session_state.agent and user_question:
-        response = handle_user_query(user_question, st.session_state.agent)
+        set_global_vectorstore(st.session_state.vectorstore)
+        response = handle_user_query(user_question)
         st.session_state.chat_history.append({"user": user_question, "bot": response})
 
     for chat in st.session_state.chat_history: