Update app.py

app.py

@@ -38,100 +38,82 @@ def generate_section_audio(project_name, section_name, text, speaker):
     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}
+# Fonction pour ajouter une section
+def add_section(sections):
+    section = {"name": f"Section_{len(sections) + 1}", "text": ""}
+    sections.append(section)
+    return sections
+
+# Fonction pour supprimer la dernière section
+def remove_section(sections):
+    if sections:
+        sections.pop()
+    return sections
+
+# Fonction pour générer les audios de toutes les sections
+def generate_all_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
 
 # Interface Gradio
 with gr.Blocks() as demo:
-    # État global pour gérer les 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.
+    ## 👋 Bienvenue sur Margaux
+    Créez des voix off naturelles à partir de textes. Divisez votre contenu en plusieurs sections pour une meilleure qualité audio.
     """)
 
-    # Étape 1 : Création du Projet
-    with gr.Row(visible=True) as step1:
-        gr.Markdown("### 🛠️ Étape 1 : Création du Projet")
+    with gr.Box():
+        gr.Markdown("### 🛠️ Configuration 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")
+    with gr.Box():
+        gr.Markdown("### ✍️ Gestion des Sections")
+        sections = gr.State(value=[])
         sections_list = gr.Column()  # Conteneur 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
-    with gr.Row(visible=False) as step3:
-        gr.Markdown("### 🎧 Étape 3 : Génération des Audios")
-        results_output = gr.Column()  # Conteneur pour afficher les résultats
+        def update_section_list(sections):
+            return [gr.Textbox(label=f"Nom : {s['name']}", value=s["text"], lines=2) for s in sections]
+
+        add_section_btn.click(add_section, inputs=[sections], outputs=[sections_list])
+        remove_section_btn.click(remove_section, inputs=[sections], outputs=[sections_list])
+
+    with gr.Box():
+        gr.Markdown("### 🎧 Génération des Audios")
         generate_btn = gr.Button("Générer les Audios ▶️")
-        prev_btn_3 = gr.Button("⬅️ Précédent")
+        results_output = gr.Column()  # Conteneur pour afficher les audios générés
+
+        def generate_audios_and_display(project_name, sections, speaker):
+            results = generate_all_audios(project_name, sections, speaker)
+            return [gr.Audio(label=f"Audio : {s['name']}", value=path) for s, path in zip(sections, results)]
+
+        generate_btn.click(
+            generate_audios_and_display,
+            inputs=[project_name, sections, speaker],
+            outputs=[results_output]
+        )
+
+    with gr.Box():
+        gr.Markdown("### 📁 Sauvegarde du Projet")
         save_project_btn = gr.Button("Sauvegarder le Projet ✅")
+        save_message = gr.Markdown("")
+
+        def save_project(project_name, sections):
+            project_path = os.path.join(output_folder, project_name)
+            if not os.path.exists(project_path):
+                return "⚠️ Aucun audio génér�� à sauvegarder."
+            return f"🎉 Projet '{project_name}' sauvegardé dans le dossier '{project_path}'."
 
-    # Actions des Boutons
-    def create_project(project_name, speaker, agree):
-        if not agree:
-            return "❗ Veuillez accepter les conditions d'utilisation.", False, False
-        if not project_name:
-            return "❗ Le nom du projet est obligatoire.", False, False
-        os.makedirs(os.path.join(output_folder, project_name), exist_ok=True)
-        return f"✅ Projet '{project_name}' créé avec succès !", True, False
-
-    next_btn_1.click(
-        create_project,
-        inputs=[project_name, speaker, agree],
-        outputs=[project_message, step1, step2],
-    )
-
-    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],
-    )
-
-    prev_btn_2.click(lambda: (True, False), inputs=None, outputs=[step1, step2])
-    next_btn_2.click(lambda: (False, True), inputs=None, outputs=[step2, step3])
-    prev_btn_3.click(lambda: (True, False), inputs=None, outputs=[step2, step3])
+        save_project_btn.click(save_project, inputs=[project_name, sections], outputs=[save_message])
 
 # Lancement de l'interface
 demo.launch(debug=True)