Update app.py

app.py

@@ -10,40 +10,26 @@ from config import openai_api_key
 llm = load_model(openai_api_key)
 
 def analyze_video(video_path, max_speakers, progress=gr.Progress()):
-    start_time = time.time()
-    if not video_path:
-        return {"error": "Please upload a video file."}
-
-    progress(0, desc="Starting analysis...")
-    progress(0.2, desc="Starting transcription and diarization")
-    transcription = diarize_audio(video_path, max_speakers)
-    print("Transcription:", transcription)  # Debug print
-    progress(0.5, desc="Transcription and diarization complete.")
-
-    progress(0.6, desc="Processing transcription")
-    results = process_input(transcription, llm)
-    progress(0.7, desc="Transcription processing complete.")
-
-    progress(0.9, desc="Generating charts")
-    charts, explanations = create_charts(results)
-    progress(1.0, desc="Charts generation complete.")
-
-    end_time = time.time()
-    execution_time = end_time - start_time
-
-    return {
-        "transcript": transcription,
-        "charts": charts,
-        "explanations": explanations,
-        "execution_time": int(execution_time)
-    }
+    # ... (keep the existing analyze_video function as is)
 
 def create_output_components():
     with gr.Row() as row:
         with gr.Column():
-            transcript = gr.Textbox(label="Transcript", lines=10)
+            transcript = gr.Textbox(label="Transcript", lines=10, visible=False)
+            
             tabs = gr.Tabs()
-            execution_info = gr.Textbox(label="Execution Information")
+            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)
+            
+            execution_info = gr.Textbox(label="Execution Information", visible=False)
     return row, transcript, tabs, execution_info
 
 with gr.Blocks() as iface:
@@ -61,35 +47,41 @@ with gr.Blocks() as iface:
         
         if "error" in results:
             return {
-                transcript_output: "",
-                tabs_output: gr.Tabs(),
-                execution_info_output: results["error"]
+                transcript_output: gr.update(value="", visible=False),
+                tabs_output: gr.update(visible=False),
+                execution_info_output: gr.update(value=results["error"], visible=True)
             }
 
-        tabs_list = []
-        for speaker_id, speaker_charts in results["charts"].items():
+        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]
-            with gr.Tab(label=speaker_id) as tab:
-                gr.Markdown(f"## {speaker_id}\n<hr style='height:2px;border-width:0;color:black;background-color:black'>", visible=True)
-                gr.Plot(value=speaker_charts.get("attachment", None), visible=True)
-                gr.Textbox(value=speaker_explanations.get("attachment", ""), label="Attachment Styles Explanation", visible=True)
-                gr.Plot(value=speaker_charts.get("dimensions", None), 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)
-            tabs_list.append(tab)
+            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)
 
-        return {
-            transcript_output: results["transcript"],
-            tabs_output: gr.Tabs(*tabs_list),
-            execution_info_output: f"Completed in {results['execution_time']} seconds."
-        }
+        # Hide unused tabs
+        for i in range(len(results["charts"]), 3):
+            updates[f"Speaker {i+1}"] = gr.update(visible=False)
+
+        return updates
 
     analyze_button.click(
         fn=run_analysis,
         inputs=[video_input, max_speakers],
-        outputs=[transcript_output, tabs_output, execution_info_output],
+        outputs=[transcript_output, tabs_output, execution_info_output] + [component for tab in tabs_output.children for component in tab.children],
         show_progress=True
     )