Update app.py

app.py

@@ -181,9 +181,6 @@ def _friendly_error(e: Exception) -> str:
 def process(input_mode, text_input, audio_input, target_language):
     """
     Called when the user clicks Generate.
-    The first yield immediately clears stale output and wakes up Gradio's
-    streaming so all subsequent yields (including errors) render on the
-    very first click.
     """
     print("TOKEN SET:", bool(os.environ.get("SUNBIRD_API_TOKEN")))
     print(f"\n=== PROCESS CALLED ===")
@@ -192,32 +189,28 @@ def process(input_mode, text_input, audio_input, target_language):
     print(f"Audio input: {audio_input}")
     print(f"Target language: {target_language}")
 
-    # ── Wake-up yield: clears stale results, enables immediate rendering ──
-    yield ("", "", "", None, gr.update(value="", visible=False))
-
+    use_audio = (input_mode == "Audio Upload")
+    
+    # Clear previous results - but don't yield if we're about to error
+    clear_yield = ("", "", "", None, gr.update(value="", visible=False))
+    
+    # Validate inputs
+    if use_audio and not audio_input:
+        error_yield = ("", "", "", None, gr.update(value="<div id='error-box'>Please upload an audio file before running the pipeline.</div>", visible=True))
+        yield clear_yield
+        yield error_yield
+        return
+    
+    if not use_audio and not (text_input and text_input.strip()):
+        error_yield = ("", "", "", None, gr.update(value="<div id='error-box'>Please enter some text before running the pipeline.</div>", visible=True))
+        yield clear_yield
+        yield error_yield
+        return
+    
+    # Now yield the clear
+    yield clear_yield
+    
     try:
-        use_audio = (input_mode == "Audio Upload")
-
-        if use_audio and not audio_input:
-            yield (
-                "",
-                "",
-                "",
-                None,
-                gr.update(value="<div id='error-box'>Please upload an audio file before running the pipeline.</div>", visible=True),
-            )
-            return
-
-        if not use_audio and not (text_input and text_input.strip()):
-            yield (
-                "",
-                "",
-                "",
-                None,
-                gr.update(value="<div id='error-box'>Please enter some text before running the pipeline.</div>", visible=True),
-            )
-            return
-
         audio_path = audio_input if use_audio else None
         user_text = None if use_audio else text_input
 
@@ -235,6 +228,7 @@ def process(input_mode, text_input, audio_input, target_language):
         tmp.close()
         print(f"Audio saved to: {tmp.name}")
 
+        # Final yield with results
         yield (
             transcript,
             summary,
@@ -250,6 +244,8 @@ def process(input_mode, text_input, audio_input, target_language):
         traceback.print_exc()
 
         error_msg = _friendly_error(e)
+        # Multiple yields to ensure error appears
+        yield ("", "", "", None, gr.update(value="", visible=False))
         yield (
             "",
             "",
@@ -258,7 +254,6 @@ def process(input_mode, text_input, audio_input, target_language):
             gr.update(value=f"<div id='error-box'>{error_msg}</div>", visible=True),
         )
 
-
 # ── Build Gradio UI ──────────────────────────────────────────────────────────
 
 def build_ui():