Update app.py

app.py

@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 """ 
-ROBOTSMALI — Sous-titrage Bambara (VERSION INTÉGRALE V6.1 - FIX FINAL OUTPUT)
+ROBOTSMALI — Sous-titrage Bambara (VERSION COMPLÈTE V6.3)
+- Tous les modèles inclus
+- Correction du clic sur exemple
+- Correction de l'affichage vidéo (Output Fix)
 """
 import os
 import shlex
@@ -10,7 +13,6 @@ import traceback
 import random
 import textwrap
 import time
-import shutil
 from pathlib import Path
 
 import numpy as np
@@ -42,12 +44,12 @@ VIDEO_EXAMPLES = [
 
 _cache = {}
 
-# ---------------------------- # FONCTIONS TECHNIQUES # ----------------------------
+# ---------------------------- # FONCTIONS IA & FFmpeg # ----------------------------
 
 def run_cmd(cmd):
     res = subprocess.run(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
     if res.returncode != 0:
-        raise RuntimeError(f"Erreur FFmpeg: {res.stdout}")
+        raise RuntimeError(f"FFmpeg: {res.stdout}")
     return res.stdout
 
 def load_model(name):
@@ -72,8 +74,9 @@ def load_model(name):
     return model
 
 def burn_subtitles(video_path, words, duration):
-    # Création d'un fichier de sortie dans un dossier temporaire Gradio
-    out_path = os.path.join(tempfile.gettempdir(), f"final_output_{int(time.time())}.mp4")
+    # On crée le fichier dans le dossier courant pour éviter les erreurs de permissions
+    out_name = f"resultat_robotsmali_{int(time.time())}.mp4"
+    out_path = os.path.abspath(out_name)
     
     chunk_size = 7
     with tempfile.NamedTemporaryFile(suffix=".srt", mode="w", encoding="utf-8", delete=False) as tf:
@@ -88,77 +91,79 @@ def burn_subtitles(video_path, words, duration):
             tf.write(f"{i+1}\n{t_srt(start)} --> {t_srt(end)}\n{txt}\n\n")
         srt_name = tf.name
 
-    # Commande d'encodage optimisée pour le Web (H.264 Baseline + Faststart)
+    # Commande ultra-compatible pour navigateur (H.264, YUV420p)
     vf = f"subtitles={shlex.quote(srt_name)}:force_style='Fontsize=22,PrimaryColour=&HFFFFFF&,OutlineColour=&H000000&'"
     cmd = (
         f'ffmpeg -hide_banner -loglevel error -y -i {shlex.quote(video_path)} '
         f'-vf {shlex.quote(vf)} -c:v libx264 -pix_fmt yuv420p -preset ultrafast -crf 28 '
-        f'-movflags +faststart -c:a copy {shlex.quote(out_path)}'
+        f'-c:a aac -b:a 128k {shlex.quote(out_path)}'
     )
     run_cmd(cmd)
     os.remove(srt_name)
     return out_path
 
+# ---------------------------- # PIPELINE # ----------------------------
+
 def pipeline(video_input, model_name):
     try:
-        if not video_input: return "❌ Veuillez charger une vidéo", None
+        if not video_input: return "❌ Vidéo introuvable", None, None
         
-        yield "⏳ Phase 1/3 : Analyse Audio...", None
+        yield "⏳ Phase 1/3 : Analyse Audio...", None, None
         with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as tf:
             wav_path = tf.name
         
-        # Extraction stable
         run_cmd(f'ffmpeg -hide_banner -loglevel error -y -i {shlex.quote(video_input)} -vn -ac 1 -ar 16000 -f wav {shlex.quote(wav_path)}')
         
-        # Récupération durée
         dur_out = subprocess.run(f'ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 {shlex.quote(video_input)}', 
                                  shell=True, stdout=subprocess.PIPE, text=True).stdout
         duration = float(dur_out.strip()) if dur_out.strip() else 10.0
         
-        yield f"⏳ Phase 2/3 : Transcription IA ({model_name})...", None
+        yield f"⏳ Phase 2/3 : Transcription {model_name}...", None, None
         model = load_model(model_name)
         res = model.transcribe([wav_path])[0]
         text = res.text if hasattr(res, 'text') else str(res)
         words = [w for w in text.split() if len(w) > 1]
         
-        if not words: return "⚠️ Pas de parole détectée.", None
+        if not words: return "⚠️ Pas de parole détectée", None, None
         
-        yield "⏳ Phase 3/3 : Encodage vidéo final...", None
+        yield "⏳ Phase 3/3 : Encodage de la vidéo...", None, None
         final_v = burn_subtitles(video_input, words, duration)
         
         if os.path.exists(wav_path): os.remove(wav_path)
         
-        # On force Gradio à renvoyer le chemin absolu
-        yield "✅ Succès !", gr.update(value=final_v)
+        yield "✅ Succès !", final_v, final_v
 
     except Exception as e:
         traceback.print_exc()
-        yield f"❌ Erreur: {str(e)}", None
+        yield f"❌ Erreur: {str(e)}", None, None
 
 # ---------------------------- # INTERFACE # ----------------------------
 
 custom_css = """
 body { background-color: #0b0e14; }
-.gradio-container { background: rgba(17, 25, 40, 0.8) !important; backdrop-filter: blur(12px); border-radius: 20px; border: 1px solid rgba(255, 255, 255, 0.1); }
-#title-header { text-align: center; padding: 20px; }
-.gr-button-primary { background: linear-gradient(135deg, #059669, #10b981) !important; border: none !important; }
+.gradio-container { background: rgba(17, 25, 40, 0.8) !important; border-radius: 20px; border: 1px solid rgba(255, 255, 255, 0.1); }
+#header { text-align: center; padding: 20px; }
+.gr-button-primary { background: linear-gradient(135deg, #059669, #10b981) !important; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
-    with gr.Column(elem_id="title-header"):
-        gr.HTML("<h1 style='color:#facc15;'>🤖 ROBOTSMALI</h1><p style='color:#94a3b8;'>Sous-titrage Bambara Professionnel</p>")
+    with gr.Column(elem_id="header"):
+        gr.HTML("<h1 style='color:#facc15;'>🤖 ROBOTSMALI</h1><p style='color:#94a3b8;'>Sous-titrage Bambara Intégral</p>")
 
     with gr.Row():
         with gr.Column():
-            v_in = gr.Video(label="Entrée", mirror_webcam=False)
-            m_sel = gr.Dropdown(list(MODELS.keys()), value="Soloba V1 (CTC)", label="Modèle")
+            v_in = gr.Video(label="Source", mirror_webcam=False)
+            m_sel = gr.Dropdown(list(MODELS.keys()), value="Soloba V1 (CTC)", label="Modèle IA")
             btn = gr.Button("🚀 GÉNÉRER", variant="primary")
         with gr.Column():
             status = gr.Markdown("*Prêt*")
-            v_out = gr.Video(label="Résultat Final")
+            v_out = gr.Video(label="Vidéo Sous-titrée")
+            f_out = gr.File(label="Lien de secours (Téléchargement)")
 
     gr.Examples(examples=VIDEO_EXAMPLES, inputs=[v_in, m_sel], cache_examples=False)
-    btn.click(pipeline, [v_in, m_sel], [status, v_out])
+    
+    # On renvoie vers status, le lecteur vidéo ET le composant fichier
+    btn.click(pipeline, [v_in, m_sel], [status, v_out, f_out])
 
 if __name__ == "__main__":
     demo.launch(share=True, debug=True)