Update visualization.py

visualization.py

@@ -18,10 +18,9 @@ def create_charts(results):
         values = [attachment_data.secured, attachment_data.anxious_preoccupied, 
                   attachment_data.dismissive_avoidant, attachment_data.fearful_avoidant]
         
-        charts[speaker_id]['attachment'] = {
-            'data': [{'x': labels, 'y': values, 'type': 'bar'}],
-            'layout': {'title': f'{speaker_id}: Attachment Styles', 'yaxis': {'range': [0, 1]}}
-        }
+        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
+        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
         
         # Big Five Traits
@@ -30,10 +29,9 @@ def create_charts(results):
         values = [bigfive_data.extraversion, bigfive_data.agreeableness, 
                   bigfive_data.conscientiousness, bigfive_data.neuroticism, bigfive_data.openness]
         
-        charts[speaker_id]['bigfive'] = {
-            'data': [{'x': labels, 'y': values, 'type': 'bar'}],
-            'layout': {'title': f'{speaker_id}: Big Five Traits', 'yaxis': {'range': [0, 10]}}
-        }
+        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
+        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
@@ -46,10 +44,9 @@ def create_charts(results):
                   personality_data.anxious_avoidant, personality_data.dependent_victimized,
                   personality_data.obsessional]
         
-        charts[speaker_id]['personality'] = {
-            'data': [{'x': labels, 'y': values, 'type': 'bar'}],
-            'layout': {'title': f'{speaker_id}: Personality Disorders', 'yaxis': {'range': [0, 5]}}
-        }
+        fig = go.Figure(data=[go.Bar(x=labels, y=values)])
+        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
 
         # Attachment Dimensions (Radar Chart)
@@ -58,18 +55,18 @@ def create_charts(results):
         values = [dimensions_data.self_model, dimensions_data.others_model, 
                   dimensions_data.anxiety, dimensions_data.avoidance]
         
-        charts[speaker_id]['dimensions'] = {
-            'data': [{
-                'type': 'scatterpolar',
-                'r': values,
-                'theta': labels,
-                'fill': 'toself'
-            }],
-            'layout': {
-                'polar': {'radialaxis': {'visible': True, 'range': [0, 10]}},
-                'showlegend': False,
-                'title': f'{speaker_id}: Attachment Dimensions'
-            }
-        }
+        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'{speaker_id}: Attachment Dimensions'
+        )
+        charts[speaker_id]['dimensions'] = fig
     
     return charts, explanations, general_impressions