Update app.py

app.py

@@ -70,7 +70,67 @@ def use_example():
     return "examples/Scenes.From.A.Marriage.US.mp4"
 
 with gr.Blocks() as iface:
-    # ... (rest of the interface definition remains the same)
+    gr.Markdown("# AI Personality Detection")
+    
+    with gr.Row():
+        with gr.Column(scale=3):
+            gr.Markdown("Upload a video")
+            video_input = gr.Video(label="Upload Video")
+            analyze_button = gr.Button("Analyze")
+        with gr.Column(scale=1):
+            gr.Markdown("Example Video")
+            example_video = gr.Video("examples/Scenes.From.A.Marriage.US.mp4", label="Example Video")
+            use_example_button = gr.Button("Use Example Video")
+
+    # Create placeholder components for output
+    with gr.Column() as output_container:
+        transcript_output = gr.Textbox(label="Transcript", lines=10, visible=False)
+        speaker_outputs = []
+        for i in range(1, 4):  # Assuming a maximum of 3 speakers
+            with gr.Column(visible=False) as speaker_column:
+                speaker_header = gr.Markdown(f"## Speaker {i}")
+                speaker_impression = gr.Textbox(label="General Impression", lines=3)
+                speaker_attachment = gr.Plot(label="Attachment Styles")
+                speaker_attachment_exp = gr.Textbox(label="Attachment Styles Explanation")
+                speaker_dimensions = gr.Plot(label="Attachment Dimensions")
+                speaker_bigfive = gr.Plot(label="Big Five Traits")
+                speaker_bigfive_exp = gr.Textbox(label="Big Five Traits Explanation")
+                speaker_personality = gr.Plot(label="Personality Disorders")
+                speaker_personality_exp = gr.Textbox(label="Personality Disorders Explanation")
+            speaker_outputs.extend([
+                speaker_header, speaker_impression, speaker_attachment, speaker_attachment_exp,
+                speaker_dimensions, speaker_bigfive, speaker_bigfive_exp, speaker_personality,
+                speaker_personality_exp
+            ])
+        execution_info = gr.Textbox(label="Execution Information", visible=True)
+
+    all_outputs = [transcript_output] + speaker_outputs + [execution_info]
+
+    analyze_button.click(
+        fn=analyze_video,
+        inputs=[video_input],
+        outputs=all_outputs,
+        show_progress=True
+    ).then(
+        fn=update_output,
+        inputs=all_outputs,
+        outputs=all_outputs
+    )
+
+    use_example_button.click(
+        fn=use_example,
+        inputs=[],
+        outputs=[video_input],
+    ).then(
+        fn=analyze_video,
+        inputs=[video_input],
+        outputs=all_outputs,
+        show_progress=True
+    ).then(
+        fn=update_output,
+        inputs=all_outputs,
+        outputs=all_outputs
+    )
 
 if __name__ == "__main__":
     iface.launch()