Update app.py

app.py

@@ -16,6 +16,7 @@ import time
 from langdetect import detect
 import plotly.graph_objs as go
 import re
+from collections import Counter
 
 # Set environment variable to disable tokenizers parallelism warning
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
@@ -264,7 +265,13 @@ def update_visibility_and_charts(status, exec_time, lang, info, attachments, big
         speakers_data = extract_speaker_data(analysis_text)
         if not speakers_data:
             print(f"No speaker data extracted from: {analysis_text}")
-        for speaker, data in speakers_data.items():
+
+        # Determine the two main speakers
+        speaker_counts = Counter(speakers_data.keys())
+        main_speakers = [speaker for speaker, count in speaker_counts.most_common(2)]
+        
+        for speaker in main_speakers:
+            data = speakers_data.get(speaker, {})
             attachment_data = {k: v for k, v in data.items() if k in ["Secured", "Anxious-Preoccupied", "Dismissive-Avoidant", "Fearful-Avoidant"]}
             if attachment_data:
                 charts.append(create_bar_chart(attachment_data, "Attachment Styles", speaker))
@@ -337,4 +344,4 @@ def create_interface():
 iface = create_interface()
 
 # Launch the app
-iface.launch()
+iface.launch()