Update app.py

app.py

@@ -3,148 +3,113 @@ import os
 import gradio as gr
 from TTS.api import TTS
 
-class TTSInterface:
-    def __init__(self):
-        # Initialisation du modèle TTS
-        self.tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
-        self.output_folder = "output_audio"
-        os.makedirs(self.output_folder, exist_ok=True)
-
-    def create_interface(self):
-        with gr.Blocks(theme=gr.themes.Soft()) as demo:
-            # Variables d'état
-            state = gr.State({
-                "project_name": "",
-                "sections": []
-            })
-
-            # En-tête
-            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.
-            3. 🎧 **Générer les audios** : Chaque section est transformée en fichier audio.
-            4. 📁 **Sauvegardez le projet** : Finalisez et récupérez les fichiers.
-            """)
-
-            # Tabs pour les étapes
-            with gr.Tabs() as tabs:
-                # Étape 1: Configuration du projet
-                with gr.Tab("🛠️ Création du Projet"):
-                    with gr.Column():
-                        gr.Markdown("### Configuration initiale")
-                        project_name = gr.Textbox(
-                            label="Nom du Projet",
-                            placeholder="Exemple : Capsule_Video_PLU",
-                            scale=1
-                        )
-                        speaker = gr.Dropdown(
-                            choices=["Margaux"],
-                            value="Margaux",
-                            label="Voix 🎙️",
-                            scale=1
-                        )
-                        terms = gr.Checkbox(
-                            label="✅ J'accepte les conditions d'utilisation"
-                        )
-                        create_btn = gr.Button("Créer le Projet ➡️", variant="primary")
-                        status_message = gr.Markdown("")  # Pour afficher les messages de statut
-
-                # Étape 2: Gestion des sections
-                with gr.Tab("✍️ Sections"):
-                    with gr.Column():
-                        gr.Markdown("### Gestion des sections")
-                        
-                        # Template pour une section
-                        section_template = gr.DataFrame(
-                            headers=["Nom", "Texte"],
-                            row_count=0,
-                            col_count=2,
-                            interactive=True
-                        )
-                        
-                        with gr.Row():
-                            add_section = gr.Button("+ Ajouter une Section ➕")
-                            remove_section = gr.Button("- Supprimer la dernière Section ➖")
-
-                # Étape 3: Génération des audios
-                with gr.Tab("🎧 Génération"):
-                    with gr.Column():
-                        gr.Markdown("### Génération des audios")
-                        generate_all = gr.Button("Générer tous les audios ▶️", variant="primary")
-                        audio_output = gr.Audio(label="Audio généré")  # Un seul composant Audio pour l'exemple
-                        status = gr.Markdown("")  # Pour les messages de statut de génération
-
-            # Fonctions de callback
-            def update_project_info(name, terms):
-                if not terms:
-                    return "⚠️ Veuillez accepter les conditions d'utilisation"
-                if not name.strip():
-                    return "⚠️ Le nom du projet est requis"
-                return f"✅ Projet '{name}' créé avec succès!"
-
-            def add_new_section(data):
-                if data is None:
-                    data = []
-                new_row = [f"Section_{len(data) + 1}", ""]
-                return data + [new_row]
-
-            def remove_last_section(data):
-                if data and len(data) > 0:
-                    return data[:-1]
-                return data
-
-            def generate_audio(text, project_name):
-                try:
-                    if not text or not project_name:
-                        return None, "⚠️ Texte ou nom de projet manquant"
-                    
-                    output_path = os.path.join(self.output_folder, f"{project_name}_output.wav")
-                    
-                    # Génération de l'audio
-                    self.tts.tts_to_file(
-                        text=text,
-                        file_path=output_path,
-                        language="fr"
-                    )
-                    
-                    return output_path, "✅ Audio généré avec succès"
-                except Exception as e:
-                    return None, f"⚠️ Erreur: {str(e)}"
-
-            # Événements
-            create_btn.click(
-                fn=update_project_info,
-                inputs=[project_name, terms],
-                outputs=[status_message]
-            )
-            
-            add_section.click(
-                fn=add_new_section,
-                inputs=[section_template],
-                outputs=[section_template]
-            )
-            
-            remove_section.click(
-                fn=remove_last_section,
-                inputs=[section_template],
-                outputs=[section_template]
-            )
-
-            generate_all.click(
-                fn=generate_audio,
-                inputs=[project_name, speaker],
-                outputs=[audio_output, status]
-            )
-
-        return demo
-
-if __name__ == "__main__":
-    interface = TTSInterface()
-    demo = interface.create_interface()
-    demo.launch(debug=True)
+# Initialisation du modèle TTS avec GPU désactivé
+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:
+        project_path = os.path.join(output_folder, project_name)
+        os.makedirs(project_path, exist_ok=True)
+
+        file_name = f"{section_name}.wav"
+        output_path = os.path.join(project_path, 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 ValueError(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  
+    except Exception as e:
+        return str(e)
+
+# 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:
+    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.
+    """)
+
+    # Étape 1 : Création du Projet
+    with gr.Box():
+        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")
+        agree = gr.Checkbox(label="✅ J'accepte les conditions d'utilisation")
+
+        next_step_btn = gr.Button("Suivant ➡️")
+        
+    # Étape 2 : Ajout des Sections
+    with gr.Box():
+        gr.Markdown("### ✍️ Étape 2 : Ajoutez vos Sections")
+        
+        sections = gr.State(value=[{"name": "Section_1", "text": ""}])
+        
+        section_inputs = gr.Column()  # Conteneur pour afficher les sections ajoutées
+        
+        def update_section_list(sections):
+            return [gr.Row([
+                gr.Textbox(label=f"Nom : {s['name']}", value=s["name"], interactive=False),
+                gr.Textbox(label="Texte", value=s["text"], lines=2)
+            ]) for s in sections]
+
+        add_section_btn = gr.Button("+ Ajouter une Section ➕")
+        remove_section_btn = gr.Button("- Supprimer la dernière Section ➖")
+
+        add_section_btn.click(lambda sections: (sections + [{"name": f"Section_{len(sections) + 1}", "text": ""}], update_section_list(sections)), inputs=sections, outputs=[sections, section_inputs])
+        
+        remove_section_btn.click(lambda sections: (sections[:-1], update_section_list(sections[:-1])) if len(sections) > 1 else (sections, update_section_list(sections)), inputs=sections, outputs=[sections, section_inputs])
+
+    # Étape 3 : Génération des Audios
+    with gr.Box():
+        gr.Markdown("### 🎧 Étape 3 : Génération des Audios")
+        
+        generate_btn = gr.Button("Générer les Audios ▶️")
+        
+        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])
+
+    # Sauvegarde du Projet
+    with gr.Box():
+        save_project_btn = gr.Button("Sauvegarder le Projet ✅")
+        
+        def save_project(project_name):
+            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}'."
+
+        save_project_btn.click(save_project, inputs=[project_name], outputs=[results_output])
+
+# Lancement de l'interface
+demo.launch(debug=True)