Update app.py

app.py

@@ -58,30 +58,14 @@ def analyze_video(video_path, progress=gr.Progress()):
         
         # Pad with None for any missing speakers
         while len(output_components) < 28:
-            output_components.append(None)
+            output_components.extend([gr.update(visible=False)] * 9)
                 
     output_components.append(f"Completed in {int(execution_time)} seconds.")  # execution info
 
     return output_components
 
 def update_output(*args):
-    updated_components = []
-    for i in range(0, len(args) - 1, 10):  # 9 components per speaker + 1 column
-        column_components = args[i:i+9]
-        column = args[i+9]
-        
-        if any(comp is not None for comp in column_components):
-            updated_components.extend([gr.update(value=arg, visible=arg is not None) for arg in column_components])
-            updated_components.append(gr.update(visible=True))  # Make the column visible
-        else:
-            updated_components.extend([gr.update(visible=False) for _ in range(9)])
-            updated_components.append(gr.update(visible=False))  # Hide the empty column
-    
-    # Handle transcript and execution info
-    updated_components.insert(0, gr.update(value=args[0], visible=args[0] is not None))
-    updated_components.append(gr.update(value=args[-1], visible=args[-1] is not None))
-    
-    return updated_components
+    return [gr.update(value=arg, visible=arg is not None) for arg in args]
 
 def use_example():
     return "examples/Scenes.From.A.Marriage.US.mp4"
@@ -105,10 +89,9 @@ with gr.Blocks() as iface:
     # Add transcript output near the top
     execution_info_box = gr.Textbox(label="Transcript", value="N/A", lines=1)
     output_components.append(execution_info_box)
-
     with gr.Row():
         for i in range(3):  # Assuming maximum of 3 speakers
-            with gr.Column(visible=False) as column:
+            with gr.Column():
                     column_components = [
                         gr.Markdown(visible=False),
                         gr.Textbox(label="General Impression", visible=False),
@@ -121,7 +104,6 @@ with gr.Blocks() as iface:
                         gr.Textbox(label="Personality Disorders Explanation", visible=False),
                     ]
             output_components.extend(column_components)
-            output_components.append(column)
         
     # Add execution info component
     transcript_output = gr.Textbox(label="Transcript", lines=10, visible=False)
@@ -132,10 +114,6 @@ with gr.Blocks() as iface:
         inputs=[video_input],
         outputs=output_components,
         show_progress=True
-    ).then(
-        fn=update_output,
-        inputs=output_components,
-        outputs=output_components
     )
 
     use_example_button.click(