Update app.py

app.py

@@ -24,7 +24,7 @@ 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(results)
+    charts, explanations = create_charts(json.loads(results))
     progress(1.0, desc="Charts generation complete.")
     
     end_time = time.time()
@@ -33,29 +33,47 @@ def analyze_video(video_path, max_speakers, progress=gr.Progress()):
     output = {
         "transcript": transcription,
         "execution_info": f"Completed in {int(execution_time)} seconds.",
-        "speakers": {}
+        "speakers": charts,
+        "explanations": explanations
     }
     
-    for speaker_id, speaker_charts in charts.items():
-        speaker_explanations = explanations[speaker_id]
-        output["speakers"][speaker_id] = {
-            "attachment": {
-                "chart": speaker_charts["attachment"],
-                "explanation": speaker_explanations["attachment"]
-            },
-            "dimensions": speaker_charts["dimensions"],
-            "bigfive": {
-                "chart": speaker_charts["bigfive"],
-                "explanation": speaker_explanations["bigfive"]
-            },
-            "personality": {
-                "chart": speaker_charts["personality"],
-                "explanation": speaker_explanations["personality"]
-            }
-        }
-    
     return output
 
+def update_interface(result, max_speakers):
+    if "error" in result:
+        return [result["error"], gr.update(visible=False)] + [gr.update(visible=False)] * 42
+    
+    outputs = [
+        result["transcript"],  # Transcript
+        result["execution_info"]  # Execution info
+    ]
+    
+    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]
+            outputs.extend([
+                gr.update(value=speaker_data["attachment"], visible=True),
+                gr.update(visible=True),  # Column visibility
+                gr.update(value=speaker_explanations["attachment"], visible=True),
+                gr.update(visible=True),  # Column visibility
+                gr.update(value=speaker_data["dimensions"], visible=True),
+                gr.update(visible=True),  # Column visibility
+                gr.update(value=speaker_data["bigfive"], visible=True),
+                gr.update(visible=True),  # Column visibility
+                gr.update(value=speaker_explanations["bigfive"], visible=True),
+                gr.update(visible=True),  # Column visibility
+                gr.update(value=speaker_data["personality"], visible=True),
+                gr.update(visible=True),  # Column visibility
+                gr.update(value=speaker_explanations["personality"], visible=True),
+                gr.update(visible=True),  # Column visibility
+            ])
+        else:
+            outputs.extend([gr.update(visible=False)] * 14)  # 7 components + 7 columns
+    
+    return outputs
+
 def update_interface(result, max_speakers):
     if "error" in result:
         return [result["error"], gr.update(visible=False)] + [gr.update(visible=False)] * 42