Update app.py

app.py

@@ -1,10 +1,13 @@
 import subprocess
-import os
 import gradio as gr
 from TTS.api import TTS
+import os
+import time
+
 # Initialisation du modèle TTS
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=False)
 
+# Dossier pour les fichiers audio générés
 output_folder = "output_audio"
 os.makedirs(output_folder, exist_ok=True)
 
@@ -12,25 +15,24 @@ def create_project(project_name, speaker, agree):
     if not agree:
         raise gr.Error("Veuillez accepter les conditions d'utilisation.")
     
-    project_folder = os.path.join(output_folder, project_name.strip())
+    project_folder = os.path.join(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)
-
+def generate_audio(prompt, speaker, project_folder, section_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}")
 
+    output_path = os.path.join(project_folder, f"{section_name}.wav")
+    
     tts.tts_to_file(
-        text=text,
+        text=prompt,
         file_path=output_path,
         speaker_wav=speaker_wav_paths,
         language="fr"
     )
-
+    
     return output_path
 
 custom_css = """
@@ -71,104 +73,66 @@ custom_css = """
 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>
+<p style='text-align: center;'>Générez une voix naturelle à partir de votre texte en français. Créez un projet et ajoutez des sections pour votre script.</p>
 """
 
 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:
+    
+    # Étape 1 : Création du Projet
     gr.Markdown(f"<h1 style='text-align: center; color: #4a90e2;'>{title}</h1>")
     gr.Markdown(description)
     
     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")
+        project_name = gr.Textbox(label="Nom du projet", placeholder="Entrez le nom du projet")
+        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)
     
-    project_folder_output = gr.State()
-
-    def on_create_project(name, spkr, agr):
-        folder_path = create_project(name, spkr, agr)
-        return folder_path
+    create_project_btn = gr.Button("Créer le Projet", variant="primary")
     
-    create_project_btn.click(
-        on_create_project,
-        inputs=[project_name, speaker, agree],
-        outputs=[project_folder_output]
-    )
-
-    section_textboxes = []
+    # Étape 2 : Gestion des Sections
+    sections = []
     
     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")
+        section_name = f"Section {len(sections) + 1}"
+        sections.append({"name": section_name, "text": ""})
+        return sections
     
-    def remove_section():
-        if section_textboxes:
-            removed_section = section_textboxes.pop()
-            return removed_section[0], removed_section[1]
+    def remove_section(index):
+        if index < len(sections):
+            sections.pop(index)
+        return sections
     
-    remove_section_btn = gr.Button("- Supprimer une section")
+    def update_section_text(index, text):
+        if index < len(sections):
+            sections[index]["text"] = text
+        return sections
     
-    add_section_btn.click(add_section)
+    section_outputs = []
     
-    # Outputs for the sections
-    added_sections_output = []
-
-    def update_sections_output():
-        return [section[0] for section in section_textboxes], [section[1] for section in section_textboxes]
-
-    add_section_btn.click(update_sections_output)
+    with gr.Column():
+        section_list = gr.State(value=sections)
+        
+        def render_sections(sections):
+            return [gr.Row([
+                gr.Textbox(label=f"Texte pour {section['name']}", value=section['text'], lines=5).change(update_section_text, inputs=[sections.index(section)], outputs=None),
+                gr.Textbox(label="Nom du fichier", value=section['name'], interactive=False),
+                gr.Button("Régénérer", variant="secondary").click(generate_audio, inputs=[section['text'], speaker, project_folder, section['name']], outputs=[gr.Audio(label=f"Audio {section['name']}")])
+            ]) for section in sections]
+        
+        with gr.Row():
+            add_section_btn = gr.Button("+ Ajouter une section")
+            remove_section_btn = gr.Button("- Supprimer la dernière section")
+        
+        add_section_btn.click(add_section, outputs=section_list)
+        remove_section_btn.click(remove_section, inputs=[len(sections)-1], outputs=section_list)
+        
+        section_output_area = gr.Column()
     
-    remove_section_btn.click(update_sections_output)
+        # Dynamically render sections
+        with section_output_area:
+            for section in render_sections(sections):
+                section_output_area.append(section)
 
-    generate_audio_btn = gr.Button("Générer l'audio")
-    
-    def generate_all_sections(proj_folder):
-        audio_outputs = []
-        for text_box, name_box in section_textboxes:
-            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=[project_folder_output],
-        outputs=audio_outputs_display
-    )
-    
 demo.launch(debug=True)