Update app.py

app.py

@@ -9,9 +9,9 @@ from huggingface_hub import snapshot_download
 from nemo.collections import asr as nemo_asr
 import gradio as gr
 
-# 1. CONFIGURATION ET MODÈLES (LISTE COMPLÈTE)
+# 1. CONFIGURATION ET MODÈLES (INTÉGRALITÉ CONSERVÉE)
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-SEGMENT_DURATION = 5.0  # Ta préférence stricte pour Soloni
+SEGMENT_DURATION = 5.0 
 
 MODELS = {
     "Soloba V3 (CTC)":           ("RobotsMali/soloba-ctc-0.6b-v3", "ctc"),
@@ -38,14 +38,13 @@ def find_example_video():
     try:
         subprocess.run(f"wget {example_url} -O {target_path}", shell=True, check=True)
         return target_path
-    except Exception as e:
-        print(f"⚠️ Impossible de télécharger l'exemple : {e}")
+    except Exception:
         return None
 
 EXAMPLE_PATH = find_example_video()
 _cache = {}
 
-# 2. GESTION MÉMOIRE ET CHARGEMENT SÉCURISÉ (CORRECTIF BUG)
+# 2. GESTION MÉMOIRE ET CHARGEMENT SÉCURISÉ
 def clear_memory():
     _cache.clear()
     gc.collect()
@@ -61,17 +60,17 @@ def get_model(name):
     nemo_file = next((os.path.join(folder, f) for f in os.listdir(folder) if f.endswith(".nemo")), None)
     if not nemo_file: raise FileNotFoundError("Fichier .nemo introuvable.")
 
-    # CORRECTIF : On instancie le connecteur explicitement pour éviter l'erreur object.init()
+    # FIX CRITIQUE : Importation et instanciation explicite du connecteur
     from nemo.core.connectors.save_restore_connector import SaveRestoreConnector
-    connector = SaveRestoreConnector()
-
+    
+    # On force NeMo à utiliser cette instance précise pour éviter le bug __init__()
     model = nemo_asr.models.ASRModel.restore_from(
         nemo_file, 
         map_location=torch.device(DEVICE),
-        save_restore_connector=connector
+        save_restore_connector=SaveRestoreConnector()
     )
     
-    model.to(DEVICE).eval()
+    model.eval()
     if DEVICE == "cuda":
         model = model.half()
         
@@ -92,22 +91,21 @@ def pipeline(video_in, model_name):
             yield "❌ Aucune vidéo sélectionnée.", None
             return
         
-        yield "⏳ Phase 1/4 : Extraction audio...", None
+        yield "⏳ Extraction audio...", None
         full_wav = os.path.join(tmp_dir, "full.wav")
         subprocess.run(f"ffmpeg -y -i {shlex.quote(video_in)} -vn -ac 1 -ar 16000 {full_wav}", shell=True, check=True)
         
-        yield f"⏳ Phase 2/4 : Segmentation ({SEGMENT_DURATION}s)...", None
+        yield f"⏳ Segmentation ({SEGMENT_DURATION}s)...", None
         subprocess.run(f"ffmpeg -i {full_wav} -f segment -segment_time {SEGMENT_DURATION} -c copy {os.path.join(tmp_dir, 'seg_%03d.wav')}", shell=True, check=True)
         
         files = sorted(glob.glob(os.path.join(tmp_dir, "seg_*.wav")))
-        # Sécurité : Ignorer les fichiers vides (<1.5 Ko)
         valid_segments = [f for f in files if os.path.getsize(f) > 1500]
         
         if not valid_segments:
-            yield "❌ Erreur : Audio invalide.", None
+            yield "❌ Erreur : Audio trop court ou invalide.", None
             return
 
-        yield f"⏳ Phase 3/4 : Chargement de {model_name}...", None
+        yield f"⏳ Chargement de {model_name}...", None
         model = get_model(model_name)
         
         yield f"🎙️ Transcription de {len(valid_segments)} segments...", None
@@ -131,7 +129,7 @@ def pipeline(video_in, model_name):
                     "end": base_time + ((i+1) * gap)
                 })
 
-        yield "⏳ Phase 4/4 : Encodage vidéo...", None
+        yield "⏳ Encodage vidéo...", None
         srt_path = os.path.join(tmp_dir, "final.srt")
         with open(srt_path, "w", encoding="utf-8") as f:
             for i in range(0, len(all_words_ts), 6):
@@ -142,7 +140,7 @@ def pipeline(video_in, model_name):
         out_path = os.path.abspath(f"resultat_{int(time.time())}.mp4")
         safe_srt = srt_path.replace("\\", "/").replace(":", "\\:")
         
-        cmd = f"ffmpeg -y -i {shlex.quote(video_in)} -vf \"subtitles='{safe_srt}':force_style='Alignment=2,FontSize=18,PrimaryColour=&H00FFFF'\" -c:v libx264 -preset ultrafast -c:a copy {out_path}"
+        cmd = f"ffmpeg -y -i {shlex.quote(video_in)} -vf \"subtitles='{safe_srt}':force_style='Alignment=2,FontSize=18,PrimaryColour=&H00FFFF,Outline=1'\" -c:v libx264 -preset ultrafast -c:a copy {out_path}"
         subprocess.run(cmd, shell=True, check=True)
         
         yield "✅ Terminé !", out_path
@@ -163,13 +161,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             m_input = gr.Dropdown(choices=list(MODELS.keys()), value="Soloni V3 (TDT-CTC)", label="Modèle")
             run_btn = gr.Button("🚀 GÉNÉRER", variant="primary")
             
-            # Rétabli : Affichage des exemples si trouvés
             if EXAMPLE_PATH:
                 gr.Examples(examples=[[EXAMPLE_PATH, "Soloni V3 (TDT-CTC)"]], inputs=[v_input, m_input])
 
         with gr.Column():
             status = gr.Markdown("### État\nPrêt.")
-            v_output = gr.Video(label="Vidéo finale")
+            v_output = gr.Video(label="Résultat")
 
     run_btn.click(pipeline, [v_input, m_input], [status, v_output])