Update visualization.py

visualization.py

@@ -3,19 +3,22 @@ import plotly.graph_objs as go
 import re
 
 def extract_data_and_explanation(text):
-    data = {}
+    speakers_data = {}
+    current_speaker = None
     explanation = ""
-    lines = text.split('\n')
-    for line in lines:
-        if ':' in line:
+    for line in text.split('\n'):
+        if line.startswith("Speaker"):
+            current_speaker = line.split(':')[0].strip()
+            speakers_data[current_speaker] = {}
+        elif ':' in line and current_speaker:
             key, value = line.split(':', 1)
             try:
-                data[key.strip()] = float(value.strip())
+                speakers_data[current_speaker][key.strip()] = float(value.strip())
             except ValueError:
                 pass
         elif line.lower().startswith("explanation:"):
             explanation = line.split("Explanation:", 1)[1].strip()
-    return data, explanation
+    return speakers_data, explanation
 
 def create_bar_chart(data, title):
     fig = go.Figure(data=[go.Bar(
@@ -34,11 +37,12 @@ def update_visibility_and_charts(status, exec_time, lang, attachments, bigfive,
     ]
     
     for analysis_text, analysis_type in [(attachments, "Attachments"), (bigfive, "Big Five"), (personalities, "Personalities")]:
-        data, explanation = extract_data_and_explanation(analysis_text)
-        if data:
-            fig = create_bar_chart(data, f"{analysis_type} Analysis")
-            outputs.append(gr.Plot(value=fig))
-            outputs.append(gr.Textbox(value=explanation, label=f"{analysis_type} Explanation"))
+        speakers_data, explanation = extract_data_and_explanation(analysis_text)
+        for speaker, data in speakers_data.items():
+            if data:
+                fig = create_bar_chart(data, f"{analysis_type} Analysis - {speaker}")
+                outputs.append(gr.Plot(value=fig))
+                outputs.append(gr.Textbox(value=explanation, label=f"{analysis_type} Explanation - {speaker}"))
     
     print("Outputs generated:", outputs)
     return outputs