Upload app.py with huggingface_hub

app.py

@@ -1321,16 +1321,20 @@ def download_upload(filename):
 @app.route('/admin/download_all_uploads')
 @login_required
 def download_all_uploads():
-    """一鍵下載所有上傳檔案 (ZIP)。?source=PRA / DSA 可篩選分類。"""
+    """一鍵下載所有上傳檔案 (ZIP)。
+    ?source=PRA / DSA：只下載該類，檔案放在 zip 根目錄。
+    無 source（預設）：下載全部，依分類放入 PRA/ 與 DSA/ 子資料夾。
+    """
     if flask_session.get('role') != 'admin':
         return 'Forbidden', 403
     import db, zipfile
     source = (request.args.get('source') or '').upper()
     uploads = db.list_uploads()
+    name_to_source = {u['name']: u['source'] for u in uploads}
     if source in ('PRA', 'DSA'):
-        wanted = {u['name'] for u in uploads if u['source'] == source}
+        wanted = {n for n, s in name_to_source.items() if s == source}
     else:
-        wanted = {u['name'] for u in uploads}
+        wanted = set(name_to_source)
     upload_dir = db.get_upload_dir()
     files = [f for f in (list(upload_dir.glob('*.xls')) + list(upload_dir.glob('*.xlsx')) + list(upload_dir.glob('*.csv')))
              if f.name in wanted]
@@ -1340,7 +1344,11 @@ def download_all_uploads():
     buf = io.BytesIO()
     with zipfile.ZipFile(buf, 'w', zipfile.ZIP_DEFLATED) as zf:
         for f in files:
-            zf.write(str(f), f.name)
+            if source in ('PRA', 'DSA'):
+                arcname = f.name
+            else:
+                arcname = f"{name_to_source.get(f.name, 'PRA')}/{f.name}"
+            zf.write(str(f), arcname)
     buf.seek(0)
     from datetime import datetime
     tag = f'_{source.lower()}' if source in ('PRA', 'DSA') else ''