Update app.py

app.py

@@ -44,19 +44,54 @@ def create_output_components():
             transcript = gr.Textbox(label="Transcript", lines=10, visible=False)
             
             tabs = gr.Tabs()
+            tab_components = []
             for i in range(3):  # Pre-create 3 tabs (max number of speakers)
                 with gr.Tab(f"Speaker {i+1}", visible=False) as tab:
-                    gr.Markdown(f"## Speaker {i+1}", visible=False)
-                    gr.Plot(label="Attachment", visible=False)
-                    gr.Textbox(label="Attachment Styles Explanation", visible=False)
-                    gr.Plot(label="Dimensions", visible=False)
-                    gr.Plot(label="Big Five", visible=False)
-                    gr.Textbox(label="Big Five Traits Explanation", visible=False)
-                    gr.Plot(label="Personality", visible=False)
-                    gr.Textbox(label="Personality Disorders Explanation", visible=False)
+                    speaker_components = [
+                        gr.Markdown(f"## Speaker {i+1}", visible=False),
+                        gr.Plot(label="Attachment", visible=False),
+                        gr.Textbox(label="Attachment Styles Explanation", visible=False),
+                        gr.Plot(label="Dimensions", visible=False),
+                        gr.Plot(label="Big Five", visible=False),
+                        gr.Textbox(label="Big Five Traits Explanation", visible=False),
+                        gr.Plot(label="Personality", visible=False),
+                        gr.Textbox(label="Personality Disorders Explanation", visible=False)
+                    ]
+                    tab_components.extend(speaker_components)
             
-            execution_info = gr.Textbox(label="Execution Information", visible=True)
-    return row, transcript, tabs, execution_info
+            execution_info = gr.Textbox(label="Execution Information", visible=False)
+    return row, transcript, tab_components, execution_info
+
+def run_analysis(video_path, max_speakers):
+    results = analyze_video(video_path, max_speakers)
+    
+    if "error" in results:
+        return [gr.update(value="", visible=False)] + [gr.update(visible=False)] * 24 + [gr.update(value=results["error"], visible=True)]
+
+    transcript = results["transcript"]
+    execution_info = f"Completed in {results['execution_time']} seconds."
+
+    tab_updates = []
+    for i in range(3):  # For each potential speaker
+        if i < len(results["charts"]):
+            speaker_id = list(results["charts"].keys())[i]
+            speaker_charts = results["charts"][speaker_id]
+            speaker_explanations = results["explanations"][speaker_id]
+            
+            tab_updates.extend([
+                gr.update(value=f"## {speaker_id}", visible=True),  # Markdown
+                gr.update(value=speaker_charts.get("attachment", None), visible=True),  # Attachment plot
+                gr.update(value=speaker_explanations.get("attachment", ""), visible=True),  # Attachment explanation
+                gr.update(value=speaker_charts.get("dimensions", None), visible=True),  # Dimensions plot
+                gr.update(value=speaker_charts.get("bigfive", None), visible=True),  # Big Five plot
+                gr.update(value=speaker_explanations.get("bigfive", ""), visible=True),  # Big Five explanation
+                gr.update(value=speaker_charts.get("personality", None), visible=True),  # Personality plot
+                gr.update(value=speaker_explanations.get("personality", ""), visible=True),  # Personality explanation
+            ])
+        else:
+            tab_updates.extend([gr.update(visible=False)] * 8)  # Hide unused tab components
+
+    return [gr.update(value=transcript, visible=True)] + tab_updates + [gr.update(value=execution_info, visible=True)]
 
 with gr.Blocks() as iface:
     gr.Markdown("# AI Personality Detection")
@@ -66,54 +101,14 @@ with gr.Blocks() as iface:
     max_speakers = gr.Slider(minimum=1, maximum=3, step=1, value=2, label="Maximum Number of Speakers")
     analyze_button = gr.Button("Analyze")
 
-    output_row, transcript_output, tabs_output, execution_info_output = create_output_components()
-
-    def run_analysis(video_path, max_speakers):
-        results = analyze_video(video_path, max_speakers)
-        
-        if "error" in results:
-            return [
-                "",  # transcript
-                None,  # tabs (no update)
-                results["error"],  # execution_info
-            ] + [gr.update(visible=False)] * 24  # 8 components per tab * 3 tabs
-    
-        transcript = results["transcript"]
-        execution_info = f"Completed in {results['execution_time']} seconds."
-    
-        tab_updates = []
-        for i in range(3):  # For each potential speaker
-            if i < len(results["charts"]):
-                speaker_id = list(results["charts"].keys())[i]
-                speaker_charts = results["charts"][speaker_id]
-                speaker_explanations = results["explanations"][speaker_id]
-                
-                tab_updates.extend([
-                    gr.update(visible=True, label=speaker_id),  # Tab visibility and label
-                    gr.update(value=f"## {speaker_id}", visible=True),  # Markdown
-                    gr.update(value=speaker_charts.get("attachment", None), visible=True),  # Attachment plot
-                    gr.update(value=speaker_explanations.get("attachment", ""), visible=True),  # Attachment explanation
-                    gr.update(value=speaker_charts.get("dimensions", None), visible=True),  # Dimensions plot
-                    gr.update(value=speaker_charts.get("bigfive", None), visible=True),  # Big Five plot
-                    gr.update(value=speaker_explanations.get("bigfive", ""), visible=True),  # Big Five explanation
-                    gr.update(value=speaker_charts.get("personality", None), visible=True),  # Personality plot
-                    gr.update(value=speaker_explanations.get("personality", ""), visible=True),  # Personality explanation
-                ])
-            else:
-                tab_updates.extend([gr.update(visible=False)] * 9)  # Hide unused tab and its components
-    
-        return [gr.update(value=transcript, visible=True), None, gr.update(value=execution_info, visible=True)] + tab_updates
-    
-    # Modify the analyze_button.click setup
+    output_row, transcript_output, tab_components, execution_info_output = create_output_components()
+
     analyze_button.click(
         fn=run_analysis,
         inputs=[video_input, max_speakers],
-        outputs=[
-            transcript_output,
-            tabs_output,
-            execution_info_output
-        ] + [component for tab in tabs_output.children for component in tab.children],
+        outputs=[transcript_output] + tab_components + [execution_info_output],
         show_progress=True
     )
+
 if __name__ == "__main__":
     iface.launch()