| import yaml | |
| class Settings: | |
| def __init__(self, config_file): | |
| self.config_file = config_file | |
| self.load() | |
| def load(self): | |
| try: | |
| with open(self.config_file, 'r') as f: | |
| data = yaml.load(f, Loader=yaml.FullLoader) | |
| self.selected_theme = data.get('selected_theme', "gstaff/xkcd") | |
| self.server_name = data.get('server_name', "") | |
| self.server_port = data.get('server_port', 0) | |
| self.server_share = data.get('server_share', False) | |
| self.input_text_desired_length = data.get('input_text_desired_length', 110) | |
| self.input_text_max_length = data.get('input_text_max_length', 170) | |
| self.silence_sentence = data.get('silence_between_sentences', 250) | |
| self.silence_speakers = data.get('silence_between_speakers', 500) | |
| self.output_folder_path = data.get('output_folder_path', 'outputs') | |
| except: | |
| self.selected_theme = "gstaff/xkcd" | |
| def save(self): | |
| data = { | |
| 'selected_theme': self.selected_theme, | |
| 'server_name': self.server_name, | |
| 'server_port': self.server_port, | |
| 'server_share': self.server_share, | |
| 'input_text_desired_length' : self.input_text_desired_length, | |
| 'input_text_max_length' : self.input_text_max_length, | |
| 'silence_between_sentences': self.silence_sentence, | |
| 'silence_between_speakers': self.silence_speakers, | |
| 'output_folder_path': self.output_folder_path | |
| } | |
| with open(self.config_file, 'w') as f: | |
| yaml.dump(data, f) | |