Update app.py

app.py

@@ -7,6 +7,7 @@ import huggingface_hub
 import numpy as np
 import onnxruntime as rt
 import pandas as pd
+import time
 from PIL import Image
 
 TITLE = "Video Tagger"
@@ -67,6 +68,20 @@ kaomojis = [
 ]
 
 
+def _format_duration(seconds: float) -> str:
+    """
+    Format a duration in seconds as MM:SS or HH:MM:SS.
+    """
+    total_seconds = int(round(seconds))
+    hours, rem = divmod(total_seconds, 3600)
+    minutes, secs = divmod(rem, 60)
+
+    if hours > 0:
+        return f"{hours:02d}:{minutes:02d}:{secs:02d}"
+    else:
+        return f"{minutes:02d}:{secs:02d}"
+
+
 def load_labels(df: pd.DataFrame):
     """
     Convert tag dataframe into:
@@ -587,6 +602,8 @@ def tag_video_interface(
     if video_path is None:
         return "", {"error": "Please upload a video file."}
 
+    start_time = time.time()
+
     try:
         # Get or create the cached VideoTagger for this model,
         # and update its batch size for this run.
@@ -595,7 +612,7 @@ def tag_video_interface(
         tag_substitutes = _normalize_tag_substitutes(tag_substitutes_df)
         tag_exclusions = _normalize_tag_exclusions(tag_exclusions_df)
 
-        return tagger.tag_video(
+        combined_tags_str, debug_info = tagger.tag_video(
             video_path=video_path,
             frame_interval=frame_interval,
             general_thresh=general_thresh,
@@ -604,6 +621,13 @@ def tag_video_interface(
             tag_exclusions=tag_exclusions,
             progress=progress,
         )
+
+        elapsed = time.time() - start_time
+        debug_info["session_duration_seconds"] = round(elapsed, 3)
+        debug_info["session_duration_hms"] = _format_duration(elapsed)
+
+        return combined_tags_str, debug_info
+
     except Exception as e:
         return "", {"error": str(e)}