Update app.py

app.py

@@ -484,45 +484,34 @@ def transcribe_multiple(audio_files, model_name, advanced_options, merge_checkbo
 
 
 
-# ----------------------- Gradio UI -----------------------
+# Defensive wrapper to surface exceptions into the Logs textbox
 def run_transcription_wrapper(files, model_name, merge, zip_file, zip_password, enable_memory, advanced_options_state):
-    audio_input = files
-    zip_path = None
-    if zip_file:
-        if isinstance(zip_file, (str, os.PathLike)):
-            zip_path = str(zip_file)
-        elif hasattr(zip_file, "name"):
-            zip_path = zip_file.name
-        elif isinstance(zip_file, dict) and zip_file.get("name"):
-            zip_path = zip_file["name"]
-    adv = {}
-    return transcribe_multiple(audio_input, model_name, adv, merge_checkbox=merge, zip_file=zip_path, zip_password=zip_password, enable_memory=enable_memory)
-
-print("DEBUG: building Gradio Blocks", flush=True)
-demo = gr.Blocks()
-
-with demo:
-    gr.Markdown("## Whisper Transcription (Spaces-ready)")
-    with gr.Row():
-        with gr.Column(scale=2):
-            file_input = gr.File(label="Upload audio files (or zip)", file_count="multiple", type="filepath")
-            zip_input = gr.File(label="Optional: Upload zip file containing audio", file_count="single", type="filepath")
-            zip_password = gr.Textbox(label="Zip password (if any)", placeholder="password (optional)")
-            model_select = gr.Dropdown(choices=["small","medium","large","base"], value="small", label="Whisper model")
-            merge_checkbox = gr.Checkbox(label="Merge transcripts to a single .docx (downloadable)", value=True)
-            memory_checkbox = gr.Checkbox(label="Enable persistent memory (word/phrase correction)", value=False)
-            submit = gr.Button("Transcribe")
-        with gr.Column(scale=3):
-            logs = gr.Textbox(label="Logs (streaming)", lines=12)
-            transcripts_out = gr.Textbox(label="Transcripts (streaming)", lines=12)
-            download_file = gr.File(label="Merged .docx (when enabled)")
-            progress_num = gr.Number(value=0, label="Progress (%)")
-
-    submit.click(
-        fn=run_transcription_wrapper,
-        inputs=[file_input, model_select, merge_checkbox, zip_input, zip_password, memory_checkbox, gr.State({})],
-        outputs=[logs, transcripts_out, download_file, progress_num],
-    )
+    import traceback, io
+    try:
+        audio_input = files
+        zip_path = None
+        if zip_file:
+            if isinstance(zip_file, (str, os.PathLike)):
+                zip_path = str(zip_file)
+            elif hasattr(zip_file, "name"):
+                zip_path = zip_file.name
+            elif isinstance(zip_file, dict) and zip_file.get("name"):
+                zip_path = zip_file["name"]
+        adv = {}
+        # return the generator directly (transcribe_multiple yields tuples)
+        return transcribe_multiple(audio_input, model_name, adv, merge_checkbox=merge, zip_file=zip_path, zip_password=zip_password, enable_memory=enable_memory)
+    except Exception as e:
+        # If anything raises before generator returned, produce a generator that yields the traceback
+        buf = io.StringIO()
+        traceback.print_exc(file=buf)
+        tb = buf.getvalue()
+        logs = f"EXCEPTION in run_transcription_wrapper:\n{tb}"
+        transcripts = "ERROR: transcription did not start due to exception."
+        # Yield once with logs and final 100% to stop spinner
+        def error_gen():
+            yield logs, transcripts, None, 100
+        return error_gen()
+
 
 # Launch
 if __name__ == "__main__":