add a time tracker

app.py

@@ -6,6 +6,7 @@ from huggingface_hub import hf_hub_download
 import glob
 from moviepy.editor import VideoFileClip
 from datetime import datetime
+import time
 
 is_shared_ui = True if "fffiloni/X-Portrait" in os.environ['SPACE_ID'] else False
 
@@ -93,6 +94,8 @@ def extract_frames_with_labels(video_path, base_output_dir="frames"):
 # Define a function to run your script with selected inputs
 def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_frames, num_mix, ddim_steps, progress=gr.Progress(track_tqdm=True)):
 
+    start_time = time.perf_counter()
+    
     # Create a unique output directory name based on current date and time
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     output_dir = f"output_{timestamp}"
@@ -126,11 +129,16 @@ def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_f
         print(video_files)
         if video_files:
             final_vid = convert_video_to_h264_aac(video_files[0])
+            
             return f"Output video saved at: {final_vid}", final_vid
         else:
             return "No video file was found in the output directory.", None
     except subprocess.CalledProcessError as e:
         return f"An error occurred: {e}", None
+    finally:
+        end_time = time.perf_counter()
+        elapsed = end_time - start_time
+        print(f"[LOG] Execution time: {elapsed:.4f} seconds")        
 
 def convert_video_to_h264_aac(video_path):
     # Get the directory and original filename