Update app.py

app.py

@@ -8,6 +8,8 @@ matplotlib.rcParams['figure.dpi'] = 300
 matplotlib.rcParams['savefig.dpi'] = 300
 
 def process_and_show_completion(video_input_path, anomaly_threshold_input, fps, progress=gr.Progress()):
+    start_time = time.time()
+    
     try:
         print("Starting video processing...")
         results = process_video(video_input_path, anomaly_threshold_input, fps, progress=progress)
@@ -31,8 +33,11 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
 
         face_samples_frequent = [Image.open(path) for path in face_samples_frequent] if face_samples_frequent is not None else []
 
+        end_time = time.time()
+        total_exec_time = end_time - start_time
+
         output = [
-            exec_time, results_summary,
+            total_exec_time, results_summary,
             df, mse_embeddings, mse_posture, mse_voice,
             mse_plot_embeddings, mse_plot_posture, mse_plot_voice,
             mse_histogram_embeddings, mse_histogram_posture, mse_histogram_voice,
@@ -54,10 +59,16 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
         return [error_message] + [None] * 27
 
 def show_results():
-    return [gr.Tab.update(visible=True) for _ in range(4)] + [gr.Tab.update(visible=False)], gr.Group.update(visible=True)
+    return [gr.Tab.update(visible=True) for _ in range(4)] + [gr.Tab.update(visible=False)]
 
 def hide_description_show_results():
-    return [gr.Tab.update(visible=False)] + [gr.Tab.update(visible=True) for _ in range(4)]
+    return [gr.Tab.update(visible=False)] + [gr.Tab.update(visible=True) for _ in range(4)], gr.Group.update(visible=True)
+
+def start_execution_timer():
+    return gr.Number.update(value=0, visible=True), gr.Markdown.update(visible=True)
+
+def update_execution_time():
+    return gr.Number.update()
 
 with gr.Blocks() as iface:
     with gr.Row():
@@ -70,7 +81,8 @@ with gr.Blocks() as iface:
     
     execution_time_group = gr.Group(visible=False)
     with execution_time_group:
-        execution_time = gr.Number(label="Execution Time (seconds)")
+        execution_time = gr.Number(label="Execution Time", visible=False)
+        execution_time_text = gr.Markdown("Execution time will be displayed here", visible=False)
 
     with gr.Tabs() as all_tabs:
         description_tab = gr.Tab("Description")
@@ -121,9 +133,13 @@ with gr.Blocks() as iface:
     mse_heatmap_voice_store = gr.State()
     
     process_btn.click(
+        start_execution_timer,
+        inputs=None,
+        outputs=[execution_time, execution_time_text]
+    ).then(
         hide_description_show_results,
         inputs=None,
-        outputs=[description_tab, facial_features_tab, body_posture_tab, voice_tab, combined_tab]
+        outputs=[description_tab, facial_features_tab, body_posture_tab, voice_tab, combined_tab, execution_time_group]
     ).then(
         process_and_show_completion,
         inputs=[video_input, anomaly_threshold, fps_slider],
@@ -142,7 +158,14 @@ with gr.Blocks() as iface:
     ).then(
         show_results,
         inputs=None,
-        outputs=[all_tabs, execution_time_group]
+        outputs=all_tabs.children
+    )
+
+    execution_time.change(
+        update_execution_time,
+        inputs=None,
+        outputs=execution_time_text,
+        every=1
     )
 
 if __name__ == "__main__":