Upload app.py

app.py

@@ -39,19 +39,14 @@ def process_document(
     progress=gr.Progress(),
 ):
     """
-    Full pipeline:
-        upload → extract → RAG → script (mode-specific) → audio
-
-    Args:
-        file          : Gradio uploaded file object (.name attribute)
-        delivery_mode : "Summary" | "Podcast" | "Song / Rap"
-        song_rap_sub  : "Song" | "Rap"  (only relevant for Song/Rap mode)
+    Full pipeline: upload → extract → RAG → script (mode-specific) → audio.
 
     Returns:
-        (script_text, audio_file_path, status_markdown)
+        (script_text, audio_file_path)
+        — no status card; technical details are logged server-side only.
     """
 
-    # ── Validate input ───────────────────────────────────────────────────────
+    # ── Validate ─────────────────────────────────────────────────────────────
     if file is None:
         raise gr.Error("Please upload a PDF or TXT file first.")
 
@@ -106,7 +101,6 @@ def process_document(
         progress(0.80, desc="🎙️ Synthesising audio…")
 
         is_podcast = delivery_mode.strip().lower() == "podcast"
-
         if is_podcast:
             audio_path, engine = generate_audio_podcast(script)
         else:
@@ -115,23 +109,7 @@ def process_document(
         logger.info("Audio generated via %s: %s", engine, audio_path)
         progress(1.00, desc="✅ Done!")
 
-        # ── Build status card ─────────────────────────────────────────────────
-        mode_icon = {"summary": "📋", "podcast": "🎙️", "song / rap": "🎵"}.get(
-            delivery_mode.lower(), "🎧"
-        )
-        status = (
-            f"### ✅ Generation complete!\n\n"
-            f"| | |\n|---|---|\n"
-            f"| {mode_icon} **Mode** | {delivery_mode}"
-            + (f" — {song_rap_sub}" if "song" in delivery_mode.lower() or "rap" in delivery_mode.lower() else "")
-            + f" |\n"
-            f"| 📄 **Document** | {os.path.basename(file_path)} |\n"
-            f"| 🧩 **Chunks** | {chunk_count} |\n"
-            f"| ✍️ **Script length** | {len(script):,} chars |\n"
-            f"| 🔊 **Voice engine** | {engine} |\n"
-        )
-
-        return script, audio_path, status
+        return script, audio_path
 
     except gr.Error:
         raise
@@ -152,16 +130,31 @@ def _mode_progress_label(mode: str, sub_mode: str) -> str:
 
 
 # ══════════════════════════════════════════════════════════════════════════════
-# Conditional UI visibility helpers
+# UI helpers
 # ══════════════════════════════════════════════════════════════════════════════
 
+def _mode_description(mode: str) -> str:
+    descriptions = {
+        "Summary": (
+            "*📋 **Summary** — A clear, structured spoken narration covering "
+            "the intro, key points, and conclusion. Single voice, neutral tone.*"
+        ),
+        "Podcast": (
+            "*🎙️ **Podcast** — A two-host conversation. Host 1 guides and "
+            "asks questions; Host 2 explains and elaborates. Dual voices.*"
+        ),
+        "Song / Rap": (
+            "*🎵 **Song / Rap** — Key ideas transformed into a rhythmic, "
+            "memorable format. Choose Song for smooth flow or Rap for punchy lines.*"
+        ),
+    }
+    return descriptions.get(mode, "")
+
+
 def _on_mode_change(mode: str):
-    """
-    Return visibility updates for mode-specific sub-controls.
-    Called whenever the delivery mode radio changes.
-    """
-    show_song_rap = "song" in mode.lower() or "rap" in mode.lower()
-    return gr.update(visible=show_song_rap)
+    """Show/hide Song/Rap sub-radio and update description text."""
+    show_sub = "song" in mode.lower() or "rap" in mode.lower()
+    return gr.update(visible=show_sub), _mode_description(mode)
 
 
 # ══════════════════════════════════════════════════════════════════════════════
@@ -171,7 +164,6 @@ def _on_mode_change(mode: str):
 def build_ui() -> gr.Blocks:
 
     css = """
-    /* ── Header ─────────────────────────────────────────────── */
     .main-header { text-align: center; margin-bottom: 1rem; }
     .main-header h1 {
         background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
@@ -183,25 +175,14 @@ def build_ui() -> gr.Blocks:
     }
     .main-header p { color: #6b7280; font-size: 1.1rem; }
 
-    /* ── Mode selector card ──────────────────────────────────── */
     .mode-card {
         background: linear-gradient(135deg, #f8f7ff 0%, #f0edff 100%);
         border: 1px solid #e0d9ff;
         border-radius: 12px;
         padding: 1rem 1.25rem;
         margin-top: 0.5rem;
+        margin-bottom: 0.75rem;
     }
-    .mode-card h3 { color: #4c3d99; margin-bottom: 0.5rem; }
-
-    /* ── Status ──────────────────────────────────────────────── */
-    .status-box {
-        border-left: 3px solid #667eea;
-        padding-left: 1rem;
-        margin: 0.5rem 0;
-    }
-
-    /* ── Sub-mode row ────────────────────────────────────────── */
-    .sub-mode-row { margin-top: 0.5rem; }
     """
 
     with gr.Blocks(
@@ -220,12 +201,9 @@ def build_ui() -> gr.Blocks:
 
         with gr.Row(equal_height=False):
 
-            # ════════════════════════════════════════════════════════════════
-            # LEFT COLUMN — Upload + Mode Selector
-            # ════════════════════════════════════════════════════════════════
+            # ── LEFT COLUMN — Upload + Mode Selector ─────────────────────────
             with gr.Column(scale=1):
 
-                # ── Upload ───────────────────────────────────────────────
                 gr.Markdown("### 📤 Upload Document")
                 file_input = gr.File(
                     label="Upload a PDF or TXT file",
@@ -233,7 +211,7 @@ def build_ui() -> gr.Blocks:
                     type="filepath",
                 )
 
-                # ── Delivery Mode Selector ───────────────────────────────
+                # Delivery Mode card
                 gr.HTML('<div class="mode-card">')
                 gr.Markdown("### 🎨 Choose Audio Experience")
 
@@ -241,42 +219,27 @@ def build_ui() -> gr.Blocks:
                     choices=["Summary", "Podcast", "Song / Rap"],
                     value="Summary",
                     label=None,
-                    elem_id="delivery-mode-radio",
                 )
 
-                # Song / Rap sub-option — hidden unless Song/Rap selected
-                with gr.Row(visible=False, elem_classes=["sub-mode-row"]) as song_rap_row:
+                # Sub-option: Song vs Rap — hidden until Song/Rap is chosen
+                with gr.Row(visible=False) as song_rap_row:
                     song_rap_sub = gr.Radio(
                         choices=["Song", "Rap"],
                         value="Rap",
                         label="Style",
-                        scale=1,
                     )
 
-                # Mode description (updates on change)
-                mode_description = gr.Markdown(
-                    value=_mode_description("Summary"),
-                    elem_id="mode-desc",
-                )
+                mode_description = gr.Markdown(value=_mode_description("Summary"))
 
-                gr.HTML("</div>")   # close .mode-card
+                gr.HTML("</div>")
 
-                # ── Generate Button ──────────────────────────────────────
                 generate_btn = gr.Button(
                     "🎙️ Generate Audio",
                     variant="primary",
                     size="lg",
                 )
 
-                # ── Status ───────────────────────────────────────────────
-                status_output = gr.Markdown(
-                    value="*Upload a document, choose your audio experience, then click Generate.*",
-                    elem_classes=["status-box"],
-                )
-
-            # ════════════════════════════════════════════════════════════════
-            # RIGHT COLUMN — Audio + Script Output
-            # ════════════════════════════════════════════════════════════════
+            # ── RIGHT COLUMN — Audio + Script Output ──────────────────────────
             with gr.Column(scale=1):
 
                 gr.Markdown("### 🎧 Generated Audio")
@@ -302,56 +265,25 @@ def build_ui() -> gr.Blocks:
             "</center>"
         )
 
-        # ════════════════════════════════════════════════════════════════════
-        # Event wiring
-        # ════════════════════════════════════════════════════════════════════
+        # ── Event wiring ─────────────────────────────────────────────────────
 
-        # Show/hide Song-Rap sub-option + update description when mode changes
         delivery_mode.change(
-            fn=_on_mode_change_full,
+            fn=_on_mode_change,
             inputs=[delivery_mode],
             outputs=[song_rap_row, mode_description],
         )
 
-        # Generate button click
+        # outputs: script first, then audio — matches return order in process_document
         generate_btn.click(
             fn=process_document,
             inputs=[file_input, delivery_mode, song_rap_sub],
-            outputs=[script_output, audio_output, status_output],
+            outputs=[script_output, audio_output],
         )
 
     return app
 
 
-# ── Mode description helper ───────────────────────────────────────────────────
-
-def _mode_description(mode: str) -> str:
-    descriptions = {
-        "Summary": (
-            "*📋 **Summary** — A clear, structured spoken narration covering "
-            "the intro, key points, and conclusion. Single voice, neutral tone.*"
-        ),
-        "Podcast": (
-            "*🎙️ **Podcast** — A two-host conversation. Host 1 guides and "
-            "asks questions; Host 2 explains and elaborates. Dual voices.*"
-        ),
-        "Song / Rap": (
-            "*🎵 **Song / Rap** — Key ideas transformed into a rhythmic, "
-            "memorable format. Choose Song for smooth flow or Rap for punchy lines.*"
-        ),
-    }
-    return descriptions.get(mode, "")
-
-
-def _on_mode_change_full(mode: str):
-    """Return (song_rap_row visibility, description markdown)."""
-    show_sub = "song" in mode.lower() or "rap" in mode.lower()
-    return gr.update(visible=show_sub), _mode_description(mode)
-
-
-# ══════════════════════════════════════════════════════════════════════════════
-# Entry point
-# ══════════════════════════════════════════════════════════════════════════════
+# ── Entry point ───────────────────────────────────────────────────────────────
 
 if __name__ == "__main__":
     logger.info("Starting VoiceVerse AI…")