Upd safe batcher and btn for download file

app.py

@@ -144,7 +144,8 @@ def transcribe(inputs, task, summarize=False):
                     running_summary += ("\n\n" if running_summary else "") + summarize_with_gemini(partial)
                 except Exception:
                     pass
-            yield running_text, (running_summary if summarize else "")
+            # Files are not ready during streaming
+            yield running_text, (running_summary if summarize else ""), None, None
 
     try:
         if isinstance(inputs, str):
@@ -165,8 +166,18 @@ def transcribe(inputs, task, summarize=False):
                     summ = summarize_with_gemini(out)
                 except Exception as e:
                     summ = f"Summary error: {e}"
-                return out, summ
-            return out, ""
+                # Create files for download
+                tf = tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False)
+                tf.write(out)
+                tf.close()
+                sf = tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False)
+                sf.write(summ)
+                sf.close()
+                return out, summ, tf.name, sf.name
+            tf = tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False)
+            tf.write(out)
+            tf.close()
+            return out, "", tf.name, None
     except Exception as e:
         raise gr.Error(f"Transcription failed: {e}")
 
@@ -233,7 +244,17 @@ def yt_transcribe(yt_url, task, summarize=False, max_filesize=75.0):
             summary = summarize_with_gemini(text)
         except Exception as e:
             summary = f"Summary error: {e}"
-    return html_embed_str, text, summary
+    # Create download files
+    tf = tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False)
+    tf.write(text)
+    tf.close()
+    sf_path = None
+    if summary:
+        sf = tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False)
+        sf.write(summary)
+        sf.close()
+        sf_path = sf.name
+    return html_embed_str, text, summary, tf.name, sf_path
 
 
 demo = gr.Blocks()
@@ -245,7 +266,12 @@ mf_transcribe = gr.Interface(
         gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
         gr.Checkbox(label="Summarize with Gemini", value=False),
     ],
-    outputs=[gr.Textbox(label="Transcription"), gr.Textbox(label="Summary")],
+    outputs=[
+        gr.Textbox(label="Transcription"),
+        gr.Textbox(label="Summary"),
+        gr.File(label="Download transcription (transcribe.txt)"),
+        gr.File(label="Download summary (summarise.txt)")
+    ],
     title="Whisper Large V3: Microphone",
     description=(
         "Transcribe long-form microphone or audio inputs."
@@ -260,7 +286,12 @@ file_transcribe = gr.Interface(
         gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
         gr.Checkbox(label="Summarize with Gemini", value=False),
     ],
-    outputs=[gr.Textbox(label="Transcription"), gr.Textbox(label="Summary")],
+    outputs=[
+        gr.Textbox(label="Transcription"),
+        gr.Textbox(label="Summary"),
+        gr.File(label="Download transcription (transcribe.txt)"),
+        gr.File(label="Download summary (summarise.txt)")
+    ],
     title="Whisper Large V3: Audio file",
     description=(
         "Transcribe long-form microphone or audio inputs."
@@ -275,7 +306,13 @@ yt_transcribe = gr.Interface(
         gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
         gr.Checkbox(label="Summarize with Gemini", value=False),
     ],
-    outputs=["html", gr.Textbox(label="Transcription"), gr.Textbox(label="Summary")],
+    outputs=[
+        "html",
+        gr.Textbox(label="Transcription"),
+        gr.Textbox(label="Summary"),
+        gr.File(label="Download transcription (transcribe.txt)"),
+        gr.File(label="Download summary (summarise.txt)")
+    ],
     title="Whisper Large V3: Transcribe YouTube",
     description=(
         "Transcribe long-form YouTube videos."