Update app.py

app.py

@@ -1,148 +1,168 @@
 import subprocess
 import os
+import time
 import gradio as gr
 from TTS.api import TTS
 
-# Initialisation du modèle TTS avec GPU désactivé
+# Initialisation du modèle TTS
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
-# 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"
-        )
+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.strip())
+    os.makedirs(project_folder, exist_ok=True)
+    return project_folder
+
+def generate_audio_section(text, section_name, project_folder, speaker):
+    file_name = f"{section_name.strip()}.wav"
+    output_path = os.path.join(project_folder, file_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}")
+
+    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 gérer la visibilité des étapes
-def update_steps(current_step, target_step):
-    return {"visible": target_step == 1}, {"visible": target_step == 2}, {"visible": target_step == 3}
-
-# Interface Gradio
-with gr.Blocks() as demo:
-    # Liste dynamique des sections
-    sections = gr.State(value=[])
-
-    # 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
-    with gr.Row(visible=True) as step1:
-        gr.Markdown("### 🛠️ Étape 1 : Création du 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
-    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 Sauvegarde
-    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,
+    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. Choisissez une voix, entrez votre texte, et écoutez le résultat ! 🎤</p>
+"""
+article = """
+<div style='margin: 20px auto; text-align: center; padding: 10px; background-color: #e8f0fe; border-radius: 10px;'>
+<p>En utilisant cette démo, vous acceptez les <a href='https://coqui.ai/cpml' target='_blank' style='color: #4a90e2; text-decoration: none;'>conditions d'utilisation du modèle Coqui Public</a> 📜</p>
+</div>
+"""
+
+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:
+    gr.Markdown(f"<h1 style='text-align: center; color: #4a90e2;'>{title}</h1>")
+    gr.Markdown(description)
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            project_name = gr.Textbox(
+                label="🗂️ Nom du projet",
+                placeholder="Entrez le nom du projet",
+                lines=1
+            )
+            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")
+    
+    project_folder_output = gr.State()
+
+    def on_create_project(name, spkr, agr):
+        folder_path = create_project(name, spkr, agr)
+        return folder_path
+    
+    create_project_btn.click(
+        on_create_project,
         inputs=[project_name, speaker, agree],
-        outputs=[project_message, step1],
+        outputs=[project_folder_output]
     )
 
-    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 = []
-        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],
-    )
+    section_textboxes = []
 
-    prev_btn_2.click(lambda: update_steps(2, 1), inputs=None, outputs=[step1, step2, step3])
-    next_btn_2.click(lambda: update_steps(2, 3), inputs=None, outputs=[step1, step2, step3])
-    prev_btn_3.click(lambda: update_steps(3, 2), inputs=None, outputs=[step1, step2, step3])
+    def add_section():
+        section_textbox = gr.Textbox(
+            label=f"✏️ Texte de la section {len(section_textboxes) + 1}",
+            placeholder="Entrez le texte pour cette section",
+            lines=5
+        )
+        section_name_textbox = gr.Textbox(
+            label=f"📝 Nom de la section {len(section_textboxes) + 1}",
+            placeholder="Entrez le nom du fichier pour cette section",
+            lines=1
+        )
+        section_textboxes.append((section_textbox, section_name_textbox))
+        return [textbox for textbox_pair in section_textboxes for textbox in textbox_pair]
+
+    add_section_btn = gr.Button("➕ Ajouter une section")
+    
+    def remove_section():
+        if section_textboxes:
+            removed_section = section_textboxes.pop()
+            return [textbox for textbox_pair in section_textboxes for textbox in textbox_pair]
+        return []
+
+    remove_section_btn = gr.Button("➖ Supprimer une section")
+
+    add_section_btn.click(add_section)
+    remove_section_btn.click(remove_section)
+
+    generate_audio_btn = gr.Button("🎶 Générer l'audio")
+    
+    audio_outputs_display = []
+
+    def generate_all_sections(sections_data, proj_folder):
+        audio_outputs = []
+        for text_box, name_box in sections_data:
+            text = text_box.value
+            name = name_box.value
+            audio_path = generate_audio_section(text, name.strip(), proj_folder.value, speaker.value)
+            audio_outputs.append(audio_path)
+        return audio_outputs
+    
+    generate_audio_btn.click(
+        generate_all_sections,
+        inputs=[section_textboxes, project_folder_output],
+        outputs=audio_outputs_display
+    )
 
-# Lancement de l'interface
-demo.launch(debug=True)
+demo.launch(debug=True)