Update app.py

app.py

@@ -72,42 +72,47 @@ with gr.Blocks() as iface:
         results = analyze_video(video_path, max_speakers)
         
         if "error" in results:
-            return {
-                transcript_output: gr.update(value="", visible=False),
-                tabs_output: gr.update(visible=False),
-                execution_info_output: gr.update(value=results["error"], visible=True)
-            }
-
-        updates = {
-            transcript_output: gr.update(value=results["transcript"], visible=True),
-            execution_info_output: gr.update(value=f"Completed in {results['execution_time']} seconds.", visible=True)
-        }
-
-        for i, (speaker_id, speaker_charts) in enumerate(results["charts"].items()):
-            speaker_explanations = results["explanations"][speaker_id]
-            tab_updates = {
-                f"Speaker {i+1}": gr.update(visible=True, label=speaker_id),
-                f"Speaker {i+1}/0": gr.update(value=f"## {speaker_id}", visible=True),
-                f"Speaker {i+1}/1": gr.update(value=speaker_charts.get("attachment", None), visible=True),
-                f"Speaker {i+1}/2": gr.update(value=speaker_explanations.get("attachment", ""), visible=True),
-                f"Speaker {i+1}/3": gr.update(value=speaker_charts.get("dimensions", None), visible=True),
-                f"Speaker {i+1}/4": gr.update(value=speaker_charts.get("bigfive", None), visible=True),
-                f"Speaker {i+1}/5": gr.update(value=speaker_explanations.get("bigfive", ""), visible=True),
-                f"Speaker {i+1}/6": gr.update(value=speaker_charts.get("personality", None), visible=True),
-                f"Speaker {i+1}/7": gr.update(value=speaker_explanations.get("personality", ""), visible=True),
-            }
-            updates.update(tab_updates)
-
-        # Hide unused tabs
-        for i in range(len(results["charts"]), 3):
-            updates[f"Speaker {i+1}"] = gr.update(visible=False)
-
-        return updates
-
+            return [
+                "",  # transcript
+                gr.Tabs(),  # tabs
+                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 [transcript, gr.Tabs.update(selected=0), execution_info] + tab_updates
+    
+    # Modify the analyze_button.click setup
     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,
+            tabs_output,
+            execution_info_output
+        ] + [component for tab in tabs_output.children for component in tab.children],
         show_progress=True
     )