Update app.py

app.py

@@ -5,55 +5,92 @@ import scipy.io.wavfile as wavfile
 import numpy as np
 import tempfile
 
-print("Chargement du modèle...")
+print("Chargement du modèle TTS Pulaar...")
 model = VitsModel.from_pretrained("morlayecis0003/tts-pulaar")
 tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-ful")
 model.eval()
-print("Modèle chargé!")
+print("✅ Modèle chargé avec succès!")
 
-def synthesize_speech(text, volume_boost=1.0):
-    if not text or not text.strip():
-        return None, "Veuillez entrer un texte"
+def synthèse_vocale(texte, volume=1.0):
+    if not texte or not texte.strip():
+        return None, "❌ Veuillez entrer un texte"
     
-    inputs = tokenizer(text, return_tensors="pt")
+    try:
+        # Tokenisation
+        entrées = tokenizer(texte, return_tensors="pt")
+        
+        # Génération
+        with torch.no_grad():
+            sorties = model(**entrées)
+        
+        # Récupérer l'audio
+        audio = sorties.waveform[0].cpu().numpy()
+        
+        # Normalisation du volume
+        max_valeur = np.max(np.abs(audio))
+        if max_valeur > 0:
+            audio = audio * (0.9 * volume / max_valeur)
+        audio = np.clip(audio, -1.0, 1.0)
+        
+        # Conversion pour sauvegarde
+        audio_int16 = (audio * 32767).astype(np.int16)
+        
+        # Sauvegarde temporaire
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
+            wavfile.write(f.name, 16000, audio_int16)
+            return f.name, f"✅ Succès! Durée: {len(audio)/16000:.1f} secondes"
     
-    with torch.no_grad():
-        outputs = model(**inputs)
-    
-    audio = outputs.waveform[0].cpu().numpy()
+    except Exception as e:
+        return None, f"❌ Erreur: {str(e)}"
+
+# Interface Gradio
+with gr.Blocks(title="TTS Pulaar", theme=gr.themes.Soft()) as interface:
+    gr.Markdown("""
+    # 🗣️ Synthèse Vocale en Pulaar
+    ### Entrez un texte en pulaar et générez l'audio correspondant
+    """)
     
-    # Normalisation
-    max_val = np.max(np.abs(audio))
-    if max_val > 0:
-        audio = audio * (0.9 * volume_boost / max_val)
-    audio = np.clip(audio, -1.0, 1.0)
+    with gr.Row():
+        with gr.Column(scale=2):
+            texte_entrée = gr.Textbox(
+                label="📝 Texte en Pulaar",
+                placeholder="Exemple: A jaaraama, No ñawdiri?",
+                lines=4
+            )
+            volume_entrée = gr.Slider(
+                label="🔊 Volume", 
+                minimum=0.5, 
+                maximum=1.5, 
+                value=1.0, 
+                step=0.1
+            )
+            bouton = gr.Button("🎵 Générer l'audio", variant="primary")
+        
+        with gr.Column(scale=1):
+            sortie_audio = gr.Audio(label="🎧 Audio généré")
+            sortie_statut = gr.Textbox(label="📊 Statut", lines=2)
     
-    audio_int16 = (audio * 32767).astype(np.int16)
+    # Exemples
+    gr.Markdown("### 📋 Exemples de textes")
+    exemples = gr.Examples(
+        examples=[
+            ["A jaaraama", 1.0],
+            ["No ñawdiri?", 1.0],
+            ["Miɗo yiɗi pulaar", 1.0],
+            ["Jamma waali", 1.0],
+            ["Miɗo hoɗi Dakaar", 1.0]
+        ],
+        inputs=[texte_entrée, volume_entrée],
+        label="Cliquez sur un exemple pour le tester"
+    )
     
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
-        wavfile.write(f.name, 16000, audio_int16)
-        return f.name, f"✅ Succès! Durée: {len(audio)/16000:.1f}s"
-
-# Interface plus simple
-demo = gr.Interface(
-    fn=synthesize_speech,
-    inputs=[
-        gr.Textbox(label="Texte en pulaar", lines=3, placeholder="Ex: A jaaraama, No ñawdiri?"),
-        gr.Slider(0.5, 1.5, value=1.0, label="Volume")
-    ],
-    outputs=[
-        gr.Audio(label="Audio généré"),
-        gr.Textbox(label="Statut")
-    ],
-    title="🗣️ Synthèse Vocale Pulaar",
-    description="Entrez du texte en pulaar pour générer l'audio",
-    examples=[
-        ["A jaaraama", 1.0],
-        ["No ñawdiri?", 1.0],
-        ["Miɗo yiɗi pulaar", 1.0],
-        ["Jamma waali", 1.0]
-    ]
-)
+    # Lier la fonction
+    bouton.click(
+        fn=synthèse_vocale,
+        inputs=[texte_entrée, volume_entrée],
+        outputs=[sortie_audio, sortie_statut]
+    )
 
+# Lancer l'application
 if __name__ == "__main__":
-    demo.launch()
+    interface.launch()