Update app.py

app.py

@@ -12,34 +12,21 @@ _last_segments = []
 def process_audio(audio_path, num_speakers):
     global _last_segments
     if audio_path is None:
-        return "⚠️ Please upload an audio file first.", "", ""
+        return "⚠️ Please upload an audio file first.", ""
     try:
         pipeline = get_pipeline()
         n = int(num_speakers) if num_speakers and int(num_speakers) > 0 else 0
         segments = pipeline.process(audio_path, num_speakers=n)
     except Exception as e:
-        return f"❌ Error: {e}\n{traceback.format_exc()}", "", ""
+        return f"❌ Error: {e}\n{traceback.format_exc()}", ""
 
     if not segments:
-        return "⚠️ No speech detected.", "", ""
+        return "⚠️ No speech detected.", ""
 
     _last_segments = segments
     abab = to_abab_text(segments)
-    rows = "".join(
-        f"<tr style='background:{'#f0f4ff' if i%2==0 else 'white'}'>"
-        f"<td style='padding:6px 8px;font-weight:bold;color:#1e3a5f'>{s.speaker}</td>"
-        f"<td style='padding:6px 8px'>{s.start_fmt}</td>"
-        f"<td style='padding:6px 8px'>{s.end_fmt}</td>"
-        f"<td style='padding:6px 8px'>{s.text}</td></tr>"
-        for i, s in enumerate(segments)
-    )
-    table = f"""<table style='width:100%;border-collapse:collapse;font-size:14px'>
-      <thead style='background:#1e3a5f;color:white'>
-        <tr><th style='padding:8px'>Speaker</th><th>Start</th><th>End</th>
-        <th style='text-align:left;padding:8px'>Transcript</th></tr>
-      </thead><tbody>{rows}</tbody></table>"""
     unique = len(set(s.speaker for s in segments))
-    return f"✅ Done — {len(segments)} segments, {unique} speaker(s) detected", abab, table
+    return f"✅ Done — {len(segments)} segments, {unique} speaker(s) detected", abab
 
 
 def export_json():
@@ -61,36 +48,23 @@ with gr.Blocks(title="Speech Annotation Pipeline") as demo:
     gr.Markdown("# 🎙️ Speech Annotation Pipeline")
     gr.Markdown("Upload audio → get speaker-labeled transcripts automatically")
 
+    audio_input = gr.Audio(label="Upload Audio (.wav/.mp3/.flac)", type="filepath")
+    num_speakers = gr.Slider(minimum=0, maximum=10, step=1, value=0,
+                             label="Number of speakers (0 = auto-detect)")
+    run_btn = gr.Button("▶ Run Annotation", variant="primary")
+    status_box = gr.Textbox(label="Status", interactive=False)
+    abab_box = gr.Textbox(label="Editable Transcript (ABAB)", lines=20,
+                          placeholder="Annotations will appear here...")
     with gr.Row():
-        with gr.Column(scale=1):
-            audio_input = gr.Audio(
-                label="Upload Audio (.wav/.mp3/.flac)",
-                type="filepath"
-            )
-            num_speakers = gr.Slider(
-                minimum=0, maximum=10, step=1, value=0,
-                label="Number of speakers (0 = auto-detect)"
-            )
-            run_btn = gr.Button("▶ Run Annotation", variant="primary")
-            status_box = gr.Textbox(label="Status", interactive=False)
-            with gr.Row():
-                json_btn = gr.Button("Export JSON")
-                csv_btn  = gr.Button("Export CSV")
-            json_file = gr.File(label="JSON Download")
-            csv_file  = gr.File(label="CSV Download")
-
-        with gr.Column(scale=2):
-            abab_box = gr.Textbox(
-                label="Editable Transcript (ABAB)",
-                lines=20,
-                placeholder="Annotations will appear here..."
-            )
-            table_html = gr.HTML()
+        json_btn = gr.Button("Export JSON")
+        csv_btn  = gr.Button("Export CSV")
+    json_file = gr.File(label="JSON Download")
+    csv_file  = gr.File(label="CSV Download")
 
     run_btn.click(
         fn=process_audio,
         inputs=[audio_input, num_speakers],
-        outputs=[status_box, abab_box, table_html]
+        outputs=[status_box, abab_box]
     )
     json_btn.click(fn=export_json, inputs=[], outputs=[json_file])
     csv_btn.click(fn=export_csv, inputs=[], outputs=[csv_file])