Update to `ruamel.yaml`

modules/utils/files_manager.py

@@ -1,39 +1,30 @@
 import os
 import fnmatch
-import yaml
+from ruamel.yaml import YAML
 from gradio.utils import NamedString
 
 from modules.utils.paths import DEFAULT_PARAMETERS_CONFIG_PATH
 
 
-class YAMLDumper(yaml.SafeDumper):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.add_representer(str, self.str_presenter)
-
-    def write_line_break(self, data=None):
-        super().write_line_break(data)
-        if len(self.indents) == 1:
-            super().write_line_break()
-
-    @staticmethod
-    def str_presenter(dumper, data):
-        special_chars = set(' \n\t:-[]{},&*#?|>!%@`"\'')
-        if any(char in special_chars for char in data) or data == '':
-            return dumper.represent_scalar('tag:yaml.org,2002:str', data, style='"')
-        return dumper.represent_scalar('tag:yaml.org,2002:str', data)
-
-
 def load_yaml(path: str = DEFAULT_PARAMETERS_CONFIG_PATH):
+    yaml = YAML(typ="safe")
+    yaml.preserve_quotes = True
     with open(path, 'r', encoding='utf-8') as file:
-        config = yaml.safe_load(file)
+        config = yaml.load(file)
     return config
 
 
 def save_yaml(data: dict, path: str = DEFAULT_PARAMETERS_CONFIG_PATH):
-    data = yaml.dump(data, sort_keys=False, default_flow_style=False, Dumper=YAMLDumper, allow_unicode=True)
+    yaml = YAML(typ="safe")
+    yaml.map_indent = 2
+    yaml.sequence_indent = 4
+    yaml.sequence_dash_offset = 2
+    yaml.preserve_quotes = True
+    yaml.default_flow_style = False
+    yaml.sort_base_mapping_type_on_output = False
+
     with open(path, 'w', encoding='utf-8') as file:
-        file.write(data)
+        yaml.dump(data, file)
     return path