Update app.py

app.py

@@ -1,138 +1,149 @@
 import subprocess
+import os
 import gradio as gr
 from TTS.api import TTS
-import os
-import time
 
-# Initialisation du modèle TTS
+# Initialisation du modèle TTS avec GPU désactivé
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
-# Dossier pour les fichiers audio générés
+# Répertoire de sortie pour tous les fichiers audio
 output_folder = "output_audio"
 os.makedirs(output_folder, exist_ok=True)
 
-def create_project(project_name, speaker, agree):
-    if not agree:
-        raise gr.Error("Veuillez accepter les conditions d'utilisation.")
-    
-    project_folder = os.path.join(output_folder, project_name)
-    os.makedirs(project_folder, exist_ok=True)
-    return project_folder
-
-def generate_audio(prompt, speaker, project_folder, section_name):
-    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 gr.Error(f"Aucun fichier audio trouvé pour le speaker : {speaker}")
-
-    output_path = os.path.join(project_folder, f"{section_name}.wav")
-    
-    tts.tts_to_file(
-        text=prompt,
-        file_path=output_path,
-        speaker_wav=speaker_wav_paths,
-        language="fr"
-    )
-    
-    return output_path
-
-custom_css = """
-.gradio-container {
-    font-family: 'Roboto', sans-serif;
-    background-color: #f7f9fc;
-}
-.gr-form {
-    background-color: white;
-    border-radius: 15px;
-    padding: 30px;
-    box-shadow: 0 8px 16px rgba(0, 0, 0, 0.1);
-}
-.gr-button {
-    background-color: #4a90e2;
-    border: none;
-    color: white;
-    font-weight: bold;
-    transition: all 0.3s ease;
-}
-.gr-button:hover {
-    background-color: #3a7bc8;
-    transform: translateY(-2px);
-}
-.gr-input, .gr-dropdown {
-    border: 1px solid #e0e0e0;
-    border-radius: 8px;
-    padding: 10px;
-}
-.gr-checkbox {
-    margin-top: 10px;
-}
-.gr-form > div {
-    margin-bottom: 20px;
-}
-"""
-
-title = "🎙️ Synthèse Vocale XTTS"
-description = """
-<h3 style='text-align: center; margin-bottom: 1em;'>Bienvenue sur notre outil de synthèse vocale XTTS !</h3>
-<p style='text-align: center;'>Générez une voix naturelle à partir de votre texte en français. Créez un projet et ajoutez des sections pour votre script.</p>
-"""
-
-available_speakers = list(set([f.split('_')[0] for f in os.listdir("examples") if f.endswith(".wav")]))
-
-with gr.Blocks(css=custom_css) as demo:
-    
+# 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
+
+# Interface Gradio
+with gr.Blocks() as demo:
+    # État global pour gérer les étapes et les sections
+    step = gr.State(value=1)
+    sections = gr.State(value=[])  # Liste dynamique des sections ajoutées
+
+    # Introduction et explication globale
+    gr.Markdown("""
+    # 🎙️ Synthèse Vocale Margaux
+    ## 👋 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. 📁 **Sauvegardez le projet** : Finalisez et récupérez les fichiers validés.
+    """)
+
     # Étape 1 : Création du Projet
-    gr.Markdown(f"<h1 style='text-align: center; color: #4a90e2;'>{title}</h1>")
-    gr.Markdown(description)
-    
-    with gr.Row():
-        project_name = gr.Textbox(label="Nom du projet", placeholder="Entrez le nom du projet")
-        speaker = gr.Dropdown(label="Voix", choices=available_speakers, value=available_speakers[0] if available_speakers else None)
-        agree = gr.Checkbox(label="J'accepte les conditions d'utilisation", value=False)
-    
-    create_project_btn = gr.Button("Créer le Projet", variant="primary")
-    
+    with gr.Row(visible=True) as step1:
+        gr.Markdown("### 🛠️ Étape 1 : Création du Projet")
+        gr.Markdown("**📂 Définissez les informations générales pour votre projet.**")
+        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")
+        project_message = gr.Markdown(value="", visible=False)  # Composant pour afficher les messages
+        next_btn_1 = gr.Button("Suivant ➡️")
+
     # Étape 2 : Gestion des Sections
-    sections = []
-    
+    with gr.Row(visible=False) as step2:
+        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_list = gr.Column()  # Conteneur dynamique pour afficher les sections ajoutées
+        add_section_btn = gr.Button("+ Ajouter une Section ➕")
+        remove_section_btn = gr.Button("- Supprimer la dernière Section ➖")
+        prev_btn_2 = gr.Button("⬅️ Précédent")
+        next_btn_2 = gr.Button("Suivant ➡️")
+
+    # Étape 3 : Génération des Audios et Gestion des Sections
+    with gr.Row(visible=False) as step3:
+        gr.Markdown("### 🎧 Étape 3 : Génération et Gestion 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 afficher les lecteurs audio et les boutons de régénération
+        prev_btn_3 = gr.Button("⬅️ Précédent")
+        save_project_btn = gr.Button("Sauvegarder le Projet ✅")
+
+    # Actions des Boutons
+    def create_project(project_name, speaker, agree):
+        if not agree:
+            return "❗ Veuillez accepter les conditions d'utilisation.", 1
+        if not project_name:
+            return "❗ Le nom du projet est obligatoire.", 1
+        os.makedirs(os.path.join(output_folder, project_name), exist_ok=True)
+        return f"✅ Projet '{project_name}' créé avec succès !", 2
+
+    next_btn_1.click(
+        create_project,
+        inputs=[project_name, speaker, agree],
+        outputs=[project_message, step]
+    )
+
     def add_section(sections):
-        section_name = f"Section {len(sections) + 1}"
-        sections.append({"name": section_name, "text": ""})
+        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
-    
-    def update_section_text(index, text, sections):
-        if index < len(sections):
-            sections[index]["text"] = text
-        return sections
-    
-    section_outputs = []
-    
-    with gr.Column():
-        section_list = gr.State(value=sections)
-        
-        def render_sections(sections):
-            return [gr.Row([
-                gr.Textbox(label=f"Texte pour {section['name']}", value=section['text'], lines=5).change(update_section_text, inputs=[sections.index(section), _, section_list], outputs=None),
-                gr.Textbox(label="Nom du fichier", value=section['name'], interactive=False),
-                gr.Button("Régénérer", variant="secondary").click(generate_audio, inputs=[section['text'], speaker, project_folder, section['name']], outputs=[gr.Audio(label=f"Audio {section['name']}")])
-            ]) for section in sections]
-        
-        with gr.Row():
-            add_section_btn = gr.Button("+ Ajouter une section")
-            remove_section_btn = gr.Button("- Supprimer la dernière section")
-        
-        add_section_btn.click(add_section, inputs=[section_list], outputs=section_list)
-        remove_section_btn.click(remove_section, inputs=[section_list], outputs=section_list)
-        
-        section_output_area = gr.Column()
-    
-        # Dynamically render sections
-        with section_output_area:
-            for section in render_sections(sections):
-                section_output_area.append(section)
-
-demo.launch(debug=True)
+
+    remove_section_btn.click(remove_section, inputs=[sections], outputs=[sections_list])
+
+    def generate_audios(project_name, sections, speaker):
+        results = []
+        for section in sections:
+            name = section["name"]
+            text = section["text"]
+            audio_path = generate_section_audio(project_name, name, text, speaker)
+            results.append(audio_path)
+        return results
+
+    generate_btn.click(
+        generate_audios,
+        inputs=[project_name, sections, speaker],
+        outputs=[results_output]
+    )
+
+    next_btn_2.click(lambda: 3, inputs=None, outputs=step)
+    prev_btn_2.click(lambda: 1, inputs=None, outputs=step)
+    step.change(lambda x: {"visible": x == 2}, inputs=step, outputs=[step1, step2])
+
+    prev_btn_3.click(lambda: 2, inputs=None, outputs=step)
+    save_project_btn.click(lambda: "🎉 Projet sauvegardé avec succès !", outputs=[project_message])
+
+# Lancement de l'interface
+demo.launch(debug=True)