Spaces:
Runtime error
Runtime error
File size: 2,553 Bytes
87c5489 |
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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
import os
import constants
from pathlib import Path
from time import time
from utils import get_image_file_extension
def join_paths(
first_path: str,
second_path: str,
) -> str:
return os.path.join(first_path, second_path)
def get_file_name(file_path: str) -> str:
return Path(file_path).stem
def get_app_path() -> str:
app_dir = os.path.dirname(__file__)
work_dir = os.path.dirname(app_dir)
return work_dir
def get_configs_path() -> str:
config_path = join_paths(get_app_path(), constants.CONFIG_DIRECTORY)
return config_path
class FastStableDiffusionPaths:
@staticmethod
def get_app_settings_path() -> str:
configs_path = get_configs_path()
settings_path = join_paths(
configs_path,
constants.APP_SETTINGS_FILE,
)
return settings_path
@staticmethod
def get_results_path() -> str:
results_path = join_paths(get_app_path(), constants.RESULTS_DIRECTORY)
return results_path
@staticmethod
def get_css_path() -> str:
app_dir = os.path.dirname(__file__)
css_path = os.path.join(
app_dir,
"frontend",
"webui",
"css",
"style.css",
)
return css_path
@staticmethod
def get_models_config_path(model_config_file: str) -> str:
configs_path = get_configs_path()
models_path = join_paths(
configs_path,
model_config_file,
)
return models_path
@staticmethod
def get_upscale_filepath(
file_path_src: str,
scale_factor: int,
format: str,
) -> str:
if file_path_src:
file_name_src = get_file_name(file_path_src)
else:
file_name_src = "fastsdcpu"
extension = get_image_file_extension(format)
upscaled_filepath = join_paths(
FastStableDiffusionPaths.get_results_path(),
f"{file_name_src}_{int(scale_factor)}x_upscale_{int(time())}{extension}",
)
return upscaled_filepath
@staticmethod
def get_lora_models_path() -> str:
lora_models_path = join_paths(get_app_path(), constants.LORA_DIRECTORY)
return lora_models_path
@staticmethod
def get_controlnet_models_path() -> str:
controlnet_models_path = join_paths(
get_app_path(), constants.CONTROLNET_DIRECTORY
)
return controlnet_models_path
def get_base_folder_name(path: str) -> str:
return os.path.basename(path)
|