Initial

app.py

@@ -206,19 +206,83 @@ def process_audio(uploaded_audio):
         return {"error": str(e)}
 
 
+def transform_output_to_tables(final_output: dict):
+    """
+    Helper function to convert data into a table view.
+    Transforms data inside `final_output` into two separate tables.
+
+    Parameters
+    ----------
+    final_output : dict
+        Dictionary containing processed results.
+
+    Returns
+    -------
+    tuple
+        Returns two lists as `(ssm_data, file_data)`.
+    """
+
+    if "error" in final_output:
+        return [], []
+
+    # Utterance Table
+    ssm_data = []
+    if "ssm" in final_output:
+        for item in final_output["ssm"]:
+            ssm_data.append([
+                item.get("speaker", ""),
+                item.get("start_time", ""),
+                item.get("end_time", ""),
+                item.get("text", ""),
+                item.get("index", ""),
+                item.get("sentiment", ""),
+                item.get("profane", "")
+            ])
+
+    # File Table
+    file_data = []
+    for key in ["summary", "conflict", "topic", "silence"]:
+        file_data.append([key, final_output.get(key, "")])
+
+    return ssm_data, file_data
+
+
 with gr.Blocks() as demo:
-    gr.Markdown("## Diarization, Transcription & Analysis")
+    gr.Markdown("Callytics Demo")
 
     with gr.Row():
         audio_input = gr.Audio(type="filepath", label="Upload your audio")
-        output_display = gr.JSON(label="Final Output")
 
     submit_btn = gr.Button("Process")
 
+    with gr.Row():
+        utterance_table = gr.Dataframe(
+            headers=["Speaker", "Start Time", "End Time", "Text", "Index", "Sentiment", "Profane"],
+            label="Utterance Table"
+        )
+
+    with gr.Row():
+        file_table = gr.Dataframe(
+            headers=["Key", "Value"],
+            label="File Table"
+        )
+
+    output_display = gr.JSON(label="Final Output (JSON)")
+
+
+    def process_and_show_tables(uploaded_audio):
+        """
+        Calls the main processing function `process_audio` and returns data suitable for the table.
+        """
+        final_output = process_audio(uploaded_audio)
+        ssm_data, file_data = transform_output_to_tables(final_output)
+        return ssm_data, file_data, final_output
+
+
     submit_btn.click(
-        fn=process_audio,
+        fn=process_and_show_tables,
         inputs=audio_input,
-        outputs=output_display
+        outputs=[utterance_table, file_table, output_display]
     )
 
 if __name__ == "__main__":