Update app.py

app.py

@@ -5,7 +5,6 @@ from transcription_diarization import diarize_audio
 from visualization import create_charts
 import time
 from config import openai_api_key
-import json
 
 # Load the model
 llm = load_model(openai_api_key)
@@ -25,12 +24,9 @@ def analyze_video(video_path, max_speakers, progress=gr.Progress()):
     progress(0.7, desc="Transcription processing complete.")
     
     progress(0.9, desc="Generating charts")
-    charts, explanations = create_charts(json.loads(results))
+    charts, explanations = create_charts(results)
     progress(1.0, desc="Charts generation complete.")
     
-    end_time = time.time()
-    execution_time = end_time - start_time
-    
     output = {
         "transcript": transcription,
         "speakers": charts,
@@ -41,7 +37,7 @@ def analyze_video(video_path, max_speakers, progress=gr.Progress()):
 
 def update_interface(result, max_speakers):
     if "error" in result:
-        return [result["error"], gr.update(visible=False)] + [gr.update(visible=False)] * 46
+        return [result["error"]] + [gr.update(visible=False)] * 47
 
     outputs = [result["transcript"]]  # Transcript
     
@@ -49,10 +45,9 @@ def update_interface(result, max_speakers):
     outputs.extend([gr.update(visible=True) for _ in range(4)])  # Make section titles visible
     
     for i in range(3):
-        speaker_id = i
-        if i < max_speakers and speaker_id in result["speakers"]:
-            speaker_data = result["speakers"][speaker_id]
-            speaker_explanations = result["explanations"][speaker_id]
+        if i < max_speakers and i in result["speakers"]:
+            speaker_data = result["speakers"][i]
+            speaker_explanations = result["explanations"][i]
             outputs.extend([
                 gr.update(value=speaker_data["attachment"], visible=True),
                 gr.update(visible=True),  # Column visibility
@@ -79,13 +74,11 @@ with gr.Blocks() as iface:
     gr.Markdown("Upload a video")
     
     with gr.Row():
-        with gr.Column(scale=1):
-            video_input = gr.Video(label="Upload Video")
-            max_speakers = gr.Slider(minimum=1, maximum=3, step=1, value=2, label="Maximum Number of Speakers")
-            analyze_button = gr.Button("Analyze")
-        
-        with gr.Column(scale=2):
-            transcript_output = gr.Textbox(label="Transcript", lines=10)
+        video_input = gr.Video(label="Upload Video")
+        max_speakers = gr.Slider(minimum=1, maximum=3, step=1, value=2, label="Maximum Number of Speakers")
+    
+    analyze_button = gr.Button("Analyze")
+    transcript_output = gr.Textbox(label="Transcript", lines=10)
 
     # Section titles (initially hidden)
     attachment_styles_title = gr.Markdown("## Attachment Styles", visible=False)