Return diarization segments as JSON

app.py

@@ -213,13 +213,15 @@ def diarize(
         f"- ZeroGPU time used: **{zerogpu_seconds:.2f}s**"
     )
 
-    table = [
-        [
-            segment["speaker"],
-            _format_timestamp(segment["start"]),
-            _format_timestamp(segment["end"]),
-            round(segment["duration"], 3),
-        ]
+    segments_json = [
+        {
+            **segment,
+            "start": round(segment["start"], 3),
+            "end": round(segment["end"], 3),
+            "duration": round(segment["duration"], 3),
+            "start_timestamp": _format_timestamp(segment["start"]),
+            "end_timestamp": _format_timestamp(segment["end"]),
+        }
         for segment in segments
     ]
 
@@ -229,7 +231,7 @@ def diarize(
     )
 
     artifacts = _write_artifacts(segments, rttm_text)
-    return summary, round(zerogpu_seconds, 3), table, turns_text, artifacts
+    return summary, round(zerogpu_seconds, 3), segments_json, turns_text, artifacts
 
 
 def build_demo() -> gr.Blocks:
@@ -269,12 +271,7 @@ def build_demo() -> gr.Blocks:
             with gr.Column(scale=1):
                 summary_output = gr.Markdown()
                 zerogpu_seconds_output = gr.Number(label="ZeroGPU seconds used", precision=3)
-                segments_output = gr.Dataframe(
-                    headers=["Speaker", "Start", "End", "Duration (s)"],
-                    datatype=["str", "str", "str", "number"],
-                    interactive=False,
-                    label="Segments",
-                )
+                segments_output = gr.JSON(label="Segments JSON")
                 turns_output = gr.Textbox(
                     label="Speaker turns",
                     lines=14,
@@ -294,7 +291,7 @@ def build_demo() -> gr.Blocks:
 
         gr.Markdown(
             """
-            Outputs include a segment table, a plain-text speaker-turn list, and downloadable
+            Outputs include segments as JSON, a plain-text speaker-turn list, and downloadable
             `CSV`, `TXT`, and `RTTM` files.
             """
         )