Update app.py

app.py

@@ -5,7 +5,6 @@ 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)
 
@@ -41,89 +40,51 @@ 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 traiter un projet complet
-def process_project(project_name, sections, speaker):
-    results = []
-    for section in sections:
-        section_name, text = section["name"], section["text"]
-        result = generate_section_audio(project_name, section_name, text, speaker)
-        results.append({"section": section_name, "result": result})
-    return results
-
-# Fonction de validation des conditions d'utilisation
-def validate_conditions(agree):
-    if not agree:
-        raise gr.Error("❗ Veuillez accepter les conditions d'utilisation pour continuer.")
+# 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}
 
 # Interface Gradio
 with gr.Blocks() as demo:
-    # Titre principal
-    gr.Markdown("# 🎙️ Synthèse Vocale Margaux")
-
-    # Introduction et explication globale
-    gr.Markdown("""
-    ## 👋 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. 🔄 **Écoutez et ajustez** : Régénérez les audios si nécessaire, indépendamment des autres.
-    5. 📁 **Sauvegardez le projet** : Finalisez et récupérez les fichiers validés.
-    """)
+    step = gr.State(value=1)  # État pour suivre l'étape active
+    sections = gr.State(value=[])  # Liste dynamique des sections
 
     # Étape 1 : Création du Projet
-    with gr.Box():
+    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.**")
-        gr.Markdown("Le nom du projet servira à organiser vos fichiers dans un dossier dédié.")
         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")
-        create_project_btn = gr.Button("Créer le Projet 🚀")
+        next_btn_1 = gr.Button("Suivant ➡️")
 
     # Étape 2 : Gestion des Sections
-    with gr.Box():
+    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 = gr.State([])  # Liste des sections dynamiques
-        sections_list = gr.Column()  # Conteneur pour les sections ajoutées dynamiquement
+        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 ➖")
-    
-    # Étape 3 : Validation des Sections et Génération des Audios
-    with gr.Box():
-        gr.Markdown("### 🎧 Étape 3 : Génération 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.  
-        """)
+        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")
         generate_btn = gr.Button("Générer les Audios ▶️")
         results_output = gr.Column()  # Conteneur pour les audios générés
-    
-    # Étape 4 : Sauvegarde Finale
-    with gr.Box():
-        gr.Markdown("### 📁 Étape 4 : Sauvegarde Finale")
-        gr.Markdown("""
-        **💾 Une fois satisfait des résultats :**  
-        - Cliquez sur **Sauvegarder** pour conserver uniquement les fichiers validés.  
-        - Les fichiers seront organisés dans un dossier au nom de votre projet.  
-        """)
+        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):
-        validate_conditions(agree)
+        if not agree:
+            return gr.Error("❗ Veuillez accepter les conditions d'utilisation.")
+        if not project_name:
+            return gr.Error("❗ Le nom du projet est obligatoire.")
         os.makedirs(os.path.join(output_folder, project_name), exist_ok=True)
-        return f"✅ Projet '{project_name}' créé avec succès !"
+        return f"✅ Projet '{project_name}' créé avec succès !", 2
 
-    create_project_btn.click(create_project, inputs=[project_name, speaker, agree], outputs=[])
+    next_btn_1.click(create_project, inputs=[project_name, speaker, agree], outputs=[None, step])
+    step.change(update_step, inputs=step, outputs=[step1, step2, step3])
 
     def add_section(sections):
         section = {"name": f"Section_{len(sections) + 1}", "text": ""}
@@ -143,6 +104,11 @@ with gr.Blocks() as demo:
         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])
 
 # Lancement de l'interface