utils/gradio_utils.py

import os

# App Pose utils
def motion_to_video_path(motion):
    videos = [
        "__assets__/dance1_corr.mp4",
        "__assets__/dance2_corr.mp4",
        "__assets__/dance3_corr.mp4",
        "__assets__/dance4_corr.mp4",
        "__assets__/dance5_corr.mp4"
    ]
    if len(motion.split(" ")) > 1 and motion.split(" ")[1].isnumeric():
        id = int(motion.split(" ")[1]) - 1
        return videos[id]
    else:
        return motion

def logo_name_to_path(name):
    logo_paths = {
        'Picsart AI Research': '__assets__/pair_watermark.png',
        'Text2Video-Zero': '__assets__/t2v-z_watermark.png',
        'None': None
    }
    if name in logo_paths:
        return logo_paths[name]
    return name