Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import time
 import torch
 
+
 # Initialisation du modèle TTS avec GPU désactivé
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
@@ -43,8 +44,8 @@ with gr.Blocks() as demo:
         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.")
+        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
@@ -55,17 +56,28 @@ with gr.Blocks() as demo:
             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.
         """)
+
+        section_inputs = []
+        
+        def add_section(sections):
+            section_name_input = gr.Textbox(label=f"Nom de la Section 🏷️", placeholder="Exemple : Introduction")
+            section_text_input = gr.TextArea(label=f"Texte de la Section ✏️", placeholder="Écrivez ici le texte à convertir en voix off.")
+            section_inputs.append((section_name_input, section_text_input))
+            sections.append({"name": "", "text": ""})
+            return sections
         
-        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"
-        )
+        def remove_section(sections):
+            if sections:
+                sections.pop()
+                section_inputs.pop()
+            return sections
         
         add_section_btn = gr.Button("Ajouter une Section ➕")
         remove_section_btn = gr.Button("Supprimer une Section ➖")
+        
+        add_section_btn.click(add_section, inputs=[sections], outputs=[sections])
+        remove_section_btn.click(remove_section, inputs=[sections], outputs=[sections])
+        
         next_btn_2 = gr.Button("Valider les Sections ➡️")
         prev_btn_2 = gr.Button("⬅️ Retour")
 
@@ -84,12 +96,22 @@ with gr.Blocks() as demo:
     # É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))
+        def generate_audios(project_info, sections):
+            outputs = []
+            for i, (name_input, text_input) in enumerate(section_inputs):
+                name = name_input.value
+                text = text_input.value
+                audio_path = generate_section_audio(
+                    project_info["name"],
+                    name,
+                    text,
+                    project_info["voice"]
+                )
+                outputs.append(gr.Audio(value=audio_path))
+            return outputs
+
+        audio_outputs_container = gr.Column()
         
         save_project_btn = gr.Button("Sauvegarder le Projet ✅")
 
@@ -113,29 +135,6 @@ with gr.Blocks() as demo:
             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):
-        sections.append({"name": f"Section {len(sections) + 1}", "text": ""})
-        return sections
-
-    def remove_section(sections):
-        if sections:
-            sections.pop()
-            return sections
-        return sections
-
-    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,
@@ -144,55 +143,43 @@ with gr.Blocks() as demo:
     ).then(
         update_visibility,
         inputs=[current_step],
-        outputs=[step1, step2, step3]
-    )
-
-    add_section_btn.click(
-         add_section,
-         inputs=[sections],
-         outputs=[sections_list]
-     )
-
-    remove_section_btn.click(
-         remove_section,
-         inputs=[sections],
-         outputs=[sections_list]
+        outputs=[step1, step2, step3, step4]
     )
 
     next_btn_2.click(
-         lambda x: x + 1,
-         inputs=[current_step],
-         outputs=[current_step]
+      lambda x: x + 1,
+      inputs=[current_step],
+      outputs=[current_step]
     ).then(
-         update_visibility,
-         inputs=[current_step],
-         outputs=[step1, step2, step3]
+      update_visibility,
+      inputs=[current_step],
+      outputs=[step1, step2, step3, step4]
     )
 
     prev_btn_2.click(
-         prev_step,
-         inputs=[current_step],
-         outputs=[current_step]
+      prev_step,
+      inputs=[current_step],
+      outputs=[current_step]
     ).then(
-         update_visibility,
-         inputs=[current_step],
-         outputs=[step1, step2, step3]
+      update_visibility,
+      inputs=[current_step],
+      outputs=[step1, step2, step3, step4]
     )
 
     generate_btn.click(
-         generate_audios,
-         inputs=[project_info, sections],
-         outputs=audio_outputs
+      generate_audios,
+      inputs=[project_info, sections],
+      outputs=audio_outputs_container
     )
 
     prev_btn_3.click(
-         prev_step,
-         inputs=[current_step],
-         outputs=[current_step]
+      prev_step,
+      inputs=[current_step],
+      outputs=[current_step]
     ).then(
-         update_visibility,
-         inputs=[current_step],
-         outputs=[step1, step2, step3]
+      update_visibility,
+      inputs=[current_step],
+      outputs=[step1, step2, step3, step4]
     )
 
 demo.launch()