Update app.py

app.py

@@ -33,49 +33,54 @@ def analyze_video(video_path, progress=gr.Progress()):
 
     output_components = [transcription]  # transcript
     
-    for i in range(3):
-        speaker_id = f"Speaker {i+1}" if i < len(charts) else f"Speaker {i+1} (Not Present)"
-        speaker_charts = charts.get(speaker_id, {})
-        speaker_explanations = explanations.get(speaker_id, {})
-        speaker_general_impression = general_impressions.get(speaker_id, "No data for this speaker")
-        
-        speaker_section1 = [
-            gr.Markdown(f"# {speaker_id}", visible=True),
-            gr.Textbox(value=speaker_general_impression, label="General Impression", visible=True)
-        ]
-        
-        speaker_section2 = [
-            gr.Plot(value=speaker_charts.get("attachment", None), visible=True),
-            gr.Plot(value=speaker_charts.get("dimensions", None), visible=True),
-            gr.Textbox(value=speaker_explanations.get("attachment", ""), label="Attachment Styles Explanation", visible=True)
-        ]
-        
-        speaker_section3 = [
-            gr.Plot(value=speaker_charts.get("bigfive", None), visible=True),
-            gr.Textbox(value=speaker_explanations.get("bigfive", ""), label="Big Five Traits Explanation", visible=True)
-        ]
-        
-        speaker_section4 = [    
-            gr.Plot(value=speaker_charts.get("personality", None), visible=True),
-            gr.Textbox(value=speaker_explanations.get("personality", ""),  label="Personality Disorders Explanation", visible=True)
-        ]
+    for i, (speaker_id, speaker_charts) in enumerate(charts.items(), start=1):
+        print(speaker_id)
+        speaker_explanations = explanations[speaker_id]
+        speaker_general_impression = general_impressions[speaker_id]
         
+        with gr.Tab(visible=True):
+            with gr.TabItem(label=f'General Impression'):  
+                speaker_section1 = [
+                    gr.Markdown(f"# {speaker_id}", visible=True),
+                    gr.Textbox(value=speaker_general_impression, label="General Impression", visible=True)
+                ]
+                
+            with gr.TabItem(label=f'Attachment Styles'):
+                with gr.Row(): 
+                    speaker_section2 = [
+                        gr.Plot(value=speaker_charts.get("attachment", None), visible=True),
+                        gr.Plot(value=speaker_charts.get("dimensions", None), visible=True),
+                        gr.Textbox(value=speaker_explanations.get("attachment", ""), label="Attachment Styles Explanation", visible=True)
+                    ]
+            
+            with gr.TabItem(label=f'Big Five Traits'):  
+                speaker_section3 = [
+                    gr.Plot(value=speaker_charts.get("bigfive", None), visible=True),
+                    gr.Textbox(value=speaker_explanations.get("bigfive", ""), label="Big Five Traits Explanation", visible=True)
+                ]
+            
+            with gr.TabItem(label=f'Personalities'):  
+                speaker_section4 = [    
+                    gr.Plot(value=speaker_charts.get("personality", None), visible=True),
+                    gr.Textbox(value=speaker_explanations.get("personality", ""),  label="Personality Disorders Explanation", visible=True)
+                ]
+                
         output_components.extend(speaker_section1)
         output_components.extend(speaker_section2)
         output_components.extend(speaker_section3)
         output_components.extend(speaker_section4)
-                
-    output_components.append(f"Completed in {int(execution_time)} seconds.")  # execution info
 
-    # Pad with None for any missing speakers
-    while len(output_components) < 28:
+        # Pad with None for any missing speakers
+        while len(output_components) < 28:
             output_components.extend([gr.update(visible=False)] * 9)
                 
+    output_components.append(f"Completed in {int(execution_time)} seconds.")  # execution info
+
     return output_components
 
 def update_output(*args):
     return [gr.update(value=arg, visible=arg is not None) for arg in args]
-    
+
 def use_example():
     return "examples/Scenes.From.A.Marriage.US.mp4"