Update app.py

app.py

@@ -85,6 +85,17 @@ def load_environment():
     load_dotenv()
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
+def preload_modtran_document():
+    with open("MODTRAN 6 User's Manual.pdf", "rb") as f:
+        file_obj = io.BytesIO(f.read())
+        file_obj.name = "MODTRAN 6 User's Manual.pdf"
+        uploaded_files = [file_obj]
+
+    raw_text = get_uploaded_text(uploaded_files)
+    text_chunks = get_text_chunks(raw_text)
+    vectorstore = get_vectorstore(text_chunks)
+    set_global_vectorstore(vectorstore)
+    st.session_state.agent = initialize_chatbot_agent()
 
 def convert_pdf_to_xml(pdf_file, xml_path):
     os.makedirs("temp", exist_ok=True)
@@ -243,31 +254,26 @@ def handle_user_query(query):
 def main():
     load_environment()
 
+    if "agent" not in st.session_state:
+        st.session_state.agent = None
     if "chat_ready" not in st.session_state:
         st.session_state.chat_ready = False
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
-    if "vectorstore" not in st.session_state:
-        st.session_state.vectorstore = None
-
-    st.header("Chat with MODTRAN Documents :satellite:")
-    user_question = st.text_input("Ask a question about your uploaded files:")
-
-    with st.sidebar:
-        uploaded_files = st.file_uploader("Upload PDF, HTML, or MODTRAN output files:", accept_multiple_files=True)
-        if st.button("Process") and uploaded_files:
-            with st.spinner("Processing..."):
-                raw_text = get_uploaded_text(uploaded_files)
-                text_chunks = get_text_chunks(raw_text)
-                st.session_state.vectorstore = get_vectorstore(text_chunks)  
-                set_global_vectorstore(st.session_state.vectorstore)         
-                st.session_state.chat_ready = True
-                st.success("Files processed successfully!")
-
-    if st.session_state.chat_ready and user_question:
-        # Restore the global vectorstore reference
-        set_global_vectorstore(st.session_state.vectorstore)
-        response = handle_user_query(user_question)
+
+    st.header("Chat with MODTRAN Documents 📄🤖")
+
+    # Preload the document once when app starts
+    if not st.session_state.chat_ready:
+        with st.spinner("Loading MODTRAN document..."):
+            preload_modtran_document()
+            st.session_state.chat_ready = True
+            st.success("MODTRAN User Manual loaded successfully!")
+
+    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)
         st.session_state.chat_history.append({"user": user_question, "bot": response})
 
     for chat in st.session_state.chat_history: