Update app.py

app.py

@@ -67,45 +67,45 @@ def main():
 
 
     if apikey:
-        if "tools_loaded" not in st.session_state:
-            try:
-                    directory = '/home/user/app/rag_docs_final_review_tex_merged'
-                    tex_files = find_tex_files(directory)
-                    
-                with st.spinner('Please wait, AMGPT is loading....'):  
-                    paper_to_tools_dict = {}
-                    for paper in tex_files:
-                        path = Path(paper)
-                        vector_tool = create_doc_tools(doc_name=path.stem, document_fp=path)
-                        paper_to_tools_dict[path.stem] = [vector_tool]
-            
-                    initial_tools = [t for paper in tex_files for t in paper_to_tools_dict[Path(paper).stem]]
-            
-                    obj_index = ObjectIndex.from_objects(
-                        initial_tools,
-                        index_cls=VectorStoreIndex,
-                    )
-            
-                    obj_retriever = obj_index.as_retriever(similarity_top_k=6)
-            
-            
-                    context = """You are an agent designed to answer scientific queries over a set of given documents.
-                    Please always use the tools provided to answer a question. Do not rely on prior knowledge.
-                    """
+    if "tools_loaded" not in st.session_state:
+        try:
+            directory = '/home/user/app/rag_docs_final_review_tex_merged'
+            tex_files = find_tex_files(directory)
             
-                    agent = ReActAgent.from_tools(
-                        tool_retriever=obj_retriever,
-                        llm=llm,
-                        verbose=True,
-                        context=context
-                    )
-                st.success('Done!, you may start asking questions now')
-
-                # store session state variables
-                st.session_state["tools_loaded"] = True
-                st.session_state["agent"] = agent
-            except Exception as e:
-                st.error(e)
+            with st.spinner('Please wait, AMGPT is loading....'):  
+                paper_to_tools_dict = {}
+                for paper in tex_files:
+                    path = Path(paper)
+                    vector_tool = create_doc_tools(doc_name=path.stem, document_fp=path)
+                    paper_to_tools_dict[path.stem] = [vector_tool]
+        
+                initial_tools = [t for paper in tex_files for t in paper_to_tools_dict[Path(paper).stem]]
+        
+                obj_index = ObjectIndex.from_objects(
+                    initial_tools,
+                    index_cls=VectorStoreIndex,
+                )
+        
+                obj_retriever = obj_index.as_retriever(similarity_top_k=6)
+        
+                context = """You are an agent designed to answer scientific queries over a set of given documents.
+                Please always use the tools provided to answer a question. Do not rely on prior knowledge.
+                """
+        
+                agent = ReActAgent.from_tools(
+                    tool_retriever=obj_retriever,
+                    llm=llm,
+                    verbose=True,
+                    context=context
+                )
+            st.success('Done!, you may start asking questions now')
+
+            # store session state variables
+            st.session_state["tools_loaded"] = True
+            st.session_state["agent"] = agent
+        except Exception as e:
+            st.error(e)
+