Upload app.py

app.py

@@ -18,6 +18,7 @@ Requirements (requirements.txt):
 import os
 import re
 import tempfile
+import zipfile
 from pathlib import Path
 
 import gradio as gr
@@ -102,6 +103,23 @@ def extract_files(text: str, tag: str):
     return files, html_source
 
 
+def zip_files(file_paths, tag: str) -> str:
+    """Bundle several generated files into a single .zip for one-click download.
+
+    Files are stored under their original name (the shared `tag_` prefix used
+    on disk to avoid collisions between turns is stripped inside the zip).
+    """
+    zip_path = OUT_DIR / f"{tag}_bundle.zip"
+    prefix = f"{tag}_"
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+        for fp in file_paths:
+            arcname = Path(fp).name
+            if arcname.startswith(prefix):
+                arcname = arcname[len(prefix):]
+            zf.write(fp, arcname=arcname)
+    return str(zip_path)
+
+
 def build_client(model: str) -> InferenceClient:
     kwargs = {"provider": "auto"}
     if HF_TOKEN:
@@ -166,10 +184,20 @@ def respond(message, chat_history, model, system_prompt, temperature, max_tokens
 
         tag = f"turn{len(chat_history)}"
         files, html_source = extract_files(partial, tag)
+
+        if len(files) > 1:
+            # Multiple files: bundle them so it's one click, not one download
+            # per file.
+            download_value = zip_files(files, tag)
+        elif files:
+            download_value = files[0]
+        else:
+            download_value = None
+
         yield (
             chat_history,
             "",
-            gr.update(value=files if files else None, visible=bool(files)),
+            gr.update(value=download_value, visible=bool(download_value)),
             gr.update(value=make_preview_html(html_source)),
         )
 
@@ -230,7 +258,10 @@ with gr.Blocks(title="Chatterbox", theme=gr.themes.Soft()) as demo:
 
         with gr.Column(scale=2):
             with gr.Tab("📄 Файлы"):
-                files_out = gr.File(label="Сгенерированные файлы", file_count="multiple")
+                files_out = gr.File(
+                    label="Сгенерированные файлы (несколько — упаковываются в .zip)",
+                    file_count="single",
+                )
             with gr.Tab("👁 Превью (HTML)"):
                 preview = gr.HTML(value=make_preview_html(None))