Function get_prompt_from_filename

mbench/utils.py:8–30

Extract prompt from video filename by removing extensions and special characters Args: filename: Video filename Returns: Extracted prompt string

(filename: str)

def get_prompt_from_filename(filename: str) -> str:
    """
    Extract prompt from video filename by removing extensions and special characters
    
    Args:
        filename: Video filename
        
    Returns:
        Extracted prompt string
    """
    # Remove file extension
    name = Path(filename).stem
    
    # Remove common suffixes like "-0", "-1", etc.
    name = re.sub(r'-\d+$', '', name)
    
    # Replace underscores and hyphens with spaces
    name = re.sub(r'[_-]', ' ', name)
    
    # Clean up multiple spaces
    name = re.sub(r'\s+', ' ', name).strip()
    
    return name.lower()


def save_json(data: Union[Dict, List], filepath: str) -> None: