Update visualization.py

visualization.py

@@ -35,20 +35,20 @@ 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_id}: Big Five Traits', yaxis_range=[-10, 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
         
         # Personality Disorders
         personality_data = data['personalities']
-        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']
+        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]
+        colors = ['gray', 'purple', 'gold', 'black', 'red', 'orange', 'green', 'cyan', 'brown']
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
         fig.update_layout(title=f'{speaker_id}: Personality Disorders', yaxis_range=[0, 5])