Update app.py

app.py

@@ -48,30 +48,30 @@ if "directory_path" not in st.session_state:
 with st.sidebar:
     st.subheader("Upload your Documents Here: ")            
     if "pdf_files" not in st.session_state:
-        st.session_state.pdf_files = st.file_uploader("Choose your PDF Files and Press OK", type=['pdf'], accept_multiple_files=True)      
-    if not st.session_state.pdf_files:   #如果没有上传文件，则程序停止执行，就不会出现documents为空的错误情况
-        st.warning("请上传文档文件")
-        st.stop()      
-    else:   #如果已经上传文件，则装载文件SimpleDirectoryReader.load_data()
-        st.session_state.pdf_files=pdf_files
-        if not os.path.exists(st.session_state.directory_path):
-            os.makedirs(st.session_state.directory_path)
-            for pdf_file in st.session_state.pdf_files:
-            #for pdf_file in pdf_files:
-                file_path = os.path.join(st.session_state.directory_path, pdf_file.name)
-                with open(file_path, 'wb') as f:
-                    f.write(pdf_file.read())
-                st.success(f"File '{pdf_file.name}' saved successfully.")
-            try:
-                start_1 = timeit.default_timer() # Start timer
-                st.write(f"QA文档加载开始：{start_1}")
-                if "documents" not in st.session_state:
-                    st.session_state.documents = SimpleDirectoryReader(st.session_state.directory_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.session_state.pdf_files = st.file_uploader("Choose your PDF Files and Press OK", type=['pdf'], accept_multiple_files=True)   
+        if not st.session_state.pdf_files:   #如果没有上传文件，则程序停止执行，就不会出现documents为空的错误情况、
+            st.warning("请上传文档文件")
+            st.stop()      
+        else:   #如果已经上传文件，则装载文件SimpleDirectoryReader.load_data()
+            st.session_state.pdf_files=pdf_files
+            if not os.path.exists(st.session_state.directory_path):
+                os.makedirs(st.session_state.directory_path)
+                for pdf_file in st.session_state.pdf_files:
+                    #for pdf_file in pdf_files:
+                    file_path = os.path.join(st.session_state.directory_path, pdf_file.name)
+                    with open(file_path, 'wb') as f:
+                        f.write(pdf_file.read())
+                    st.success(f"File '{pdf_file.name}' saved successfully.")
+                try:
+                    start_1 = timeit.default_timer() # Start timer
+                    st.write(f"QA文档加载开始：{start_1}")
+                    if "documents" not in st.session_state:
+                        st.session_state.documents = SimpleDirectoryReader(st.session_state.directory_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.")  
 
 # Load documents from a directory
 #documents = SimpleDirectoryReader('data').load_data()