Update visualization.py

visualization.py

@@ -4,10 +4,14 @@ from plotly.subplots import make_subplots
 def create_charts(results):
     charts = {}
     explanations = {}
+    general_impressions = {}
     
     for speaker_id, data in results.items():
         charts[speaker_id] = {}
         explanations[speaker_id] = {}
+
+        # Extract general impression
+        general_impressions[speaker_id] = data.get('general_impression', "No general impression provided.")
         
         # Attachment Styles
         attachment_data = data['attachments']
@@ -71,4 +75,4 @@ def create_charts(results):
         )
         charts[speaker_id]['dimensions'] = fig
     
-    return charts, explanations
+    return charts, explanations, general_impressions