Update visualization.py

visualization.py

@@ -27,7 +27,7 @@ def create_radar_chart(data, title):
     )
     return fig
 
-def update_visibility_and_charts(status, exec_time, lang, transcription, attachments_data, bigfive_data, personalities_data, chart_data):
+def update_visibility_and_charts(status, exec_time, lang, transcription, attachments_data, bigfive_data, personalities_data, chart_data, max_speakers):
     outputs = [
         gr.update(value=status, visible=True),
         gr.update(value=exec_time, visible=True),
@@ -35,13 +35,17 @@ def update_visibility_and_charts(status, exec_time, lang, transcription, attachm
         gr.update(value=transcription, visible=True),
     ]
     
-    sorted_speakers = sorted(chart_data.keys())
+    # Collect all unique speaker names from all datasets
+    all_speakers = set(attachments_data.keys()) | set(bigfive_data.keys()) | set(personalities_data.keys()) | set(chart_data.keys())
     
-    for speaker in sorted_speakers[:2]:  # Limit to 2 speakers
+    # Sort speakers and limit to max_speakers (up to 3)
+    sorted_speakers = sorted(all_speakers)[:min(max_speakers, 3)]
+    
+    for speaker in sorted_speakers:
         speaker_outputs = []
         
         # Attachments
-        attachment_data = attachments_data.get(f"Speaker: {speaker}", {}) or attachments_data.get(speaker, {})
+        attachment_data = attachments_data.get(speaker, {})
         if attachment_data:
             attachment_chart_data = {k: v for k, v in attachment_data.items() if k in ["Secured", "Anxious-Preoccupied", "Dismissive-Avoidant", "Fearful-Avoidant"]}
             attachment_chart = create_bar_chart(attachment_chart_data, f"Attachments Analysis - {speaker}")
@@ -57,7 +61,7 @@ def update_visibility_and_charts(status, exec_time, lang, transcription, attachm
             speaker_outputs.extend([gr.update(visible=False)] * 3)
         
         # Big Five
-        bigfive_data_speaker = bigfive_data.get(f"Speaker: {speaker}", {}) or bigfive_data.get(speaker, {})
+        bigfive_data_speaker = bigfive_data.get(speaker, {})
         if bigfive_data_speaker:
             bigfive_chart_data = {k: v for k, v in bigfive_data_speaker.items() if k not in ["explanation"] and isinstance(v, (int, float))}
             bigfive_chart = create_bar_chart(bigfive_chart_data, f"Big Five Analysis - {speaker}")
@@ -69,7 +73,7 @@ def update_visibility_and_charts(status, exec_time, lang, transcription, attachm
             speaker_outputs.extend([gr.update(visible=False)] * 2)
         
         # Personalities
-        personalities_data_speaker = personalities_data.get(f"Speaker: {speaker}", {}) or personalities_data.get(speaker, {})
+        personalities_data_speaker = personalities_data.get(speaker, {})
         if personalities_data_speaker:
             personalities_chart_data = {k: v for k, v in personalities_data_speaker.items() if k not in ["explanation"] and isinstance(v, (int, float))}
             personalities_chart = create_bar_chart(personalities_chart_data, f"Personalities Analysis - {speaker}")
@@ -82,14 +86,15 @@ def update_visibility_and_charts(status, exec_time, lang, transcription, attachm
         
         outputs.extend(speaker_outputs)
     
-    # Hide unused speaker components if there's only one speaker
-    if len(sorted_speakers) == 1:
+    # Hide unused speaker components
+    for _ in range(3 - len(sorted_speakers)):
         outputs.extend([gr.update(visible=False)] * 7)  # 7 components per speaker
     
     print("Debug: Attachments Data:", attachments_data)
     print("Debug: Big Five Data:", bigfive_data)
     print("Debug: Personalities Data:", personalities_data)
     print("Debug: Chart Data:", chart_data)
+    print("Debug: Sorted Speakers:", sorted_speakers)
     
     return outputs