|
import os |
|
|
|
|
|
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 |
|
|