Update app.py

app.py

@@ -9,18 +9,14 @@ UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
 
 _last_segments = []
 
-def process_audio(audio_path, num_speakers, min_speakers, max_speakers):
+def process_audio(audio_path, num_speakers):
     global _last_segments
     if audio_path is None:
         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
-        mn = int(min_speakers) if min_speakers and int(min_speakers) > 0 else None
-        mx = int(max_speakers) if max_speakers and int(max_speakers) > 0 else None
-        segments = pipeline.process(audio_path, num_speakers=n,
-                                    min_speakers=mn, max_speakers=mx)
+        segments = pipeline.process(audio_path, num_speakers=n)
     except Exception as e:
         return f"❌ Error: {e}\n{traceback.format_exc()}", "", ""
 
@@ -29,7 +25,6 @@ def process_audio(audio_path, num_speakers, min_speakers, max_speakers):
 
     _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>"
@@ -43,7 +38,6 @@ def process_audio(audio_path, num_speakers, min_speakers, max_speakers):
         <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
 
@@ -68,32 +62,38 @@ with gr.Blocks(title="Speech Annotation Pipeline") as demo:
     gr.Markdown("Upload audio → get speaker-labeled transcripts automatically")
 
     with gr.Row():
-        with gr.Column():
-            audio_input = gr.Audio(label="Upload Audio (.wav/.mp3/.flac)", type="filepath")
-            gr.Markdown("### 🎯 Speaker Settings")
-            num_speakers = gr.Slider(minimum=0, maximum=10, step=1, value=0,
-                                     label="Number of speakers (0 = auto-detect)")
+        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")
-            csv_file  = gr.File(label="CSV")
+            json_file = gr.File(label="JSON Download")
+            csv_file  = gr.File(label="CSV Download")
 
-        with gr.Column():
-            abab_box   = gr.Textbox(label="Editable Transcript (ABAB)", lines=20,
-                                    placeholder="Annotations will appear here...")
-            table_html = gr.HTML(label="Segment Table")
+        with gr.Column(scale=2):
+            abab_box = gr.Textbox(
+                label="Editable Transcript (ABAB)",
+                lines=20,
+                placeholder="Annotations will appear here..."
+            )
+            table_html = gr.HTML()
 
     run_btn.click(
         fn=process_audio,
-        inputs=[audio_input, num_speakers,
-                gr.State(0), gr.State(0)],
+        inputs=[audio_input, num_speakers],
         outputs=[status_box, abab_box, table_html]
     )
     json_btn.click(fn=export_json, inputs=[], outputs=[json_file])
-    csv_btn.click(fn=export_csv,  inputs=[], outputs=[csv_file])
+    csv_btn.click(fn=export_csv, inputs=[], outputs=[csv_file])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)