Update app.py

app.py

@@ -1,115 +1,167 @@
 import subprocess
-import gradio as gr
-from TTS.api import TTS
 import os
 import time
-import torch
-import json
+import gradio as gr
+from TTS.api import TTS
 
 # Initialisation du modèle TTS
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
-# Dossier de base pour les projets
-base_output_folder = "projects"
-os.makedirs(base_output_folder, exist_ok=True)
+output_folder = "output_audio"
+os.makedirs(output_folder, exist_ok=True)
 
-# Fonction pour créer ou charger un projet
-def create_or_load_project(project_name, speaker, agree):
+def create_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(base_output_folder, project_name)
+    project_folder = os.path.join(output_folder, project_name.strip())
     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)
+
+    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}")
+
+    tts.tts_to_file(
+        text=text,
+        file_path=output_path,
+        speaker_wav=speaker_wav_paths,
+        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)
     
-    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)
+    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")
     
-    return project_data
+    project_folder_output = gr.State()
 
-# 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
+    def on_create_project(name, spkr, agr):
+        folder_path = create_project(name, spkr, agr)
+        return folder_path
     
-    project_data["sections"].append({
-        "name": section_name,
-        "text": section_text,
-        "audio_path": None
-    })
-    return project_data
+    create_project_btn.click(
+        on_create_project,
+        inputs=[project_name, speaker, agree],
+        outputs=[project_folder_output]
+    )
 
-# 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
+    section_textboxes = []
     
-    raise gr.Error(f"❌ Section '{section_name}' non trouvée.")
-
-# Interface Gradio
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎙️ Projet Margaux TTS")
+    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
     
-    # État du projet
-    project_state = gr.State(value={})
+    add_section_btn = gr.Button("➕ Ajouter une section")
     
-    # É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")
-        create_btn = gr.Button("🚀 Créer/Charger le Projet")
+    def remove_section():
+        if section_textboxes:
+            removed_section = section_textboxes.pop()
+            return removed_section[0], removed_section[1]
     
-    # É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é")
+    remove_section_btn = gr.Button("➖ Supprimer une 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, gr.List.update(value=sections)
+    add_section_btn.click(add_section)
+    remove_section_btn.click(remove_section)
+
+    generate_audio_btn = gr.Button("🎶 Générer l'audio")
     
-    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_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
     
-    def generate_audio(project_data, section_name):
-        updated_project, audio_path = generate_section_audio(project_data, section_name)
-        return updated_project, audio_path
+    audio_outputs_display = [gr.Audio(label=f"🔊 Audio Section {i+1}") for i in range(len(section_textboxes))]
     
-    # 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()
+    generate_audio_btn.click(
+        generate_all_sections,
+        inputs=[section_textboxes, project_folder_output],
+        outputs=audio_outputs_display
+    )
+    
+demo.launch(debug=True)