FlowState-Creator-Nodes/FS_Assets.py at main · flowstatelabs/FlowState-Creator-Nodes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# Project: FlowState Assets
# Description: Paths to assets needed by nodes.
# Author: Johnathan Chivington
# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng


##
# OUTSIDE IMPORTS
##
from .FS_Utils import *


##
# OUTSIDE IMPORTS
##
import os
import comfy
import folder_paths


##
# ASSETS
##
WEB_DIRECTORY = './web'
FONT_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'fonts/ShareTechMono-Regular.ttf')


DIFFUSION_MODELS_LIST = lambda: sorted(set(folder_paths.get_filename_list('unet') + folder_paths.get_filename_list('diffusion_models')))
CHECKPOINTS_LIST = lambda: sorted(set(folder_paths.get_filename_list('checkpoints')))
CLIPS_LIST = lambda: sorted(set(folder_paths.get_filename_list('clip')))
VAES_LIST = lambda: sorted(set(get_vae_list()))

CONTROL_NETS_LIST = lambda: sorted(set(folder_paths.get_filename_list('controlnet')))
LORAS_LIST = lambda: sorted(set(folder_paths.get_filename_list('loras')))

SAMPLERS_LIST = lambda: sorted(set(comfy.samplers.KSampler.SAMPLERS))
SCHEDULERS_LIST = lambda: sorted(set(comfy.samplers.KSampler.SCHEDULERS))

ALL_MODELS_LIST = lambda: sorted(set(DIFFUSION_MODELS_LIST() + CHECKPOINTS_LIST()))

INPUT_FILES = lambda: get_input_files()
OUTPUT_FILES = lambda: get_output_files()