Update visualization.py

visualization.py

@@ -15,45 +15,45 @@ def create_charts(results):
         # Attachment Styles
         attachment_data = data['attachments']
         labels = ['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant']
-        values = [attachment_data.secured, attachment_data.anxious_preoccupied, 
-                  attachment_data.dismissive_avoidant, attachment_data.fearful_avoidant]
+        values = [attachment_data.get('secured', 0), attachment_data.get('anxious_preoccupied', 0), 
+                  attachment_data.get('dismissive_avoidant', 0), attachment_data.get('fearful_avoidant', 0)]
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=['blue', 'orange', 'green', 'red'])])
         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
+        explanations[speaker_id]['attachment'] = attachment_data.get('explanation', "No explanation provided")
         
         # Big Five Traits
         bigfive_data = data['bigfive']
         labels = ['Extraversion', 'Agreeableness', 'Conscientiousness', 'Neuroticism', 'Openness']
-        values = [bigfive_data.extraversion, bigfive_data.agreeableness, 
-                  bigfive_data.conscientiousness, bigfive_data.neuroticism, bigfive_data.openness]
+        values = [bigfive_data.get('extraversion', 0), bigfive_data.get('agreeableness', 0), 
+                  bigfive_data.get('conscientiousness', 0), bigfive_data.get('neuroticism', 0), bigfive_data.get('openness', 0)]
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=['blue', 'green', 'red', 'purple', 'orange'])])
         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
+        explanations[speaker_id]['bigfive'] = bigfive_data.get('explanation', "No explanation provided")
         
         # 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]
+        values = [personality_data.get('depressed', 0), personality_data.get('paranoid', 0),
+                  personality_data.get('schizoid_schizotypal', 0), personality_data.get('antisocial_psychopathic', 0),
+                  personality_data.get('borderline_dysregulated', 0), personality_data.get('narcissistic', 0),
+                  personality_data.get('anxious_avoidant', 0), personality_data.get('dependent_victimized', 0),
+                  personality_data.get('obsessional', 0)]
         
         fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=['black', 'orange', 'gray', 'green', 'brown', 'purple', 'red', 'cyan', 'magenta'])])
         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
+        explanations[speaker_id]['personality'] = personality_data.get('explanation', "No explanation provided")
 
         # Attachment Dimensions (Radar Chart)
         dimensions_data = data['attachments']
         labels = ['Self', 'Others', 'Anxiety', 'Avoidance']
-        values = [dimensions_data.self_model, dimensions_data.others_model, 
-                  dimensions_data.anxiety, dimensions_data.avoidance]
+        values = [dimensions_data.get('self_model', 0), dimensions_data.get('others_model', 0), 
+                  dimensions_data.get('anxiety', 0), dimensions_data.get('avoidance', 0)]
         
         fig = go.Figure(data=go.Scatterpolar(
           r=values,
@@ -71,6 +71,4 @@ def create_charts(results):
         )
         charts[speaker_id]['dimensions'] = fig
 
-    
-    return charts, explanations, general_impressions
-    
+    return charts, explanations, general_impressions