Update app.py

app.py

@@ -10,6 +10,7 @@ from config import openai_api_key
 # Load the model
 llm = load_model(openai_api_key)
 
+
 def analyze_video(video_path, progress=gr.Progress()):
     start_time = time.time()
     if not video_path:
@@ -40,57 +41,55 @@ def analyze_video(video_path, progress=gr.Progress()):
         print(speaker_id)
         speaker_explanations = explanations[speaker_id]
         speaker_general_impression = general_impressions[speaker_id]
-        
+
         with gr.Tab():
-            with gr.TabItem(label=f'General Impression'):  
+            with gr.TabItem(label=f'General Impression'):
                 speaker_section1 = [
                     gr.Markdown(f"### {speaker_id}", visible=True),
                     gr.Textbox(value=speaker_general_impression, label="General Impression", visible=True)
                 ]
-                
+
             with gr.TabItem(label=f'Attachment Styles'):
-                with gr.Row():
-                    speaker_section2_1 = [
-                        gr.Column(scale=2, components=[
-                            gr.Plot(value=speaker_charts.get("attachment", None), visible=True)
-                        ]),
-                        gr.Column(scale=1, components=[
-                            gr.Plot(value=speaker_charts.get("dimensions", None), visible=True)
-                        ])
-                    ]
-                     
-                speaker_section2_2 = [gr.Textbox(value=speaker_explanations.get("attachment", ""), label="Attachment Styles Explanation", visible=True)]
-            
-            with gr.TabItem(label=f'Big Five Traits'):  
+                speaker_section2 = [
+                    gr.Plot(value=speaker_charts.get("attachment", None), visible=True),
+                    gr.Plot(value=speaker_charts.get("dimensions", None), visible=True),
+                    gr.Textbox(value=speaker_explanations.get("attachment", ""), label="Attachment Styles Explanation",
+                               visible=True)
+                ]
+
+            with gr.TabItem(label=f'Big Five Traits'):
                 speaker_section3 = [
                     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.Textbox(value=speaker_explanations.get("bigfive", ""), label="Big Five Traits Explanation",
+                               visible=True)
                 ]
-            
-            with gr.TabItem(label=f'Personalities'):  
-                speaker_section4 = [    
+
+            with gr.TabItem(label=f'Personalities'):
+                speaker_section4 = [
                     gr.Plot(value=speaker_charts.get("personality", None), visible=True),
-                    gr.Textbox(value=speaker_explanations.get("personality", ""),  label="Personality Disorders Explanation", visible=True)
+                    gr.Textbox(value=speaker_explanations.get("personality", ""),
+                               label="Personality Disorders Explanation", visible=True)
                 ]
-                
+
         output_components.extend(speaker_section1)
-        output_components.extend(speaker_section2_1)
-        output_components.extend(speaker_section2_2)
+        output_components.extend(speaker_section2)
         output_components.extend(speaker_section3)
         output_components.extend(speaker_section4)
 
     # Pad with None for any missing speakers
     while len(output_components) < 28:
         output_components.extend([gr.update(visible=False)] * 9)
-    
+
     return output_components
 
+
 def use_example():
     return "examples/Scenes.From.A.Marriage.US.mp4"
 
+
 with gr.Blocks() as iface:
     gr.Markdown("# AI Personality Detection")
-    
+
     with gr.Row():
         with gr.Column(scale=3):
             gr.Markdown("Upload a video")
@@ -110,39 +109,31 @@ with gr.Blocks() as iface:
     transcript = gr.Textbox(label="Transcript", lines=10, visible=False)
     output_components.append(transcript)
 
-    
     for n in range(3):  # Assuming maximum of 3 speakers
-        with gr.Tab(label=f'Speaker {n+1}', visible=True):
-            with gr.TabItem(label=f'General Impression'):                
+        with gr.Tab(label=f'Speaker {n + 1}', visible=True):
+            with gr.TabItem(label=f'General Impression'):
                 column_components1 = [
                     gr.Markdown(visible=False),
                     gr.Textbox(label="General Impression", visible=False)]
-                    
+
             with gr.TabItem(label=f'Attachment Styles'):
-                with gr.Row():
-                    column_components2_1 = [
-                        gr.Column(scale=2, components=[
-                            gr.Plot(visible=False)
-                        ]),
-                        gr.Column(scale=1, components=[
-                            gr.Plot(visible=False)
-                        ])
-                    ]
-                column_components2_2 = [gr.Textbox(label="Attachment Styles Explanation", visible=False)]
-                
+                column_components2 = [
+                    gr.Plot(visible=False),
+                    gr.Plot(visible=False),
+                    gr.Textbox(label="Attachment Styles Explanation", visible=False)]
+
             with gr.TabItem(label=f'Big Five Traits'):
                 column_components3 = [
                     gr.Plot(visible=False),
                     gr.Textbox(label="Big Five Traits Explanation", visible=False)]
-                    
+
             with gr.TabItem(label=f'Personalities'):
                 column_components4 = [
                     gr.Plot(visible=False),
                     gr.Textbox(label="Personality Disorders Explanation", visible=False)]
-                
+
         output_components.extend(column_components1)
-        output_components.extend(column_components2_1)
-        output_components.extend(column_components2_2)
+        output_components.extend(column_components2)
         output_components.extend(column_components3)
         output_components.extend(column_components4)
 
@@ -158,10 +149,10 @@ with gr.Blocks() as iface:
         inputs=[],
         outputs=[video_input],
     ).then(fn=analyze_video,
-        inputs=[video_input],
-        outputs=output_components,
-        show_progress=True
-    )
+           inputs=[video_input],
+           outputs=output_components,
+           show_progress=True
+           )
 
 if __name__ == "__main__":
     iface.launch()