Update app.py

app.py

@@ -41,8 +41,8 @@ def generate_random_string(length):
 #if "pdf_files" not in st.session_state:
     #st.session_state.pdf_files = None
 
-#if "documents" not in st.session_state:
-#    st.session_state.documents = None
+if "documents" not in st.session_state:
+    st.session_state.documents = None
 
 if "query_engine" not in st.session_state:
     st.session_state.query_engine = None
@@ -67,13 +67,15 @@ with st.sidebar:
         try:
             start_1 = timeit.default_timer() # Start timer
             st.write(f"QA文档加载开始：{start_1}")
-            #st.session_state.documents = SimpleDirectoryReader(uploadedfile_path).load_data()
-            documents = SimpleDirectoryReader(uploadedfile_path).load_data()
+            st.session_state.documents = SimpleDirectoryReader(uploadedfile_path).load_data()
+            #documents = SimpleDirectoryReader(uploadedfile_path).load_data()
             end_1 = timeit.default_timer() # Start timer
             st.write(f"QA文档加载结束：{end_1}")
             st.write(f"QA文档加载耗时：{end_1 - start_1}")
         except Exception as e:
             print("文档加载出现问题/Waiting for path creation.")
+            st.warning("文档加载出现问题/Waiting for path creation.")
+            st.stop()                 
         start_2 = timeit.default_timer() # Start timer
         st.write(f"向量模型加载开始：{start_2}")
         if "embed_model" not in st.session_state: