Update app.py

app.py

@@ -11,7 +11,6 @@ from pathlib import Path
 from time import sleep
 import random
 import string
-import time
 
 import os
 from dotenv import load_dotenv
@@ -72,28 +71,32 @@ if "user_question " not in st.session_state:
 
 with st.sidebar:
     st.subheader("Upload your Documents Here: ")
-    while not st.session_state.pdf_files:
-        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:
-            st.warning("请上传文档文件")
-            time.sleep(1)  # 等待1秒后再次检查上传文件
-    # 上传文件后的处理代码
-    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:
-            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}")
-            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.")
+    #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)
+    pdf_files = st.file_uploader("Choose your PDF Files and Press OK", type=['pdf'], accept_multiple_files=True)    
+    #if st.session_state.pdf_files:
+    if not pdf_files:
+        st.warning("请上传文档文件")
+        st.stop()      
+    else:
+        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}")
+                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()