Upload 2 files

app.py

@@ -8,6 +8,7 @@ from doc_process import doc_process
 
 
 extract_folder = 'documents'
+input_folder = None
 
 # 压缩目录到zip文件
 def compress_directory(directory_path, output_zip):
@@ -32,8 +33,6 @@ def zip_extract_all(src_zip_file: ZipFile, target_path: str) -> None:
 
     # 遍历压缩包内所有内容，创建所有目录
     for file_or_path in file.namelist():
-
-        print(file_or_path)
         
         # 若当前节点是文件夹
         if file_or_path.endswith('/'):
@@ -50,8 +49,6 @@ def zip_extract_all(src_zip_file: ZipFile, target_path: str) -> None:
 
     # 遍历压缩包内所有内容，解压文件
     for file_or_path in file.namelist():
-
-        print(file_or_path)
         
         # 若当前节点是文件夹
         if file_or_path.endswith('/'):
@@ -87,13 +84,12 @@ if uploaded_file:
     # 显示解压缩完成的消息
     st.success(f"ZIP文件已成功解压缩到目录 {extract_folder}")
     input_folder = os.listdir(extract_folder)[0] if os.listdir(extract_folder)[0] != 'output' else os.listdir(extract_folder)[1]
-    print(input_folder)
 
     # 删除临时文件
     os.remove("temp.zip")
 
 if st.button('自动处理并生成ZIP文件'):
-    if 'input_folder' in globals():
+    if input_folder:
         input_path = os.path.join(extract_folder, input_folder)
         output_path = os.path.join(extract_folder, 'output')
         result = doc_process(input_path=input_path, output_path=output_path)
@@ -101,7 +97,7 @@ if st.button('自动处理并生成ZIP文件'):
         result = '请先上传ZIP文件'
     st.markdown(result)
 
-    if 'input_folder' in globals():
+    if input_folder:
         # 在Streamlit中压缩目录
         compressed_file_name = "output"
         if compress_directory(os.path.join(extract_folder, 'output'), compressed_file_name):
@@ -109,13 +105,17 @@ if st.button('自动处理并生成ZIP文件'):
 
             # 创建下载链接
             with open(f"{compressed_file_name}.zip", "rb") as file:
-                st.download_button("点击此处下载压缩文件", file.read(), f"{compressed_file_name}.zip")
+                st.download_button("点击此处下载ZIP文件", file.read(), f"{compressed_file_name}.zip")
         else:
             st.error("目录压缩失败。")
 
-if st.button('清空所有文档'):
+if st.button('清空输出文档', type='primary'):
     try:
         shutil.rmtree(extract_folder)
-        st.write(f"文件夹 {extract_folder} 已成功删除。")
     except Exception as e:
-        st.write(f"删除文件夹 {extract_folder} 时发生错误：{str(e)}")
+        st.write(f"删除文件夹 {extract_folder} 时发生错误：{str(e)}")
+
+if extract_folder in os.listdir():
+    st.markdown('#### :red[完成任务后请点击“清空输出文档”]')
+else:
+    st.markdown('#### 所有输出文档已清空')