Update app.py

app.py

@@ -1,9 +1,6 @@
 # -*- coding: utf-8 -*-
 """ 
-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)
+ROBOTSMALI — Sous-titrage Bambara (VERSION COMPLÈTE V6.4 - FORCE OUTPUT)
 """
 import os
 import shlex
@@ -25,9 +22,6 @@ import gradio as gr
 
 # ---------------------------- # CONFIGURATION # ----------------------------
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-random.seed(1234)
-np.random.seed(1234)
-torch.manual_seed(1234)
 
 MODELS = {
     "Soloni V1 (RNNT)":        ("RobotsMali/soloni-114m-tdt-ctc-v1", "rnnt"),
@@ -54,9 +48,7 @@ def run_cmd(cmd):
 
 def load_model(name):
     if name in _cache: return _cache[name]
-    if len(_cache) > 0:
-        _cache.clear()
-        if torch.cuda.is_available(): torch.cuda.empty_cache()
+    _cache.clear() # Nettoyage RAM
     
     repo, mode = MODELS[name]
     folder = snapshot_download(repo, local_dir_use_symlinks=False)
@@ -74,9 +66,12 @@ def load_model(name):
     return model
 
 def burn_subtitles(video_path, words, duration):
-    # 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)
+    # SAUVEGARDE LOCALE PERSISTANTE
+    out_name = f"resultat_final.mp4" 
+    out_path = os.path.join(os.getcwd(), out_name)
+    
+    # Nettoyage d'un ancien fichier si présent
+    if os.path.exists(out_path): os.remove(out_path)
     
     chunk_size = 7
     with tempfile.NamedTemporaryFile(suffix=".srt", mode="w", encoding="utf-8", delete=False) as tf:
@@ -91,12 +86,12 @@ 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 ultra-compatible pour navigateur (H.264, YUV420p)
-    vf = f"subtitles={shlex.quote(srt_name)}:force_style='Fontsize=22,PrimaryColour=&HFFFFFF&,OutlineColour=&H000000&'"
+    # Commande FFmpeg forcée sur les codecs web standards
+    vf = f"subtitles={shlex.quote(srt_name)}:force_style='Fontsize=22,PrimaryColour=&HFFFFFF&'"
     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'-c:a aac -b:a 128k {shlex.quote(out_path)}'
+        f'-c:a aac -b:a 128k -movflags +faststart {shlex.quote(out_path)}'
     )
     run_cmd(cmd)
     os.remove(srt_name)
@@ -109,16 +104,14 @@ def pipeline(video_input, model_name):
         if not video_input: return "❌ Vidéo introuvable", None, None
         
         yield "⏳ Phase 1/3 : Analyse Audio...", None, None
-        with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as tf:
-            wav_path = tf.name
-        
+        wav_path = os.path.abspath("temp_audio.wav")
         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)}')
         
         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 {model_name}...", None, None
+        yield f"⏳ Phase 2/3 : Transcription IA...", None, None
         model = load_model(model_name)
         res = model.transcribe([wav_path])[0]
         text = res.text if hasattr(res, 'text') else str(res)
@@ -126,44 +119,36 @@ def pipeline(video_input, model_name):
         
         if not words: return "⚠️ Pas de parole détectée", None, None
         
-        yield "⏳ Phase 3/3 : Encodage de la vidéo...", None, None
+        yield "⏳ Phase 3/3 : Création de la vidéo...", None, None
         final_v = burn_subtitles(video_input, words, duration)
         
-        if os.path.exists(wav_path): os.remove(wav_path)
-        
-        yield "✅ Succès !", final_v, final_v
+        yield "✅ Succès ! Vidéo générée.", final_v, final_v
 
     except Exception as e:
-        traceback.print_exc()
         yield f"❌ Erreur: {str(e)}", None, None
 
 # ---------------------------- # INTERFACE # ----------------------------
 
 custom_css = """
 body { background-color: #0b0e14; }
-.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; }
+.gradio-container { background: rgba(17, 25, 40, 0.9) !important; border-radius: 15px; }
+.gr-button-primary { background: #10b981 !important; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
-    with gr.Column(elem_id="header"):
-        gr.HTML("<h1 style='color:#facc15;'>🤖 ROBOTSMALI</h1><p style='color:#94a3b8;'>Sous-titrage Bambara Intégral</p>")
+    gr.HTML("<h1 style='color:#facc15; text-align:center;'>🤖 ROBOTSMALI V6.4</h1>")
 
     with gr.Row():
         with gr.Column():
-            v_in = gr.Video(label="Source", mirror_webcam=False)
+            v_in = gr.Video(label="Source")
             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="Vidéo Sous-titrée")
-            f_out = gr.File(label="Lien de secours (Téléchargement)")
+            status = gr.Markdown("### Statut\n*Prêt*")
+            v_out = gr.Video(label="Résultat")
+            f_out = gr.File(label="Télécharger le fichier MP4")
 
     gr.Examples(examples=VIDEO_EXAMPLES, inputs=[v_in, m_sel], cache_examples=False)
-    
-    # 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)
+demo.launch(share=True, debug=True)