Update app.py

app.py

@@ -16,12 +16,11 @@ def process_audio(audio_path, num_speakers, min_speakers, max_speakers):
 
     try:
         pipeline = get_pipeline()
-        segments = pipeline.process(
-            audio_path,
-            num_speakers=int(num_speakers) if num_speakers else 0,
-            min_speakers=int(min_speakers) if min_speakers else None,
-            max_speakers=int(max_speakers) if max_speakers else None,
-        )
+        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)
     except Exception as e:
         return f"❌ Error: {e}\n{traceback.format_exc()}", "", ""
 
@@ -45,8 +44,8 @@ def process_audio(audio_path, num_speakers, min_speakers, max_speakers):
         <th style='text-align:left;padding:8px'>Transcript</th></tr>
       </thead><tbody>{rows}</tbody></table>"""
 
-    unique_speakers = len(set(s.speaker for s in segments))
-    return f"✅ Done — {len(segments)} segments, {unique_speakers} speaker(s) detected", abab, table
+    unique = len(set(s.speaker for s in segments))
+    return f"✅ Done — {len(segments)} segments, {unique} speaker(s) detected", abab, table
 
 
 def export_json():
@@ -70,26 +69,12 @@ with gr.Blocks(title="Speech Annotation Pipeline") as demo:
 
     with gr.Row():
         with gr.Column():
-            audio_input = gr.Audio(
-                label="Upload Audio (.wav/.mp3/.flac)",
-                type="filepath"
-            )
-
+            audio_input = gr.Audio(label="Upload Audio (.wav/.mp3/.flac)", type="filepath")
             gr.Markdown("### 🎯 Speaker Settings")
-            gr.Markdown("*If you know how many speakers, set it below for better accuracy*")
-
-            num_speakers = gr.Number(
-                label="Exact number of speakers (0 = auto-detect)",
-                value=0, precision=0
-            )
-            with gr.Row():
-                min_speakers = gr.Number(label="Min speakers", value=None, precision=0)
-                max_speakers = gr.Number(label="Max speakers", value=None, precision=0)
-
+            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)
-
-            gr.Markdown("### 📥 Export Results")
             with gr.Row():
                 json_btn = gr.Button("Export JSON")
                 csv_btn  = gr.Button("Export CSV")
@@ -97,16 +82,14 @@ with gr.Blocks(title="Speech Annotation Pipeline") as demo:
             csv_file  = gr.File(label="CSV")
 
         with gr.Column():
-            abab_box   = gr.Textbox(
-                label="Editable Transcript (ABAB)",
-                lines=20,
-                placeholder="Annotations will appear here..."
-            )
+            abab_box   = gr.Textbox(label="Editable Transcript (ABAB)", lines=20,
+                                    placeholder="Annotations will appear here...")
             table_html = gr.HTML(label="Segment Table")
 
     run_btn.click(
         fn=process_audio,
-        inputs=[audio_input, num_speakers, min_speakers, max_speakers],
+        inputs=[audio_input, num_speakers,
+                gr.State(0), gr.State(0)],
         outputs=[status_box, abab_box, table_html]
     )
     json_btn.click(fn=export_json, inputs=[], outputs=[json_file])