Spaces:
Sleeping
Sleeping
| 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', True) | |
| 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', 8) | |
| self.provider = self.default_get(data, 'provider', 'cuda') | |
| 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, | |
| 'provider' : self.provider | |
| } | |
| with open(self.config_file, 'w') as f: | |
| yaml.dump(data, f) | |