Update visualization.py

visualization.py

@@ -9,8 +9,6 @@ def create_charts(results):
         charts[speaker_id] = {}
         explanations[speaker_id] = {}
         
-        # Extract the actual name from the speaker_id
-        
         # Attachment Styles
         attachment_data = data['attachments']
         labels = ['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant']
@@ -21,7 +19,7 @@ def create_charts(results):
         colors = ['blue', 'orange', 'green', 'red']
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
-        fig.update_layout(title=f'Speaker {speaker_id}: Attachment Styles', yaxis_range=[0, 1])
+        fig.update_layout(title=f'{speaker_id}: Attachment Styles', yaxis_range=[0, 1])
         charts[speaker_id]['attachment'] = fig
         explanations[speaker_id]['attachment'] = attachment_data.Explanation
         
@@ -33,7 +31,7 @@ def create_charts(results):
         colors = ['blue', 'green', 'red', 'purple', 'orange']
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
-        fig.update_layout(title=f'Speaker {speaker_id}: Big Five Traits', yaxis_range=[0, 10])
+        fig.update_layout(title=f'{speaker_id}: Big Five Traits', yaxis_range=[0, 10])
         charts[speaker_id]['bigfive'] = fig
         explanations[speaker_id]['bigfive'] = bigfive_data.Explanation
         
@@ -49,7 +47,7 @@ def create_charts(results):
         colors = ['black', 'orange', 'gray', 'green', 'brown', 'purple', 'red', 'cyan', 'gold']
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
-        fig.update_layout(title=f'Speaker {speaker_id}: Personality Disorders', yaxis_range=[0, 5])
+        fig.update_layout(title=f'{speaker_id}: Personality Disorders', yaxis_range=[0, 5])
         charts[speaker_id]['personality'] = fig
         explanations[speaker_id]['personality'] = personality_data.Explanation
 
@@ -69,7 +67,7 @@ def create_charts(results):
             radialaxis=dict(visible=True, range=[0, 10])
           ),
           showlegend=False,
-          title=f'Speaker {speaker_id}: Attachment Dimensions'
+          title=f'{speaker_id}: Attachment Dimensions'
         )
         charts[speaker_id]['dimensions'] = fig