Update app.py

app.py

@@ -3,134 +3,146 @@ import os
 import time
 import gradio as gr
 from TTS.api import TTS
-import json
 
-# Initialisation du modèle TTS
-tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
-# Dossier de base pour les projets
-base_output_folder = "projects"
-os.makedirs(base_output_folder, exist_ok=True)
+# Initialisation du modèle TTS avec GPU désactivé
+tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
-# Fonction pour créer ou charger un projet
-def create_or_load_project(project_name, speaker, agree):
+# Répertoire de sortie pour tous les fichiers audio
+output_folder = "output_audio"
+os.makedirs(output_folder, exist_ok=True)
+
+# Fonction pour générer un fichier audio à partir d'une section
+def generate_section_audio(project_name, section_name, text, speaker):
+    try:
+        # Création du sous-dossier pour le projet
+        project_path = os.path.join(output_folder, project_name)
+        os.makedirs(project_path, exist_ok=True)
+
+        # Définir le chemin de sortie pour cette section
+        file_name = f"{section_name}.wav"
+        output_path = os.path.join(project_path, file_name)
+
+        # Vérifier la disponibilité des fichiers audio pour le speaker
+        speaker_wav_paths = [os.path.join("examples", f) for f in os.listdir("examples") if f.startswith(speaker) and f.endswith(".wav")]
+        if not speaker_wav_paths:
+            raise ValueError(f"Aucun fichier audio trouvé pour le speaker : {speaker}")
+
+        # Génération de l'audio
+        tts.tts_to_file(
+            text=text,
+            file_path=output_path,
+            speaker_wav=speaker_wav_paths,
+            language="fr"
+        )
+
+        return output_path  # Retourne le chemin de l'audio généré
+    except Exception as e:
+        return str(e)  # Retourne l'erreur pour gestion dans l'interface
+
+# Fonction pour traiter un projet complet
+def process_project(project_name, sections, speaker):
+    results = []
+    for section in sections:
+        section_name, text = section["name"], section["text"]
+        result = generate_section_audio(project_name, section_name, text, speaker)
+        results.append({"section": section_name, "result": result})
+    return results
+
+# Fonction de validation des conditions d'utilisation
+def validate_conditions(agree):
     if not agree:
-        raise gr.Error("❗ Veuillez accepter les conditions d'utilisation.")
-    
-    project_folder = os.path.join(base_output_folder, project_name)
-    os.makedirs(project_folder, exist_ok=True)
-    
-    project_data = {
-        "name": project_name,
-        "speaker": speaker,
-        "sections": []
-    }
-    
-    # Vérifier si le projet existe déjà
-    project_file = os.path.join(project_folder, "project_data.json")
-    if os.path.exists(project_file):
-        with open(project_file, "r") as f:
-            project_data = json.load(f)
-    else:
-        with open(project_file, "w") as f:
-            json.dump(project_data, f)
-    
-    return project_data
-
-# Fonction pour ajouter ou mettre à jour une section
-def update_section(project_data, section_name, section_text):
-    for section in project_data["sections"]:
-        if section["name"] == section_name:
-            section["text"] = section_text
-            return project_data
-    
-    project_data["sections"].append({
-        "name": section_name,
-        "text": section_text,
-        "audio_path": None
-    })
-    return project_data
-
-# Fonction pour générer l'audio d'une section
-def generate_section_audio(section_name, text, speaker):
-    output_path = os.path.join(base_output_folder, f"{section_name}.wav")
-    
-    tts.tts_to_file(
-        text=text,
-        file_path=output_path,
-        speaker_wav=[os.path.join("examples", f) for f in os.listdir("examples") if f.startswith(speaker) and f.endswith(".wav")],
-        language="fr"
-    )
-    
-    return output_path
+        raise gr.Error("❗ Veuillez accepter les conditions d'utilisation pour continuer.")
 
 # Interface Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎙️ Projet Margaux TTS")
-    
-    # Étape 1: Création ou chargement du projet
-    with gr.Tab("🔧 Création du Projet"):
-        project_name = gr.Textbox(label="📝 Nom du projet")
-        speaker = gr.Dropdown(label="🔊 Voix", choices=["Margaux"], value="Margaux")
+    # Titre principal
+    gr.Markdown("# 🎙️ Synthèse Vocale Margaux")
+
+    # Introduction et explication globale
+    gr.Markdown("""
+    ## 👋 Bienvenue sur Margaux - Votre outil de synthèse vocale avancée
+    Margaux vous permet de générer des voix off naturelles à partir de textes, structurées par sections pour une meilleure qualité audio.  
+    **Étapes principales :**
+    1. 🛠️ **Créer un projet** : Définissez le nom du projet et choisissez la voix.
+    2. ✍️ **Ajouter des sections** : Divisez votre texte en parties claires, chacune avec un nom unique.
+    3. 🎧 **Générer les audios** : Chaque section est transformée en fichier audio individuel.
+    4. 🔄 **Écoutez et ajustez** : Régénérez les audios si nécessaire, indépendamment des autres.
+    5. 📁 **Sauvegardez le projet** : Finalisez et récupérez les fichiers validés.
+    """)
+
+    # Étape 1 : Création du Projet
+    with gr.Box():
+        gr.Markdown("### 🛠️ Étape 1 : Création du Projet")
+        gr.Markdown("**📂 Définissez les informations générales pour votre projet.**")
+        gr.Markdown("Le nom du projet servira à organiser vos fichiers dans un dossier dédié.")
+        project_name = gr.Textbox(label="Nom du Projet", placeholder="Exemple : Capsule_Video_PLU")
+        speaker = gr.Dropdown(label="Voix 🎙️", choices=["Margaux"], value="Margaux")  # Liste de voix
         agree = gr.Checkbox(label="✅ J'accepte les conditions d'utilisation")
-        create_btn = gr.Button("🚀 Créer/Charger le Projet")
+        create_project_btn = gr.Button("Créer le Projet 🚀")
+
+    # Étape 2 : Gestion des Sections
+    with gr.Box():
+        gr.Markdown("### ✍️ Étape 2 : Ajoutez vos Sections")
+        gr.Markdown("""
+        **📝 Divisez votre script en plusieurs sections pour une meilleure qualité.**  
+        Chaque section doit avoir :  
+        - Un **nom unique** 🏷️ qui servira à nommer le fichier audio.  
+        - Un **texte clair et concis** ✏️.  
+        """)
+        sections = gr.State([])  # Liste des sections dynamiques
+        sections_list = gr.Column()  # Conteneur pour les sections ajoutées dynamiquement
+        add_section_btn = gr.Button("+ Ajouter une Section ➕")
+        remove_section_btn = gr.Button("- Supprimer la dernière Section ➖")
     
-    # Étape 2: Gestion des sections
-    with gr.Tab("📋 Gestion des Sections"):
-        project_info = gr.JSON(label="📁 Informations du projet", visible=False)
-        sections_list = gr.List(label="📜 Sections du projet")
-        
-        # Entrées pour ajouter une nouvelle section
-        section_name = gr.Textbox(label="📝 Nom de la section")
-        section_text = gr.Textbox(label="✍️ Texte de la section", lines=5)
-        
-        add_section_btn = gr.Button("➕ Ajouter/Mettre à jour la section")
-
-        # Section pour générer l'audio de chaque section individuellement
-        audio_outputs_display = []
-
-    # Fonctions de callback
-    def load_project(project_name, speaker, agree):
-        project_data = create_or_load_project(project_name, speaker, agree)
-        sections = [section["name"] for section in project_data["sections"]]
-        
-        return project_data, sections
-
-    def add_section(project_data, section_name, section_text):
-        updated_project = update_section(project_data, section_name, section_text)
-        sections = [section["name"] for section in updated_project["sections"]]
-        
-        return updated_project, sections
-
-    def generate_audio(section_name, text):
-        audio_path = generate_section_audio(section_name, text, speaker.value)
-        
-        return audio_path
-
-    # Connexion des événements
-    create_btn.click(load_project, inputs=[project_name, speaker, agree], outputs=[project_info, sections_list])
-    add_section_btn.click(add_section, inputs=[project_info, section_name, section_text], outputs=[project_info, sections_list])
-
-    # Dynamique : Créer des boutons de génération audio pour chaque section
-    def update_audio_buttons(sections):
-        global audio_outputs_display
-        
-        audio_outputs_display.clear()
-        
-        for sec in sections:
-            audio_output_btn = gr.Button(f"🎤 Générer l'audio pour {sec}")
-            audio_output_display = gr.Audio(label=f"🔊 Audio de {sec}", type="filepath", visible=False)
-
-            def generate_for_section(sec=sec):
-                text_for_section = next((s['text'] for s in project_info['sections'] if s['name'] == sec), "")
-                return generate_audio(sec, text_for_section)
-
-            audio_output_btn.click(generate_for_section)
-            audio_outputs_display.append(audio_output_display)
-
-            # Ajouter les boutons et les affichages audio à l'interface
-            demo.add(audio_output_btn)
-            demo.add(audio_output_display)
+    # Étape 3 : Validation des Sections et Génération des Audios
+    with gr.Box():
+        gr.Markdown("### 🎧 Étape 3 : Génération des Audios")
+        gr.Markdown("""
+        **🎶 Générez un fichier audio pour chaque section.**  
+        - 🔄 Régénérez l’audio d’une section indépendamment si nécessaire.  
+        - ⚠️ En cas d’erreur, seuls les audios de sections valides seront disponibles.  
+        """)
+        generate_btn = gr.Button("Générer les Audios ▶️")
+        results_output = gr.Column()  # Conteneur pour les audios générés
+    
+    # Étape 4 : Sauvegarde Finale
+    with gr.Box():
+        gr.Markdown("### 📁 Étape 4 : Sauvegarde Finale")
+        gr.Markdown("""
+        **💾 Une fois satisfait des résultats :**  
+        - Cliquez sur **Sauvegarder** pour conserver uniquement les fichiers validés.  
+        - Les fichiers seront organisés dans un dossier au nom de votre projet.  
+        """)
+        save_project_btn = gr.Button("Sauvegarder le Projet ✅")
+
+    # Actions des Boutons
+    def create_project(project_name, speaker, agree):
+        validate_conditions(agree)
+        os.makedirs(os.path.join(output_folder, project_name), exist_ok=True)
+        return f"✅ Projet '{project_name}' créé avec succès !"
+
+    create_project_btn.click(create_project, inputs=[project_name, speaker, agree], outputs=[])
+
+    def add_section(sections):
+        section = {"name": f"Section_{len(sections) + 1}", "text": ""}
+        sections.append(section)
+        return sections
+
+    add_section_btn.click(add_section, inputs=[sections], outputs=[sections_list])
+
+    def remove_section(sections):
+        if sections:
+            sections.pop()
+        return sections
+
+    remove_section_btn.click(remove_section, inputs=[sections], outputs=[sections_list])
+
+    def generate_audios(project_name, sections, speaker):
+        results = process_project(project_name, sections, speaker)
+        return results
+
+    generate_btn.click(generate_audios, inputs=[project_name, sections, speaker], outputs=[results_output])
 
 # Lancement de l'interface
-demo.launch()
+demo.launch(debug=True)