Update visualization.py

visualization.py

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