Update app.py

app.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """ 
-ROBOTSMALI — Sous-titrage Bambara (VERSION COMPLÈTE V6.4 - FORCE OUTPUT)
+ROBOTSMALI — Sous-titrage Bambara (VERSION COMPLÈTE V6.5 - INTERFACE ÉPURÉE)
 """
 import os
 import shlex
@@ -22,6 +22,9 @@ 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"),
@@ -48,7 +51,9 @@ def run_cmd(cmd):
 
 def load_model(name):
     if name in _cache: return _cache[name]
-    _cache.clear() # Nettoyage RAM
+    if len(_cache) > 0:
+        _cache.clear()
+        if torch.cuda.is_available(): torch.cuda.empty_cache()
     
     repo, mode = MODELS[name]
     folder = snapshot_download(repo, local_dir_use_symlinks=False)
@@ -66,12 +71,9 @@ def load_model(name):
     return model
 
 def burn_subtitles(video_path, words, duration):
-    # 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)
+    # Sauvegarde dans le dossier de travail pour la stabilité
+    out_name = f"resultat_robotsmali.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:
@@ -86,8 +88,7 @@ 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 FFmpeg forcée sur les codecs web standards
-    vf = f"subtitles={shlex.quote(srt_name)}:force_style='Fontsize=22,PrimaryColour=&HFFFFFF&'"
+    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 '
@@ -101,9 +102,9 @@ def burn_subtitles(video_path, words, duration):
 
 def pipeline(video_input, model_name):
     try:
-        if not video_input: return "❌ Vidéo introuvable", None, None
+        if not video_input: return "❌ Vidéo introuvable", None
         
-        yield "⏳ Phase 1/3 : Analyse Audio...", None, None
+        yield "⏳ Phase 1/3 : Analyse Audio...", None
         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)}')
         
@@ -111,44 +112,53 @@ def pipeline(video_input, model_name):
                                  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...", None, None
+        yield f"⏳ Phase 2/3 : Transcription {model_name}...", 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, None
+        if not words: return "⚠️ Pas de parole détectée", None
         
-        yield "⏳ Phase 3/3 : Création de la vidéo...", None, None
+        yield "⏳ Phase 3/3 : Encodage de la vidéo...", None
         final_v = burn_subtitles(video_input, words, duration)
         
-        yield "✅ Succès ! Vidéo générée.", final_v, final_v
+        if os.path.exists(wav_path): os.remove(wav_path)
+        
+        yield "✅ Sous-titrage terminé !", final_v
 
     except Exception as e:
-        yield f"❌ Erreur: {str(e)}", None, None
+        traceback.print_exc()
+        yield f"❌ Erreur: {str(e)}", None
 
 # ---------------------------- # INTERFACE # ----------------------------
 
 custom_css = """
 body { background-color: #0b0e14; }
-.gradio-container { background: rgba(17, 25, 40, 0.9) !important; border-radius: 15px; }
-.gr-button-primary { background: #10b981 !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; border: none !important; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
-    gr.HTML("<h1 style='color:#facc15; text-align:center;'>🤖 ROBOTSMALI V6.4</h1>")
+    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="Source")
+            gr.Markdown("### 📥 Source")
+            v_in = gr.Video(label=None, 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("### Statut\n*Prêt*")
-            v_out = gr.Video(label="Résultat")
-            f_out = gr.File(label="Télécharger le fichier MP4")
+            gr.Markdown("### 📤 Résultat")
+            status = gr.Markdown("*Prêt*")
+            # Une seule case vidéo pour le résultat
+            v_out = gr.Video(label="Vidéo finale (lecture et 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, f_out])
+    
+    btn.click(pipeline, [v_in, m_sel], [status, v_out])
 
-demo.launch(share=True, debug=True)
+if __name__ == "__main__":
+    demo.launch(share=True, debug=True)