Update visualization.py

visualization.py

@@ -9,6 +9,9 @@ def create_charts(results):
         charts[speaker_id] = {}
         explanations[speaker_id] = {}
         
+        # Extract the actual name from the speaker_id
+        speaker_name = speaker_id.split(": ", 1)[-1]
+        
         # Attachment Styles
         attachment_data = data['attachments']
         labels = ['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant']
@@ -16,9 +19,10 @@ def create_charts(results):
                   getattr(attachment_data, 'Anxious_Preoccupied', 0), 
                   getattr(attachment_data, 'Dismissive_Avoidant', 0), 
                   getattr(attachment_data, 'Fearful_Avoidant', 0)]
+        colors = ['blue', 'orange', 'green', 'red']
         
-        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
-        fig.update_layout(title=f'Attachment Styles - {speaker_id}', yaxis_range=[0, 1])
+        fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
+        fig.update_layout(title=f'Attachment Styles: {speaker_name}', yaxis_range=[0, 1])
         charts[speaker_id]['attachment'] = fig
         explanations[speaker_id]['attachment'] = attachment_data.Explanation
         
@@ -27,36 +31,47 @@ def create_charts(results):
         labels = ['Extraversion', 'Agreeableness', 'Conscientiousness', 'Neuroticism', 'Openness']
         values = [bigfive_data.Extraversion, bigfive_data.Agreeableness, 
                   bigfive_data.Conscientiousness, bigfive_data.Neuroticism, bigfive_data.Openness]
+        colors = ['blue', 'green', 'red', 'purple', 'orange']
         
-        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
-        fig.update_layout(title=f'Big Five Traits - {speaker_id}', yaxis_range=[0, 10])
+        fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
+        fig.update_layout(title=f'Big Five Traits: {speaker_name}', yaxis_range=[0, 10])
         charts[speaker_id]['bigfive'] = fig
         explanations[speaker_id]['bigfive'] = bigfive_data.Explanation
         
         # Personality Disorders
         personality_data = data['personalities']
-        labels = ['Depressed', 'Paranoid', 'Schizoid-Schizotypal', 'Antisocial-Psychopathic',
-                  'Borderline-Dysregulated', 'Narcissistic', 'Anxious-Avoidant',
-                  'Dependent-Victimized', 'Obsessional']
-        values = [personality_data.Depressed, personality_data.Paranoid,
-                  personality_data.Schizoid_Schizotypal, personality_data.Antisocial_Psychopathic,
-                  personality_data.Borderline_Dysregulated, personality_data.Narcissistic,
-                  personality_data.Anxious_Avoidant, personality_data.Dependent_Victimized,
-                  personality_data.Obsessional]
-        
-        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
-        fig.update_layout(title=f'Personality Disorders - {speaker_id}', yaxis_range=[0, 5])
+        labels = ['Antisocial', 'Narcissistic', 'Depressed', 'Anxious-Avoidant',
+                  'Obsessive', 'Paranoid', 'Borderline', 'Dependent', 'Schizoid-Schizotypal']
+        values = [personality_data.Antisocial_Psychopathic, personality_data.Narcissistic,
+                  personality_data.Depressed, personality_data.Anxious_Avoidant,
+                  personality_data.Obsessional, personality_data.Paranoid,
+                  personality_data.Borderline_Dysregulated, personality_data.Dependent_Victimized,
+                  personality_data.Schizoid_Schizotypal]
+        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'Personality Disorders: {speaker_name}', yaxis_range=[0, 5])
         charts[speaker_id]['personality'] = fig
         explanations[speaker_id]['personality'] = personality_data.Explanation
 
-        # Dimensions
+        # Attachment Dimensions (Radar Chart)
         dimensions_data = data['attachments']
         labels = ['Self', 'Others', 'Anxiety', 'Avoidance']
         values = [dimensions_data.Self, dimensions_data.Others, 
                   dimensions_data.Anxiety, dimensions_data.Avoidance]
         
-        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
-        fig.update_layout(title=f'Attachment Dimensions - {speaker_id}', yaxis_range=[0, 10])
+        fig = go.Figure(data=go.Scatterpolar(
+          r=values,
+          theta=labels,
+          fill='toself'
+        ))
+        fig.update_layout(
+          polar=dict(
+            radialaxis=dict(visible=True, range=[0, 10])
+          ),
+          showlegend=False,
+          title=f'Attachment Dimensions: {speaker_name}'
+        )
         charts[speaker_id]['dimensions'] = fig
     
     return charts, explanations