Spaces:
Configuration error
Configuration error
File size: 2,849 Bytes
632bd54 |
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 |
import yaml
class Settings:
def __init__(self, config_file):
self.config_file = config_file
self.load()
def default_get(_, data, name, default):
value = default
try:
value = data.get(name, default)
except:
pass
return value
def load(self):
try:
with open(self.config_file, "r") as f:
data = yaml.load(f, Loader=yaml.FullLoader)
except:
data = None
self.selected_theme = self.default_get(data, "selected_theme", "Default")
self.server_name = self.default_get(data, "server_name", "")
self.server_port = self.default_get(data, "server_port", 0)
self.server_share = self.default_get(data, "server_share", False)
self.output_image_format = self.default_get(data, "output_image_format", "png")
self.output_video_format = self.default_get(data, "output_video_format", "mp4")
self.output_video_codec = self.default_get(
data, "output_video_codec", "libx264"
)
self.video_quality = self.default_get(data, "video_quality", 14)
self.clear_output = self.default_get(data, "clear_output", True)
self.max_threads = self.default_get(data, "max_threads", 2)
self.memory_limit = self.default_get(data, "memory_limit", 0)
self.provider = self.default_get(data, "provider", "cuda")
self.force_cpu = self.default_get(data, "force_cpu", False)
self.output_template = self.default_get(
data, "output_template", "{file}_{time}"
)
self.use_os_temp_folder = self.default_get(data, "use_os_temp_folder", False)
self.output_show_video = self.default_get(data, "output_show_video", True)
self.launch_browser = self.default_get(data, "launch_browser", True)
def save(self):
data = {
"selected_theme": self.selected_theme,
"server_name": self.server_name,
"server_port": self.server_port,
"server_share": self.server_share,
"output_image_format": self.output_image_format,
"output_video_format": self.output_video_format,
"output_video_codec": self.output_video_codec,
"video_quality": self.video_quality,
"clear_output": self.clear_output,
"max_threads": self.max_threads,
"memory_limit": self.memory_limit,
"provider": self.provider,
"force_cpu": self.force_cpu,
"output_template": self.output_template,
"use_os_temp_folder": self.use_os_temp_folder,
"output_show_video": self.output_show_video,
}
with open(self.config_file, "w") as f:
yaml.dump(data, f)
|