Update app.py

app.py

@@ -1,92 +1,102 @@
 import gradio as gr
 import zipfile
+import py7zr
 import os
 import tempfile
 import shutil
 from datetime import datetime
 
-def process_zips(uploaded_files, position, options, custom_text):
+def process_archives(uploaded_files, position, options, custom_text):
     if not uploaded_files:
         return []
 
     processed_files = []
     temp_dir = tempfile.mkdtemp()
     
-    # 取得當前時間，確保同一批處理的檔案時間戳記一致
     current_time_str = datetime.now().strftime("%Y%m%d_%H%M%S")
 
     for file_path in uploaded_files:
         filename = os.path.basename(file_path)
+        ext = os.path.splitext(filename)[1].lower()
         
-        if not filename.lower().endswith('.zip'):
+        # 確保是 zip 或 7z 檔
+        if ext not in ['.zip', '.7z']:
             continue
 
-        zip_basename = os.path.splitext(filename)[0]
-        extract_dir = os.path.join(temp_dir, zip_basename)
+        archive_basename = os.path.splitext(filename)[0]
+        extract_dir = os.path.join(temp_dir, archive_basename)
         os.makedirs(extract_dir, exist_ok=True)
 
+        # 根據副檔名選擇解壓縮方式
         try:
-            with zipfile.ZipFile(file_path, 'r') as zip_ref:
-                zip_ref.extractall(extract_dir)
-        except zipfile.BadZipFile:
-            print(f"無法讀取壓縮檔: {filename}")
+            if ext == '.zip':
+                with zipfile.ZipFile(file_path, 'r') as zip_ref:
+                    zip_ref.extractall(extract_dir)
+            elif ext == '.7z':
+                with py7zr.SevenZipFile(file_path, mode='r') as z_ref:
+                    z_ref.extractall(path=extract_dir)
+        except Exception as e:
+            print(f"無法讀取壓縮檔 {filename}: {e}")
             continue
 
-        renamed_dir = os.path.join(temp_dir, f"{zip_basename}_renamed")
+        renamed_dir = os.path.join(temp_dir, f"{archive_basename}_renamed")
         os.makedirs(renamed_dir, exist_ok=True)
 
-        # 根據使用者的勾選，組合要增加的文字
+        # 組合要增加的文字
         components = []
         if "壓縮檔名" in options:
-            components.append(zip_basename)
+            components.append(archive_basename)
         if "日期時間" in options:
             components.append(current_time_str)
         if "自訂文字" in options and custom_text.strip():
             components.append(custom_text.strip())
         
-        # 將選取的元素用底線連接 (例如: zipname_20231025_120000_mytext)
         add_str = "_".join(components)
 
+        # 重新命名檔案並攤平放到新資料夾
         for root, dirs, files in os.walk(extract_dir):
             for f in files:
                 old_path = os.path.join(root, f)
                 
-                # 如果使用者什麼都沒勾，或是自訂文字留白，就不做更改
                 if not add_str:
                     new_filename = f
                 else:
                     if position == "前綴":
-                        # 加在最前面: 附加文字_原檔名
                         new_filename = f"{add_str}_{f}"
                     else: 
-                        # 後綴 (加在副檔名之前): 原檔名_附加文字.副檔名
-                        name, ext = os.path.splitext(f)
-                        new_filename = f"{name}_{add_str}{ext}"
+                        name, file_ext = os.path.splitext(f)
+                        new_filename = f"{name}_{add_str}{file_ext}"
                         
                 new_path = os.path.join(renamed_dir, new_filename)
                 shutil.copy2(old_path, new_path)
 
-        # 將重新命名後的檔案打包成新的 ZIP 檔
-        output_zip_name = f"processed_{filename}"
-        output_zip_path = os.path.join(temp_dir, output_zip_name)
+        # 將重新命名後的檔案打包成原本的格式
+        output_archive_name = f"processed_{filename}"
+        output_archive_path = os.path.join(temp_dir, output_archive_name)
         
-        with zipfile.ZipFile(output_zip_path, 'w', zipfile.ZIP_DEFLATED) as out_zip:
-            for root, dirs, files in os.walk(renamed_dir):
-                for f in files:
-                    file_to_zip = os.path.join(root, f)
-                    out_zip.write(file_to_zip, arcname=f)
+        if ext == '.zip':
+            with zipfile.ZipFile(output_archive_path, 'w', zipfile.ZIP_DEFLATED) as out_zip:
+                for root, dirs, files in os.walk(renamed_dir):
+                    for f in files:
+                        file_to_zip = os.path.join(root, f)
+                        out_zip.write(file_to_zip, arcname=f)
+        elif ext == '.7z':
+            with py7zr.SevenZipFile(output_archive_path, 'w') as out_7z:
+                for root, dirs, files in os.walk(renamed_dir):
+                    for f in files:
+                        file_to_7z = os.path.join(root, f)
+                        out_7z.write(file_to_7z, arcname=f)
 
-        processed_files.append(output_zip_path)
+        processed_files.append(output_archive_path)
 
     return processed_files
 
 # 建立 Gradio 網頁介面
 with gr.Blocks(title="壓縮檔內容批次重新命名工具", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 📦 壓縮檔內容批次重新命名工具 (進階版)")
-    gr.Markdown("上傳 ZIP 檔案後，自訂你想增加的文字與位置。系統處理完畢後會提供新的 ZIP 檔下載。")
+    gr.Markdown("## 📦 壓縮檔內容批次重新命名工具 (支援 ZIP 與 7Z)")
+    gr.Markdown("上傳 `.zip` 或 `.7z` 檔案後，自訂你想增加的文字與位置。系統處理完畢後會提供相同格式的新壓縮檔下載。")
     
     with gr.Row():
-        # 左側：設定區
         with gr.Column(scale=1):
             gr.Markdown("### ⚙️ 命名規則設定")
             position_radio = gr.Radio(
@@ -109,16 +119,15 @@ with gr.Blocks(title="壓縮檔內容批次重新命名工具", theme=gr.themes.
                 info="需在上方勾選「自訂文字」才會生效"
             )
 
-        # 右側：檔案上傳區
         with gr.Column(scale=1):
             gr.Markdown("### 📂 檔案上傳與下載")
-            file_input = gr.File(label="上傳壓縮檔 (可多選)", file_count="multiple", type="filepath", file_types=[".zip"])
+            # 更新 file_types，讓介面允許選擇 .7z 檔案
+            file_input = gr.File(label="上傳壓縮檔 (可多選 ZIP 或 7Z)", file_count="multiple", type="filepath", file_types=[".zip", ".7z"])
             submit_btn = gr.Button("🚀 開始處理", variant="primary")
             file_output = gr.File(label="下載處理後的新壓縮檔")
 
-    # 綁定按鈕點擊事件，傳入所有設定值
     submit_btn.click(
-        fn=process_zips, 
+        fn=process_archives, 
         inputs=[file_input, position_radio, options_checkbox, custom_text_input], 
         outputs=file_output
     )