Upload 2 files

app.py

@@ -9,6 +9,14 @@ from doc_process import doc_process
 
 extract_folder = 'documents'
 
+# 压缩目录到zip文件
+def compress_directory(directory_path, output_zip):
+    try:
+        shutil.make_archive(output_zip, 'zip', directory_path)
+        return True
+    except Exception as e:
+        return str(e)
+
 def recode(raw: str) -> str:
     '''
     编码修正
@@ -56,9 +64,7 @@ def zip_extract_all(src_zip_file: ZipFile, target_path: str) -> None:
                 # 这里基于Zipfile.open()提取文件内容时需要使用原始的乱码文件名
                 shutil.copyfileobj(src_zip_file.open(file_or_path), z)
 
-st.title('起诉书 & 委托书 - 自动处理程序')
-
-st.markdown('#### 请上传ZIP压缩文件：')
+st.header('起诉书 & 委托书 - 自动处理程序')
 
 # 添加一个文件上传组件
 uploaded_file = st.file_uploader("选择要上传的文件", type=["zip"])
@@ -86,8 +92,19 @@ if uploaded_file:
     # 删除临时文件
     os.remove("temp.zip")
 
-if st.button('自动处理并压缩成文件'):
+if st.button('自动处理并生成ZIP文件'):
     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)
-    st.markdown(result)
+    st.markdown(result)
+
+    # 在Streamlit中压缩目录
+    compressed_file_name = "output"
+    if compress_directory(os.path.join(extract_folder, 'output'), compressed_file_name):
+        st.success("导出目录已成功压缩为ZIP文件")
+
+        # 创建下载链接
+        with open(f"{compressed_file_name}.zip", "rb") as file:
+            st.download_button("点击此处下载压缩文件", file.read(), f"{compressed_file_name}.zip")
+    else:
+        st.error("目录压缩失败。")

doc_process.py

@@ -51,6 +51,7 @@ def doc_process(input_path, output_path):
     num_shex = 0
     num_fjzy = 0
     num_hnsx = 0
+    num_other_lawyer = 0
 
     # 改管辖法院，并存到对应融担公司的目录中
     for qsz_file in qsz_file_list:
@@ -110,6 +111,7 @@ def doc_process(input_path, output_path):
 
         # 替换律师和电话
         if lawyer != '王磊':
+            num_other_lawyer += 1
             for p in doc.paragraphs:
                 if '王磊' in p.text and user not in p.text:
                     text_new_name = p.text.replace('王磊', '张立人')
@@ -172,7 +174,8 @@ def doc_process(input_path, output_path):
     
     total_have = len(qsz_file_list)
     total_done = num_shex+num_fjzy+num_hnsx
-    result = f'共 {total_have} 条\n完成 {total_done} 条\n\n上海耳序：共 {num_shex} 条\n\n福建智云：共 {num_fjzy} 条\n\n海南申信：共 {num_hnsx} 条\n\n所有文档已完成自动编辑！'
+    result = f'共 {total_have} 条\n完成 {total_done} 条\n\n上海耳序：共 {num_shex} 条 | 福建智云：共 {num_fjzy} 条 | 海南申信：共 {num_hnsx} 条\n\n \
+需要修改律师的委托书数量：共 {num_other_lawyer} 条\n\n所有文档已完成自动编辑！'
     
     return result