Spaces:
Sleeping
Sleeping
File size: 4,832 Bytes
a5c86e8 163771f 9883968 6bca271 f5875da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
import subprocess
import os
import gradio as gr
from TTS.api import TTS
# 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)
|