Ajout OpenWeb-UI fonctionnel

add_voice.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python
+
+import argparse
+import os
+import shutil
+import yaml
+
+print("!! WARNING EXPERIMENTAL !! - THIS TOOL WILL ERASE ALL COMMENTS FROM THE CONFIG FILES .. OR WORSE!!")
+
+parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+parser.add_argument('sample', action='store', help="Set the wav sample file")
+parser.add_argument('-n', '--name', action='store', help="Set the name for the voice (by default will use the WAV file name)")
+parser.add_argument('-l', '--language', action='store', default="auto", help="Set the language for the voice",
+                    choices=['auto', 'en', 'es', 'fr', 'de', 'it', 'pt', 'pl', 'tr', 'ru', 'nl', 'cs', 'ar', 'zh-cn', 'ja', 'hu', 'ko', 'hi'])
+parser.add_argument('--openai-model', action='store', default="tts-1-hd", help="Set the openai model for the voice")
+parser.add_argument('--xtts-model', action='store', default="xtts", help="Set the xtts model for the voice (if using a custom model, also set model_path)")
+parser.add_argument('--model-path', action='store', default=None, help="Set the path for a custom xtts model")
+parser.add_argument('--config-path', action='store', default="config/voice_to_speaker.yaml", help="Set the config file path")
+parser.add_argument('--voice-path', action='store', default="voices", help="Set the default voices file path")
+parser.add_argument('--default-path', action='store', default="voice_to_speaker.default.yaml", help="Set the default config file path")
+
+args = parser.parse_args()
+
+basename = os.path.basename(args.sample)
+name_noext, ext = os.path.splitext(basename)
+
+if not args.name:
+    args.name = name_noext
+else:
+    basename = f"{args.name}.wav"
+
+dest_file = os.path.join(args.voice_path, basename)
+if args.sample != dest_file:
+    shutil.copy2(args.sample, dest_file)
+
+if not os.path.exists(args.config_path):
+    shutil.copy2(args.default_path, args.config_path)
+
+with open(args.config_path, 'r', encoding='utf8') as file:
+    voice_map = yaml.safe_load(file)
+
+model_conf = voice_map.get(args.openai_model, {})
+model_conf[args.name] = {
+    'model': args.xtts_model,
+    'speaker': os.path.join(args.voice_path, basename),
+    'language': args.language,
+}
+if args.model_path:
+    model_conf[args.name]['model_path'] = args.model_path
+voice_map[args.openai_model] = model_conf
+
+with open(args.config_path, 'w', encoding='utf8') as ofile:
+    yaml.safe_dump(voice_map, ofile, default_flow_style=False, allow_unicode=True)
+
+print(f"Updated: {args.config_path}")
+print(f"Added voice: {args.openai_model}/{args.name}")
+print(f"Added section:")
+print(f"{args.openai_model}:")
+print(f"  {args.name}:")
+print(f"    model: {model_conf[args.name]['model']}")
+print(f"    speaker: {model_conf[args.name]['speaker']}")
+print(f"    language: {model_conf[args.name]['language']}")