Update app.py

app.py

@@ -37,33 +37,51 @@ def analyze_video(video_path, progress=gr.Progress()):
         speaker_general_impression = general_impressions[speaker_id]
 
         speaker_section = [
-            gr.Markdown(f"## {speaker_id}", visible=True),
-            gr.Textbox(value=speaker_general_impression, label="General Impression",
-                       visible=True),
-            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),
-            gr.Plot(value=speaker_charts.get("bigfive", None), visible=True),
-            gr.Textbox(value=speaker_explanations.get("bigfive", ""), label="Big Five Traits Explanation",
-                       visible=True),
-            gr.Plot(value=speaker_charts.get("personality", None), visible=True),
-            gr.Textbox(value=speaker_explanations.get("personality", ""), label="Personality Disorders Explanation",
-                       visible=True),
+            f"## {speaker_id}",
+            speaker_general_impression,
+            speaker_charts.get("attachment", None),
+            speaker_charts.get("dimensions", None),
+            speaker_explanations.get("attachment", ""),
+            speaker_charts.get("bigfive", None),
+            speaker_explanations.get("bigfive", ""),
+            speaker_charts.get("personality", None),
+            speaker_explanations.get("personality", ""),
         ]
         output_components.extend(speaker_section)
         
     # Pad with None for any missing speakers
     while len(output_components) < 28:
-        output_components.extend([gr.update(visible=False)] * 9)
+        output_components.extend([None] * 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 analyze_and_update(video_path):
+    results = analyze_video(video_path)
+    outputs = results[:-1]  # All outputs except the last one (execution info)
+    execution_info = results[-1]  # Last item is execution info
+    
+    visible_speakers = sum(1 for output in outputs[1:28:9] if output is not None)
+    
+    # Update visibility of components
+    for i in range(len(outputs)):
+        if outputs[i] is None:
+            outputs[i] = gr.update(visible=False)
+        else:
+            outputs[i] = gr.update(value=outputs[i], visible=True)
+    
+    # Update tab visibility
+    for i in range(3):
+        if i < visible_speakers:
+            outputs[i*9+1] = gr.update(value=f"## Speaker {i+1}", visible=True)
+        else:
+            for j in range(9):
+                outputs[i*9+j+1] = gr.update(visible=False)
+    
+    outputs.append(gr.update(value=execution_info, visible=True))
+    
+    return outputs
 
 def use_example():
     return "examples/Scenes.From.A.Marriage.US.mp4"
@@ -83,57 +101,45 @@ with gr.Blocks() as iface:
 
     # Create output components
     output_components = []
-    tab_visibilities = []
-    
+
     # Add transcript output near the top
-    execution_info_box = gr.Textbox(label="Transcript", value="N/A", lines=1)
-    output_components.append(execution_info_box)
+    transcript_output = gr.Textbox(label="Transcript", lines=10, visible=False)
+    output_components.append(transcript_output)
 
-    for i in range(3):  # Assuming maximum of 3 speakers
-        with gr.Tab(f"Speaker {i+1}", visible=False) as tab:
-            tab_visibilities.append(tab)
-            with gr.Row():
-                output_components.extend([
-                    gr.Markdown(visible=False),
-                    gr.Textbox(label="General Impression", visible=False),
-                    gr.Plot(visible=False),
-                    gr.Plot(visible=False),
-                    gr.Textbox(label="Attachment Styles Explanation", visible=False),
-                    gr.Plot(visible=False),
-                    gr.Textbox(label="Big Five Traits Explanation", visible=False),
-                    gr.Plot(visible=False),
-                    gr.Textbox(label="Personality Disorders Explanation", visible=False),
-                ])
+    tabs = gr.Tabs()
+    with tabs:
+        for i in range(3):  # Assuming maximum of 3 speakers
+            with gr.Tab(f"Speaker {i+1}"):
+                with gr.Row():
+                    output_components.extend([
+                        gr.Markdown(visible=False),
+                        gr.Textbox(label="General Impression", visible=False),
+                        gr.Plot(visible=False),
+                        gr.Plot(visible=False),
+                        gr.Textbox(label="Attachment Styles Explanation", visible=False),
+                        gr.Plot(visible=False),
+                        gr.Textbox(label="Big Five Traits Explanation", visible=False),
+                        gr.Plot(visible=False),
+                        gr.Textbox(label="Personality Disorders Explanation", visible=False),
+                    ])
 
     # Add execution info component
-    transcript_output = gr.Textbox(label="Transcript", lines=10, visible=False)
-    output_components.append(transcript_output)
-    
-    def update_interface(results):
-        outputs = results[:-3]  # All outputs except the last 3 (visible tabs info)
-        visible_tabs = results[-3:]  # Last 3 items are visible tabs info
-        
-        # Update tab visibility
-        tab_updates = [gr.update(visible=(i in visible_tabs)) for i in range(1, 4)]
-        
-        return outputs + tab_updates
-        
+    execution_info_box = gr.Textbox(label="Execution Info", value="N/A", lines=1)
+    output_components.append(execution_info_box)
+
     analyze_button.click(
-        fn=analyze_video,
+        fn=analyze_and_update,
         inputs=[video_input],
-        outputs=output_components + tab_visibilities,
+        outputs=output_components,
         show_progress=True
-    ).then(
-        fn=update_interface,
-        inputs=output_components + tab_visibilities,
-        outputs=output_components + tab_visibilities
     )
 
     use_example_button.click(
         fn=use_example,
         inputs=[],
         outputs=[video_input],
-    ).then(fn=analyze_video,
+    ).then(
+        fn=analyze_and_update,
         inputs=[video_input],
         outputs=output_components,
         show_progress=True