add automatic FPS tracking

finetrainers_utils.py

@@ -4,7 +4,7 @@ import logging
 import shutil
 from typing import Any, Optional, Dict, List, Union, Tuple
 from config import STORAGE_PATH, TRAINING_PATH, STAGING_PATH, TRAINING_VIDEOS_PATH, MODEL_PATH, OUTPUT_PATH, HF_API_TOKEN, MODEL_TYPES
-from utils import extract_scene_info, make_archive, is_image_file, is_video_file
+from utils import get_video_fps, extract_scene_info, make_archive, is_image_file, is_video_file
 
 logger = logging.getLogger(__name__)
 
@@ -110,6 +110,14 @@ def copy_files_to_training_dir(prompt_prefix: str) -> int:
         if parent_caption and not caption.endswith(parent_caption):
             caption = f"{caption}\n{parent_caption}"
 
+        # Add FPS information for videos
+        if is_video_file(file_path) and caption:
+            # Only add FPS if not already present
+            if not any(f"FPS, " in line for line in caption.split('\n')):
+                fps_info = get_video_fps(file_path)
+                if fps_info:
+                    caption = f"{fps_info}{caption}"
+
         if prompt_prefix and not caption.startswith(prompt_prefix):
             caption = f"{prompt_prefix}{caption}"
             

utils.py

@@ -1,11 +1,15 @@
 import os
 import shutil
+import subprocess
 from huggingface_hub import HfApi, create_repo
 from pathlib import Path
 import json
 import re
+import logging
 from typing import Any, Optional, Dict, List, Union, Tuple
 
+logger = logging.getLogger(__name__)
+
 def make_archive(source: str | Path, destination: str | Path):
     source = str(source)
     destination = str(destination)
@@ -18,6 +22,44 @@ def make_archive(source: str | Path, destination: str | Path):
     shutil.make_archive(name, format, archive_from, archive_to)
     shutil.move('%s.%s'%(name,format), destination)
 
+def get_video_fps(video_path: Path) -> Optional[str]:
+    """Get FPS information from video file using ffprobe
+    
+    Args:
+        video_path: Path to video file
+        
+    Returns:
+        FPS string (e.g. "24 FPS, ") or None if unable to determine
+    """
+    try:
+        cmd = [
+            'ffprobe',
+            '-v', 'error',
+            '-select_streams', 'v:0',
+            '-show_entries', 'stream=avg_frame_rate',
+            '-of', 'default=noprint_wrappers=1:nokey=1',
+            str(video_path)
+        ]
+        
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        if result.returncode != 0:
+            logger.warning(f"Error getting FPS for {video_path}: {result.stderr}")
+            return None
+            
+        fps = result.stdout.strip()
+        if '/' in fps:
+            # Convert fraction to decimal
+            num, den = map(int, fps.split('/'))
+            if den == 0:
+                return None
+            fps = str(round(num / den))
+            
+        return f"{fps} FPS, "
+        
+    except Exception as e:
+        logger.warning(f"Failed to get FPS for {video_path}: {e}")
+        return None
+
 def extract_scene_info(filename: str) -> Tuple[str, Optional[int]]:
     """Extract base name and scene number from filename