"""
def make_table_html(segments):
if not segments:
return ""
speaker_list = list(dict.fromkeys(s.speaker for s in segments))
color_map = {spk: SPEAKER_COLORS[i % len(SPEAKER_COLORS)] for i, spk in enumerate(speaker_list)}
rows = "".join(
f"
"
f"
{s.speaker}
"
f"
{s.start_fmt}
"
f"
{s.end_fmt}
"
f"
{s.text}
"
f"
"
for s in segments
)
return f"""
Speaker
Start
End
Transcript
{rows}
"""
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
segments = pipeline.process(audio_path, num_speakers=n)
except Exception as e:
return f"❌ Error: {e}\n{traceback.format_exc()}", "", ""
if not segments:
return "⚠️ No speech detected.", "", ""
_last_segments = segments
unique = len(set(s.speaker for s in segments))
status = f"✅ Done — {len(segments)} segments · {unique} speaker(s) detected"
return status, make_conversation_html(segments), make_table_html(segments)
def export_json():
if not _last_segments:
return None
out = str(UPLOAD_DIR / "annotation.json")
to_json(_last_segments, out)
return out
def export_csv():
if not _last_segments:
return None
out = str(UPLOAD_DIR / "annotation.csv")
to_csv(_last_segments, out)
return out
css = """
.gradio-container { max-width: 1100px !important; margin: auto !important; }
footer { display: none !important; }
"""
with gr.Blocks(title="Speech Annotation Pipeline", css=css) as demo:
gr.Markdown(\"\"\"# 🎙️ Speech Annotation Pipeline
*Upload audio · Detect speakers · Export transcript*\"\"\")
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", size="lg")
status_box = gr.Textbox(label="Status", value="Ready.", interactive=False)
gr.Markdown("### 📥 Export")
with gr.Row():
json_btn = gr.Button("⬇ JSON", size="sm")
csv_btn = gr.Button("⬇ CSV", size="sm")
json_file = gr.File(label="JSON Download", visible=True)
csv_file = gr.File(label="CSV Download", visible=True)
with gr.Column(scale=2):
gr.Markdown("### 💬 Conversation View")
conversation_html = gr.HTML(
value="