Update app.py

app.py

@@ -5,181 +5,110 @@ import os
 import time
 import torch
 
-
-# Initialisation du modèle TTS avec GPU désactivé
+# Initialisation du modèle TTS
 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)
+# Dossier de base pour les projets
+base_output_folder = "projects"
+os.makedirs(base_output_folder, exist_ok=True)
+
+# Fonction pour créer ou charger un projet
+def create_or_load_project(project_name, speaker, agree):
+    if not agree:
+        raise gr.Error("❗ Veuillez accepter les conditions d'utilisation.")
+    
+    project_folder = os.path.join(base_output_folder, project_name)
+    os.makedirs(project_folder, exist_ok=True)
+    
+    project_data = {
+        "name": project_name,
+        "speaker": speaker,
+        "sections": []
+    }
+    
+    project_file = os.path.join(project_folder, "project_data.json")
+    if os.path.exists(project_file):
+        with open(project_file, "r") as f:
+            project_data = json.load(f)
+    else:
+        with open(project_file, "w") as f:
+            json.dump(project_data, f)
+    
+    return project_data
+
+# Fonction pour ajouter ou mettre à jour une section
+def update_section(project_data, section_name, section_text):
+    for section in project_data["sections"]:
+        if section["name"] == section_name:
+            section["text"] = section_text
+            return project_data
+    
+    project_data["sections"].append({
+        "name": section_name,
+        "text": section_text,
+        "audio_path": None
+    })
+    return project_data
+
+# Fonction pour générer l'audio d'une section
+def generate_section_audio(project_data, section_name):
+    project_folder = os.path.join(base_output_folder, project_data["name"])
+    for section in project_data["sections"]:
+        if section["name"] == section_name:
+            output_path = os.path.join(project_folder, f"{section_name}.wav")
+            tts.tts_to_file(
+                text=section["text"],
+                file_path=output_path,
+                speaker_wav=[os.path.join("examples", f) for f in os.listdir("examples") if f.startswith(project_data["speaker"]) and f.endswith(".wav")],
+                language="fr"
+            )
+            section["audio_path"] = output_path
+            return project_data, output_path
+    
+    raise gr.Error(f"❌ Section '{section_name}' non trouvée.")
 
 # Interface Gradio
 with gr.Blocks() as demo:
-    current_step = gr.State(1)
-    project_info = gr.State({})
-    sections = gr.State([])
-
-    # Étape 1 : Création du Projet
-    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.")
+    gr.Markdown("# 🎙️ Projet Margaux TTS")
+    
+    # État du projet
+    project_state = gr.State(value={})
+    
+    # Étape 1: Création ou chargement du projet
+    with gr.Tab("🆕 Création/Chargement du Projet"):
+        project_name = gr.Textbox(label="📝 Nom du projet")
+        speaker = gr.Dropdown(label="🔊 Voix", choices=["Margaux"], value="Margaux")
         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.Group(visible=False) as step2:
-        gr.Markdown("## ✍️ Étape 2 : Ajoutez vos Sections")
-        gr.Markdown("""
-            **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.
-        """)
-
-        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
-        
-        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")
-
-    # Étape 3 : Validation des Sections
-    with gr.Group(visible=False) as step3:
-        gr.Markdown("## 🎧 Étape 3 : Validation des Sections")
-        gr.Markdown("""
-            **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 ▶️")
-        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")
-        
-        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 ✅")
-
-    # Fonctions de navigation et de traitement
-    def next_step(step):
-        return step + 1
-
-    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
-
-    # É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, step4]
-    )
-
-    next_btn_2.click(
-      lambda x: x + 1,
-      inputs=[current_step],
-      outputs=[current_step]
-    ).then(
-      update_visibility,
-      inputs=[current_step],
-      outputs=[step1, step2, step3, step4]
-    )
-
-    prev_btn_2.click(
-      prev_step,
-      inputs=[current_step],
-      outputs=[current_step]
-    ).then(
-      update_visibility,
-      inputs=[current_step],
-      outputs=[step1, step2, step3, step4]
-    )
-
-    generate_btn.click(
-      generate_audios,
-      inputs=[project_info, sections],
-      outputs=audio_outputs_container
-    )
-
-    prev_btn_3.click(
-      prev_step,
-      inputs=[current_step],
-      outputs=[current_step]
-    ).then(
-      update_visibility,
-      inputs=[current_step],
-      outputs=[step1, step2, step3, step4]
-    )
-
+        create_btn = gr.Button("🚀 Créer/Charger le Projet")
+    
+    # Étape 2: Gestion des sections
+    with gr.Tab("📋 Gestion des Sections"):
+        sections_list = gr.List(label="📜 Sections du projet")
+        section_name = gr.Textbox(label="📝 Nom de la section")
+        section_text = gr.Textbox(label="✍️ Texte de la section", lines=5)
+        add_section_btn = gr.Button("➕ Ajouter/Mettre à jour la section")
+        generate_section_btn = gr.Button("🎤 Générer l'audio de la section")
+        audio_output = gr.Audio(label="🔊 Audio généré")
+    
+    # Fonctions de callback
+    def load_project(project_name, speaker, agree):
+        project_data = create_or_load_project(project_name, speaker, agree)
+        sections = [section["name"] for section in project_data["sections"]]
+        return project_data, gr.List.update(value=sections)
+    
+    def add_section(project_data, section_name, section_text):
+        updated_project = update_section(project_data, section_name, section_text)
+        sections = [section["name"] for section in updated_project["sections"]]
+        return updated_project, gr.List.update(value=sections)
+    
+    def generate_audio(project_data, section_name):
+        updated_project, audio_path = generate_section_audio(project_data, section_name)
+        return updated_project, audio_path
+    
+    # Connexion des événements
+    create_btn.click(load_project, inputs=[project_name, speaker, agree], outputs=[project_state, sections_list])
+    add_section_btn.click(add_section, inputs=[project_state, section_name, section_text], outputs=[project_state, sections_list])
+    generate_section_btn.click(generate_audio, inputs=[project_state, section_name], outputs=[project_state, audio_output])
+
+# Lancement de l'interface
 demo.launch()