VERSION TACOTRON 2

data/generate_audio.py

@@ -2,35 +2,42 @@
 import sys
 import os
 import json
-import wave 
+import wave
 from TTS.api import TTS
 
-if len(sys.argv) < 3: 
-    print("Usage: python3 generate_audio.py \"Votre texte ici\" /chemin/vers/fichier.wav", file=sys.stderr)
+if len(sys.argv) < 3:
+    print(
+        "Usage: python3 generate_audio.py \"Votre texte ici\" /chemin/vers/fichier.wav",
+        file=sys.stderr
+    )
     sys.exit(1)
 
-text = sys.argv[1].replace('"', '') 
-output_file = sys.argv[2] 
+text = sys.argv[1].replace('"', '')
+output_file = sys.argv[2]
 
 try:
     print("Chargement du modèle Coqui TTS Tacotron 2 DDC sur CPU...", file=sys.stderr)
-    
-    # CHARGEMENT DU MODÈLE TACOTRON 2 (pour la meilleure qualité d'intonation)
-    tts = TTS(model_name="tts_models/fr/css10/tacotron2-DDC", progress_bar=False, gpu=False)
-    
+
+    # CHARGEMENT DU MODÈLE TACOTRON 2 (meilleure qualité d’intonation FR)
+    tts = TTS(
+        model_name="tts_models/fr/css10/tacotron2-DDC",
+        progress_bar=False,
+        gpu=False
+    )
+
     print(f"Génération de l'audio et sauvegarde vers {output_file}...", file=sys.stderr)
-    
-    # Génération
-    tts.tts_to_file(text=text, file_path=output_file) 
 
-    # Calcul de la Durée
+    # Génération de l’audio
+    tts.tts_to_file(text=text, file_path=output_file)
+
+    # Calcul de la durée de l’audio
     print("Calcul de la durée de l'audio...", file=sys.stderr)
     with wave.open(output_file, 'rb') as wf:
         frames = wf.getnframes()
         rate = wf.getframerate()
         duration = frames / float(rate)
 
-    # Sortie JSON
+    # Sortie JSON (pour n8n / API / pipeline)
     output_data = {
         "audioPath": output_file,
         "duration": duration
@@ -38,5 +45,8 @@ try:
     print(json.dumps(output_data))
 
 except Exception as e:
-    print(f"Erreur lors de la génération ou du calcul de durée : {e}", file=sys.stderr)
-    sys.exit(1)
+    print(
+        f"Erreur lors de la génération ou du calcul de durée : {e}",
+        file=sys.stderr
+    )
+    sys.exit(1)