Upload 2 files

app.py

@@ -8,6 +8,7 @@ from doc_process import doc_process
 
 
 extract_folder = 'documents'
+compressed_file_name = "output"
 input_folder = None
 
 # 压缩目录到zip文件
@@ -105,7 +106,6 @@ if st.button('自动处理并生成ZIP文件'):
 
     if input_folder:
         # 在Streamlit中压缩目录
-        compressed_file_name = "output"
         if compress_directory(os.path.join(extract_folder, 'output'), compressed_file_name):
             st.success("导出目录已成功压缩为ZIP文件")
 
@@ -114,14 +114,24 @@ if st.button('自动处理并生成ZIP文件'):
                 st.download_button("点击此处下载ZIP文件", file.read(), f"{compressed_file_name}.zip")
         else:
             st.error("目录压缩失败。")
+        
+        if 'output.zip' in os.listdir():
+            st.write("yes")
 
 if st.button('清空输出文档', type='primary'):
     try:
         shutil.rmtree(extract_folder)
     except Exception as e:
-        st.write(f"删除文件夹 {extract_folder} 时发生错误：{str(e)}")
+        # st.write(f"删除文件夹 {extract_folder} 时发生错误：{str(e)}")
+        pass
+    
+    try:
+        os.remove(f"{compressed_file_name}.zip")
+    except Exception as e:
+        # st.write(f"删除文件 f'{compressed_file_name}.zip' 时发生错误：{str(e)}")
+        pass
 
-if extract_folder in os.listdir():
-    st.markdown('#### :red[完成任务后请点击“清空输出文档”]')
+if extract_folder in os.listdir() or f"{compressed_file_name}.zip" in os.listdir():
+    st.markdown(':red[完成任务后请点击“清空输出文档”]')
 else:
-    st.markdown('#### 所有输出文档已清空')
+    st.markdown('所有输出文档已清空')