Update app.py

app.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 """
 ROBOTSMALI V38 FINAL — SOUS-TITRAGE BAMBARA (STYLE NETFLIX)
-Correction V38 : Durée exacte, QuartzNet fonctionnel, pipeline simplifiée
+Correction V38.1 : FFmpeg fixe (pas de -c copy), durée exacte, QuartzNet fonctionnel
 """
 
 import os, tempfile, traceback, random, textwrap
@@ -159,16 +159,29 @@ def align_vad(text,audio,sr,total_dur,top_db=28):
 def burn(video, subs):
     tmp_srt = tempfile.mktemp(suffix=".srt")
     out_file = "RobotsMali_Subtitled.mp4"
-    # Écriture SRT
+
     def sec_to_srt(t):
-        h=int(t//3600); m=int((t%3600)//60); s=int(t%60); ms=int((t-int(t))*1000)
+        h = int(t // 3600)
+        m = int((t % 3600) // 60)
+        s = int(t % 60)
+        ms = int((t - int(t)) * 1000)
         return f"{h:02}:{m:02}:{s:02},{ms:03}"
-    with open(tmp_srt,"w",encoding="utf-8") as f:
-        for i,(start,end,text) in enumerate(subs,1):
+
+    # Écriture du SRT
+    with open(tmp_srt, "w", encoding="utf-8") as f:
+        for i, (start, end, text) in enumerate(subs, 1):
             f.write(f"{i}\n{sec_to_srt(start)} --> {sec_to_srt(end)}\n{text}\n\n")
-    # Fusion vidéo + sous-titres sans changer durée
-    os.system(f'ffmpeg -y -i "{video}" -vf "subtitles={tmp_srt}" -c:v copy -c:a aac -b:a 192k "{out_file}"')
-    if os.path.exists(tmp_srt): os.remove(tmp_srt)
+
+    # ✅ Fusion avec réencodage (corrigé)
+    cmd = (
+        f'ffmpeg -y -i "{video}" '
+        f'-vf "subtitles={tmp_srt}:force_style=\'Fontsize=24,PrimaryColour=&HFFFFFF&,OutlineColour=&H000000&\'" '
+        f'-c:v libx264 -preset ultrafast -crf 23 -c:a copy "{out_file}"'
+    )
+    os.system(cmd)
+
+    if os.path.exists(tmp_srt):
+        os.remove(tmp_srt)
     return out_file
 
 # ----------------------------
@@ -177,7 +190,6 @@ def burn(video, subs):
 def pipeline(video, model_name):
     try:
         wav=tempfile.mktemp(suffix=".wav")
-        # Extraction audio
         extract_audio(video,wav)
         clean,audio,sr=clean_audio(wav)
         model=load_model(model_name)
@@ -210,8 +222,8 @@ def pipeline(video, model_name):
 # ----------------------------
 # INTERFACE GRADIO
 # ----------------------------
-with gr.Blocks(title="RobotsMali V38 Final") as demo:
-    gr.Markdown("## ⚡ RobotsMali V38 — Sous-titrage Style Netflix (QuartzNet & RNNT stable)")
+with gr.Blocks(title="RobotsMali V38.1 Final") as demo:
+    gr.Markdown("## ⚡ RobotsMali V38.1 — Sous-titrage Style Netflix (QuartzNet & RNNT stable)")
     v = gr.Video(label="Vidéo à sous-titrer")
     m = gr.Dropdown(list(MODELS.keys()), value="Soloba V1 (CTC)", label="Modèle ASR")
     b = gr.Button("▶️ Générer")