Update app.py

app.py

@@ -3,165 +3,134 @@ import os
 import time
 import gradio as gr
 from TTS.api import TTS
+import json
 
 # Initialisation du modèle TTS
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
-output_folder = "output_audio"
-os.makedirs(output_folder, exist_ok=True)
+# Dossier de base pour les projets
+base_output_folder = "projects"
+os.makedirs(base_output_folder, exist_ok=True)
 
-def create_project(project_name, speaker, agree):
+# 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.")
+        raise gr.Error("❗ Veuillez accepter les conditions d'utilisation.")
     
-    project_folder = os.path.join(output_folder, project_name.strip())
+    project_folder = os.path.join(base_output_folder, project_name)
     os.makedirs(project_folder, exist_ok=True)
-    return project_folder
-
-def generate_audio_section(text, section_name, project_folder, speaker):
-    file_name = f"{section_name.strip()}.wav"
-    output_path = os.path.join(project_folder, file_name)
+    
+    project_data = {
+        "name": project_name,
+        "speaker": speaker,
+        "sections": []
+    }
+    
+    # Vérifier si le projet existe déjà
+    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
 
-    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 gr.Error(f"🚨 Aucun fichier audio trouvé pour le speaker : {speaker}")
+# 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(section_name, text, speaker):
+    output_path = os.path.join(base_output_folder, f"{section_name}.wav")
+    
     tts.tts_to_file(
         text=text,
         file_path=output_path,
-        speaker_wav=speaker_wav_paths,
+        speaker_wav=[os.path.join("examples", f) for f in os.listdir("examples") if f.startswith(speaker) and f.endswith(".wav")],
         language="fr"
     )
-
+    
     return output_path
 
-custom_css = """
-.gradio-container {
-    font-family: 'Roboto', sans-serif;
-    background-color: #f7f9fc;
-}
-.gr-form {
-    background-color: white;
-    border-radius: 15px;
-    padding: 30px;
-    box-shadow: 0 8px 16px rgba(0, 0, 0, 0.1);
-}
-.gr-button {
-    background-color: #4a90e2;
-    border: none;
-    color: white;
-    font-weight: bold;
-    transition: all 0.3s ease;
-}
-.gr-button:hover {
-    background-color: #3a7bc8;
-    transform: translateY(-2px);
-}
-.gr-input, .gr-dropdown {
-    border: 1px solid #e0e0e0;
-    border-radius: 8px;
-    padding: 10px;
-}
-.gr-checkbox {
-    margin-top: 10px;
-}
-.gr-form > div {
-    margin-bottom: 20px;
-}
-"""
-
-title = "🎙️ Synthèse Vocale XTTS"
-description = """
-<h3 style='text-align: center; margin-bottom: 1em;'>Bienvenue sur notre outil de synthèse vocale XTTS ! 🌟</h3>
-<p style='text-align: center;'>Générez une voix naturelle à partir de votre texte en français. Choisissez une voix, entrez votre texte, et écoutez le résultat ! 🎤</p>
-"""
-article = """
-<div style='margin: 20px auto; text-align: center; padding: 10px; background-color: #e8f0fe; border-radius: 10px;'>
-<p>En utilisant cette démo, vous acceptez les <a href='https://coqui.ai/cpml' target='_blank' style='color: #4a90e2; text-decoration: none;'>conditions d'utilisation du modèle Coqui Public</a> 📜</p>
-</div>
-"""
-
-available_speakers = list(set([f.split('_')[0] for f in os.listdir("examples") if f.endswith(".wav")]))
-
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown(f"<h1 style='text-align: center; color: #4a90e2;'>{title}</h1>")
-    gr.Markdown(description)
+# Interface Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎙️ Projet Margaux TTS")
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            project_name = gr.Textbox(
-                label="🗂️ Nom du projet",
-                placeholder="Entrez le nom du projet",
-                lines=1
-            )
-            speaker = gr.Dropdown(
-                label="🎤 Voix",
-                choices=available_speakers,
-                value=available_speakers[0] if available_speakers else None
-            )
-            agree = gr.Checkbox(
-                label="✅ J'accepte les conditions d'utilisation",
-                value=False
-            )
-            create_project_btn = gr.Button("🔨 Créer le projet", variant="primary")
+    # Étape 1: Création ou chargement du projet
+    with gr.Tab("🔧 Création 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")
+        create_btn = gr.Button("🚀 Créer/Charger le Projet")
     
-    project_folder_output = gr.State()
+    # Étape 2: Gestion des sections
+    with gr.Tab("📋 Gestion des Sections"):
+        project_info = gr.JSON(label="📁 Informations du projet", visible=False)
+        sections_list = gr.List(label="📜 Sections du projet")
+        
+        # Entrées pour ajouter une nouvelle section
+        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")
 
-    def on_create_project(name, spkr, agr):
-        folder_path = create_project(name, spkr, agr)
-        return folder_path
-    
-    create_project_btn.click(
-        on_create_project,
-        inputs=[project_name, speaker, agree],
-        outputs=[project_folder_output]
-    )
+        # Section pour générer l'audio de chaque section individuellement
+        audio_outputs_display = []
 
-    section_textboxes = []
-    
-    def add_section():
-        section_textbox = gr.Textbox(
-            label=f"✏️ Texte de la section {len(section_textboxes) + 1}",
-            placeholder="Entrez le texte pour cette section",
-            lines=5
-        )
-        section_name_textbox = gr.Textbox(
-            label=f"📝 Nom de la section {len(section_textboxes) + 1}",
-            placeholder="Entrez le nom du fichier pour cette section",
-            lines=1
-        )
-        section_textboxes.append((section_textbox, section_name_textbox))
-        return section_textbox, section_name_textbox
-    
-    add_section_btn = gr.Button("➕ Ajouter une section")
-    
-    def remove_section():
-        if section_textboxes:
-            removed_section = section_textboxes.pop()
-            return removed_section[0], removed_section[1]
-    
-    remove_section_btn = gr.Button("➖ Supprimer une section")
-    
-    add_section_btn.click(add_section)
-    remove_section_btn.click(remove_section)
+    # 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, sections
 
-    generate_audio_btn = gr.Button("🎶 Générer l'audio")
-    
-    def generate_all_sections(sections_data, proj_folder):
-        audio_outputs = []
-        for text_box, name_box in sections_data:
-            text = text_box.value
-            name = name_box.value
-            audio_path = generate_audio_section(text, name.strip(), proj_folder.value, speaker.value)
-            audio_outputs.append(audio_path)
-        return audio_outputs
-    
-    audio_outputs_display = [gr.Audio(label=f"🔊 Audio Section {i+1}") for i in range(len(section_textboxes))]
-    
-    generate_audio_btn.click(
-        generate_all_sections,
-        inputs=[section_textboxes, project_folder_output],
-        outputs=audio_outputs_display
-    )
-    
-demo.launch(debug=True)
+    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, sections
+
+    def generate_audio(section_name, text):
+        audio_path = generate_section_audio(section_name, text, speaker.value)
+        
+        return audio_path
+
+    # Connexion des événements
+    create_btn.click(load_project, inputs=[project_name, speaker, agree], outputs=[project_info, sections_list])
+    add_section_btn.click(add_section, inputs=[project_info, section_name, section_text], outputs=[project_info, sections_list])
+
+    # Dynamique : Créer des boutons de génération audio pour chaque section
+    def update_audio_buttons(sections):
+        global audio_outputs_display
+        
+        audio_outputs_display.clear()
+        
+        for sec in sections:
+            audio_output_btn = gr.Button(f"🎤 Générer l'audio pour {sec}")
+            audio_output_display = gr.Audio(label=f"🔊 Audio de {sec}", type="filepath", visible=False)
+
+            def generate_for_section(sec=sec):
+                text_for_section = next((s['text'] for s in project_info['sections'] if s['name'] == sec), "")
+                return generate_audio(sec, text_for_section)
+
+            audio_output_btn.click(generate_for_section)
+            audio_outputs_display.append(audio_output_display)
+
+            # Ajouter les boutons et les affichages audio à l'interface
+            demo.add(audio_output_btn)
+            demo.add(audio_output_display)
+
+# Lancement de l'interface
+demo.launch()