65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
import os
|
|
import folder_paths
|
|
now_dir = os.path.dirname(os.path.abspath(__file__))
|
|
input_dir = folder_paths.get_input_directory()
|
|
output_dir = folder_paths.get_output_directory()
|
|
|
|
class LoadFile:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
|
|
return {"required":
|
|
{"file": (sorted(files),)},
|
|
}
|
|
|
|
CATEGORY = "AIFSH_RealisDance"
|
|
|
|
RETURN_TYPES = ("PATH",)
|
|
FUNCTION = "load_srt"
|
|
|
|
def load_srt(self, file):
|
|
srt_path = folder_paths.get_annotated_filepath(file)
|
|
return (srt_path,)
|
|
|
|
class LoadVideo:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f)) and f.split('.')[-1] in ["mp4", "webm","mkv","avi"]]
|
|
return {"required":{
|
|
"video":(files,),
|
|
}}
|
|
|
|
CATEGORY = "AIFSH_RealisDance"
|
|
DESCRIPTION = "hello world!"
|
|
|
|
RETURN_TYPES = ("VIDEO",)
|
|
|
|
OUTPUT_NODE = False
|
|
|
|
FUNCTION = "load_video"
|
|
|
|
def load_video(self, video):
|
|
video_path = os.path.join(input_dir,video)
|
|
return (video_path,)
|
|
|
|
class PreViewVideo:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required":{
|
|
"video":("VIDEO",),
|
|
}}
|
|
|
|
CATEGORY = "AIFSH_RealisDance"
|
|
DESCRIPTION = "hello world!"
|
|
|
|
RETURN_TYPES = ()
|
|
|
|
OUTPUT_NODE = True
|
|
|
|
FUNCTION = "load_video"
|
|
|
|
def load_video(self, video):
|
|
video_name = os.path.basename(video)
|
|
video_path_name = os.path.basename(os.path.dirname(video))
|
|
return {"ui":{"video":[video_name,video_path_name]}}
|