Update app.py

app.py

@@ -137,19 +137,36 @@ def analyze_video(video_path, language_display_name, max_speakers, progress=gr.P
     for speaker_id, speaker_charts in charts.items():
         speaker_explanations = explanations[speaker_id]
 
-        output_components.extend([
-            gr.Markdown(f"### Speaker {speaker_id}"),
-            gr.Plot(speaker_charts["attachment"]),
-            gr.Textbox(value=speaker_explanations["attachment"],
-                       label=f"Attachment Styles Explanation - Speaker {speaker_id}", lines=2),
-            gr.Plot(speaker_charts["dimensions"]),
-            gr.Plot(speaker_charts["bigfive"]),
-            gr.Textbox(value=speaker_explanations["bigfive"],
-                       label=f"Big Five Traits Explanation - Speaker {speaker_id}", lines=2),
-            gr.Plot(speaker_charts["personality"]),
-            gr.Textbox(value=speaker_explanations["personality"],
-                       label=f"Personality Disorders Explanation - Speaker {speaker_id}", lines=2)
-        ])
+    # Prepare outputs for each speaker
+    output_components = []
+    for speaker_id, speaker_charts in charts.items():
+        speaker_explanations = explanations[speaker_id]
+
+        output_components.append(gr.Markdown(f"### {speaker_id}"))
+        
+        if "attachment" in speaker_charts:
+            output_components.extend([
+                gr.Plot(speaker_charts["attachment"]),
+                gr.Textbox(value=speaker_explanations.get("attachment", ""),
+                           label=f"Attachment Styles Explanation - {speaker_id}", lines=2),
+            ])
+        
+        if "dimensions" in speaker_charts:
+            output_components.append(gr.Plot(speaker_charts["dimensions"]))
+        
+        if "bigfive" in speaker_charts:
+            output_components.extend([
+                gr.Plot(speaker_charts["bigfive"]),
+                gr.Textbox(value=speaker_explanations.get("bigfive", ""),
+                           label=f"Big Five Traits Explanation - {speaker_id}", lines=2),
+            ])
+        
+        if "personality" in speaker_charts:
+            output_components.extend([
+                gr.Plot(speaker_charts["personality"]),
+                gr.Textbox(value=speaker_explanations.get("personality", ""),
+                           label=f"Personality Disorders Explanation - {speaker_id}", lines=2),
+            ])
 
     # Add the transcript at the end
     output_components.append(gr.Textbox(value=transcription, label="Transcript", lines=10))