Update visualization.py

visualization.py

@@ -4,67 +4,78 @@ from plotly.subplots import make_subplots
 def create_charts(results):
     charts = {}
     explanations = {}
+    
+    # Assuming results contain 'attachments', 'bigfive', and 'personalities' directly
+    speaker_data = {
+        'attachments': results.get('attachments', {}),
+        'bigfive': results.get('bigfive', {}),
+        'personalities': results.get('personalities', {})
+    }
+    
+    speaker_id = "Speaker 1"  # Default speaker ID
+    speaker_charts = {}
+    speaker_explanations = {}
 
-    # Loop through each detected speaker in the results
-    for speaker_id, speaker_data in results['speakers'].items():
-        speaker_charts = {}
-        speaker_explanations = {}
-
-        # Attachment Styles for each speaker
-        attachment_data = speaker_data['attachments']
+    # Attachment Styles
+    attachment_data = speaker_data['attachments']
+    if attachment_data:
         fig_attachment = go.Figure(go.Bar(
             x=['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant'],
-            y=[attachment_data.secured, attachment_data.anxious_preoccupied,
-               attachment_data.dismissive_avoidant, attachment_data.fearful_avoidant],
+            y=[attachment_data.get('secured', 0), attachment_data.get('anxious_preoccupied', 0),
+               attachment_data.get('dismissive_avoidant', 0), attachment_data.get('fearful_avoidant', 0)],
             marker_color=['blue', 'orange', 'green', 'red']
         ))
-        fig_attachment.update_layout(title_text=f"Attachment Styles - Speaker {speaker_id}", showlegend=False)
+        fig_attachment.update_layout(title_text=f"Attachment Styles - {speaker_id}", showlegend=False)
         speaker_charts["attachment"] = fig_attachment
-        speaker_explanations["attachment"] = attachment_data.explanation
+        speaker_explanations["attachment"] = attachment_data.get('explanation', '')
 
-        # Attachment Dimensions (Radar Chart) for each speaker
+        # Attachment Dimensions (Radar Chart)
         fig_dimensions = go.Figure(go.Scatterpolar(
-            r=[attachment_data.avoidance, attachment_data.anxiety, attachment_data.self_rating, attachment_data.others_rating],
+            r=[attachment_data.get('avoidance', 0), attachment_data.get('anxiety', 0), 
+               attachment_data.get('self_rating', 0), attachment_data.get('others_rating', 0)],
             theta=['Avoidance', 'Anxiety', 'Self', 'Others'],
             fill='toself'
         ))
-        fig_dimensions.update_layout(title_text=f"Attachment Dimensions - Speaker {speaker_id}", showlegend=False)
+        fig_dimensions.update_layout(title_text=f"Attachment Dimensions - {speaker_id}", showlegend=False)
         speaker_charts["dimensions"] = fig_dimensions
 
-        # Big Five Traits for each speaker
-        bigfive_data = speaker_data['bigfive']
+    # Big Five Traits
+    bigfive_data = speaker_data['bigfive']
+    if bigfive_data:
         fig_bigfive = go.Figure(go.Bar(
             x=['Extraversion', 'Agreeableness', 'Conscientiousness', 'Neuroticism', 'Openness'],
-            y=[bigfive_data.extraversion, bigfive_data.agreeableness,
-               bigfive_data.conscientiousness, bigfive_data.neuroticism, bigfive_data.openness],
+            y=[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)],
             marker_color=['blue', 'green', 'red', 'purple', 'orange']
         ))
-        fig_bigfive.update_layout(title_text=f"Big Five Traits - Speaker {speaker_id}", showlegend=False)
+        fig_bigfive.update_layout(title_text=f"Big Five Traits - {speaker_id}", showlegend=False)
         speaker_charts["bigfive"] = fig_bigfive
-        speaker_explanations["bigfive"] = bigfive_data.explanation
+        speaker_explanations["bigfive"] = bigfive_data.get('explanation', '')
 
-        # Personality Disorders for each speaker
-        personality_data = speaker_data['personalities']
+    # Personality Disorders
+    personality_data = speaker_data['personalities']
+    if personality_data:
         fig_personality = go.Figure(go.Bar(
             x=['Antisocial', 'Narcissistic', 'Depressed', 'Anxious-Avoidant',
                'Obsessive', 'Paranoid', 'Borderline', 'Dependent', 'Schizoid-Schizotypal'],
-            y=[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],
+            y=[personality_data.get('antisocial_psychopathic', 0), personality_data.get('narcissistic', 0),
+               personality_data.get('depressed', 0), personality_data.get('anxious_avoidant', 0),
+               personality_data.get('obsessional', 0), personality_data.get('paranoid', 0),
+               personality_data.get('borderline_dysregulated', 0), personality_data.get('dependent_victimized', 0),
+               personality_data.get('schizoid_schizotypal', 0)],
             marker_color=['black', 'orange', 'gray', 'green', 'brown', 'purple', 'red', 'cyan', 'magenta']
         ))
-        fig_personality.update_layout(title_text=f"Personality Disorders - Speaker {speaker_id}", showlegend=False)
+        fig_personality.update_layout(title_text=f"Personality Disorders - {speaker_id}", showlegend=False)
         speaker_charts["personality"] = fig_personality
-        speaker_explanations["personality"] = personality_data.explanation
+        speaker_explanations["personality"] = personality_data.get('explanation', '')
 
-        # Update all charts to take full width
-        for fig in speaker_charts.values():
-            fig.update_layout(height=400, width=None, margin=dict(l=50, r=50, t=100, b=50))
+    # Update all charts to take full width
+    for fig in speaker_charts.values():
+        fig.update_layout(height=400, width=None, margin=dict(l=50, r=50, t=100, b=50))
 
-        # Store the charts and explanations for each speaker
-        charts[speaker_id] = speaker_charts
-        explanations[speaker_id] = speaker_explanations
+    # Store the charts and explanations
+    charts[speaker_id] = speaker_charts
+    explanations[speaker_id] = speaker_explanations
 
-    return charts, explanations
+    return charts, explanations