Update app.py

app.py

@@ -15,132 +15,184 @@ 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é
+        
+        tts.tts_to_file(text=text, file_path=output_path, speaker_wav=speaker_wav_paths, language="fr")
+        return output_path
     except Exception as e:
-        return str(e)  # Retourne l'erreur pour gestion dans l'interface
-
-# Fonction pour gérer l'état d'avancement de l'interface
-def update_step(step):
-    return {"visible": step == 1}, {"visible": step == 2}, {"visible": step == 3}
+        return str(e)
 
 # Interface Gradio
 with gr.Blocks() as demo:
-    # État pour suivre l'étape active
-    step = gr.State(value=1)
-    # Liste dynamique des sections
-    sections = gr.State(value=[])
-
-    # Introduction générale
-    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.
-    """)
+    current_step = gr.State(1)
+    project_info = gr.State({})
+    sections = gr.State([])
 
     # Étape 1 : Création du Projet
-    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 ➡️")
+    with gr.Group() as step1:
+        gr.Markdown("## 🛠️ Étape 1 : Création du Projet")
+        gr.Markdown("**Définissez les informations générales pour votre projet de synthèse vocale.**")
+        project_name = gr.Textbox(label="Nom du Projet 📂", placeholder="Exemple : Capsule_Video_PLU")
+        gr.Markdown("Le nom du projet sera utilisé pour organiser les fichiers dans un dossier dédié.")
+        speaker = gr.Dropdown(label="Voix 🎙️", choices=["Margaux"], value="Margaux")
+        gr.Markdown("Choisissez la voix souhaitée pour générer vos fichiers audio. Plus de voix seront ajoutées prochainement.")
+        agree = gr.Checkbox(label="✅ J’accepte les conditions d’utilisation")
+        gr.Markdown("Vous devez accepter les conditions d’utilisation pour continuer.")
+        next_btn_1 = gr.Button("Créer le Projet ➡️")
 
     # Étape 2 : Gestion des Sections
-    with gr.Row(visible=False) as step2:
-        gr.Markdown("### ✍️ Étape 2 : Ajoutez vos Sections")
+    with gr.Group(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** ✏️.  
+            **Divisez votre texte en plusieurs sections pour une meilleure qualité audio.**  
+            Chaque section correspond à une partie de votre script. Par exemple, une introduction, une explication, ou une conclusion.
+            Donnez un nom unique à chaque section et saisissez le texte correspondant.
         """)
-        sections_list = gr.Column()  # Conteneur pour les sections
-        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 Sauvegarde")
+        
+        sections_list = gr.List(
+            [gr.Group([
+                gr.Textbox(label="Nom de la Section 🏷️", placeholder="Exemple : Introduction"),
+                gr.TextArea(label="Texte de la Section ✏️", placeholder="Écrivez ici le texte à convertir en voix off.")
+            ])],
+            label="Sections"
+        )
+        
+        add_section_btn = gr.Button("Ajouter une Section ➕")
+        remove_section_btn = gr.Button("Supprimer une Section ➖")
+        next_btn_2 = gr.Button("Valider les Sections ➡️")
+        prev_btn_2 = gr.Button("⬅️ Retour")
+
+    # Étape 3 : Validation des Sections
+    with gr.Group(visible=False) as step3:
+        gr.Markdown("## 🎧 Étape 3 : Validation des Sections")
         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.  
+            **Vérifiez vos sections avant de lancer la génération des audios.**  
+            Assurez-vous que chaque section a un nom unique et que les textes sont complets.
+            Cliquez sur Générer les Audios pour commencer.
         """)
+        
         generate_btn = gr.Button("Générer les Audios ▶️")
-        results_output = gr.Column()  # Conteneur pour les audios générés
-        prev_btn_3 = gr.Button("⬅️ Précédent")
+        prev_btn_3 = gr.Button("⬅️ Retour")
+
+    # Étape 4 : Génération des Audios
+    with gr.Group(visible=False) as step4:
+        gr.Markdown("## 🎧 Étape 4 : Génération des Audios")
+        audio_outputs = []
+        
+        def render_audio_outputs(sections):
+            return [gr.Audio(label=f"Audio de {section['name']}") for section in sections]
+
+        audio_outputs.extend(render_audio_outputs(sections))
+        
         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
+    # Fonctions de navigation et de traitement
+    def next_step(step):
+        return step + 1
 
-    next_btn_1.click(
-        create_project,
-        inputs=[project_name, speaker, agree],
-        outputs=[project_message, step]  # Le message s'affiche dans `project_message`, et `step` gère l'étape active
-    )
-    step.change(update_step, inputs=step, outputs=[step1, step2, step3])
+    def prev_step(step):
+        return max(1, step - 1)
+
+    def update_visibility(step):
+        return {
+            step1: gr.update(visible=step == 1),
+            step2: gr.update(visible=step == 2),
+            step3: gr.update(visible=step == 3),
+            step4: gr.update(visible=step == 4)
+        }
+
+    def create_project(name, voice, terms):
+        if not name or not terms:
+            return "Veuillez remplir tous les champs et accepter les conditions.", None
+        return f"Projet '{name}' créé avec succès !", {"name": name, "voice": voice}, 2
 
     def add_section(sections):
-        section = {"name": f"Section_{len(sections) + 1}", "text": ""}
-        sections.append(section)
+        sections.append({"name": f"Section {len(sections) + 1}", "text": ""})
         return sections
 
-    add_section_btn.click(add_section, inputs=[sections], outputs=[sections_list])
-
     def remove_section(sections):
         if sections:
             sections.pop()
+            return sections
         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
-
-    next_btn_2.click(lambda: 3, inputs=None, outputs=step)
-    prev_btn_2.click(lambda: 1, inputs=None, outputs=step)
-    step.change(update_step, inputs=step, outputs=[step1, step2, step3])
-
-    prev_btn_3.click(lambda: 2, inputs=None, outputs=step)
-    generate_btn.click(generate_audios, inputs=[project_name, sections, speaker], outputs=[results_output])
+    def generate_audios(project_info, sections):
+        outputs = []
+        for section in sections:
+            audio_path = generate_section_audio(
+                project_info["name"],
+                section["name"],
+                section["text"],
+                project_info["voice"]
+            )
+            outputs.append(audio_path)
+        
+        return outputs
+
+    # Événements
+    next_btn_1.click(
+        create_project,
+        inputs=[project_name, speaker, agree],
+        outputs=[None, project_info, current_step]
+    ).then(
+        update_visibility,
+        inputs=[current_step],
+        outputs=[step1, step2, step3]
+    )
 
-# Lancement de l'interface
-demo.launch(debug=True)
+    add_section_btn.click(
+         add_section,
+         inputs=[sections],
+         outputs=[sections_list]
+     )
+
+     remove_section_btn.click(
+         remove_section,
+         inputs=[sections],
+         outputs=[sections_list]
+     )
+
+     next_btn_2.click(
+         lambda x: x + 1,
+         inputs=[current_step],
+         outputs=[current_step]
+     ).then(
+         update_visibility,
+         inputs=[current_step],
+         outputs=[step1, step2, step3]
+     )
+
+     prev_btn_2.click(
+         prev_step,
+         inputs=[current_step],
+         outputs=[current_step]
+     ).then(
+         update_visibility,
+         inputs=[current_step],
+         outputs=[step1, step2, step3]
+     )
+
+     generate_btn.click(
+         generate_audios,
+         inputs=[project_info, sections],
+         outputs=audio_outputs
+     )
+
+     prev_btn_3.click(
+         prev_step,
+         inputs=[current_step],
+         outputs=[current_step]
+     ).then(
+         update_visibility,
+         inputs=[current_step],
+         outputs=[step1, step2, step3]
+     )
+
+demo.launch()