Update visualization.py

visualization.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import plotly.graph_objs as go
 import re
+import json
 
 def extract_data_and_explanation(text):
     speakers_data = {}
@@ -20,6 +21,7 @@ def extract_data_and_explanation(text):
             explanation = line.split("Explanation:", 1)[1].strip()
     return speakers_data, explanation
 
+
 def create_bar_chart(data, title):
     fig = go.Figure(data=[go.Bar(
         x=list(data.keys()), 
@@ -41,7 +43,7 @@ def update_visibility_and_charts(status, exec_time, lang, attachments, bigfive,
         for speaker, data in speakers_data.items():
             if data:
                 fig = create_bar_chart(data, f"{analysis_type} Analysis - {speaker}")
-                outputs.append(fig.to_json())
+                outputs.append(fig)
                 outputs.append(gr.update(value=explanation, visible=True, label=f"{analysis_type} Explanation - {speaker}"))
             else:
                 outputs.append(None)