Update app.py

app.py

@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 """ 
-ROBOTSMALI — Sous-titrage Bambara (VERSION COMPLÈTE V6.5 - INTERFACE ÉPURÉE)
+ROBOTSMALI — Sous-titrage Bambara (VERSION COMPLÈTE V6.6)
+- Interface à deux colonnes (Entrée / Résultat unique)
+- Conservation du Statut détaillé (Phases de traitement)
+- Tous les modèles RobotsMali inclus
 """
 import os
 import shlex
@@ -41,12 +44,12 @@ VIDEO_EXAMPLES = [
 
 _cache = {}
 
-# ---------------------------- # FONCTIONS IA & FFmpeg # ----------------------------
+# ---------------------------- # FONCTIONS TECHNIQUES # ----------------------------
 
 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"FFmpeg: {res.stdout}")
+        raise RuntimeError(f"FFmpeg Error: {res.stdout}")
     return res.stdout
 
 def load_model(name):
@@ -71,8 +74,7 @@ def load_model(name):
     return model
 
 def burn_subtitles(video_path, words, duration):
-    # Sauvegarde dans le dossier de travail pour la stabilité
-    out_name = f"resultat_robotsmali.mp4"
+    out_name = f"robotsmali_output_{int(time.time())}.mp4"
     out_path = os.path.abspath(out_name)
     
     chunk_size = 7
@@ -102,9 +104,12 @@ def burn_subtitles(video_path, words, duration):
 
 def pipeline(video_input, model_name):
     try:
-        if not video_input: return "❌ Vidéo introuvable", None
+        if not video_input:
+            yield "❌ Erreur : Aucune vidéo chargée.", None
+            return
         
-        yield "⏳ Phase 1/3 : Analyse Audio...", None
+        # Phase 1 : Audio
+        yield "⏳ Phase 1/3 : Analyse et extraction de l'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)}')
         
@@ -112,26 +117,30 @@ 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 {model_name}...", None
+        # Phase 2 : Transcription
+        yield f"⏳ Phase 2/3 : Transcription IA ({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
+        if not words:
+            yield "⚠️ Phase 2 : Aucune parole détectée dans l'audio.", None
+            return
         
-        yield "⏳ Phase 3/3 : Encodage de la vidéo...", None
+        # Phase 3 : Encodage
+        yield "⏳ Phase 3/3 : Génération de la vidéo sous-titrée...", None
         final_v = burn_subtitles(video_input, words, duration)
         
         if os.path.exists(wav_path): os.remove(wav_path)
         
-        yield "✅ Sous-titrage terminé !", final_v
+        yield "✅ Traitement terminé avec succès !", final_v
 
     except Exception as e:
         traceback.print_exc()
-        yield f"❌ Erreur: {str(e)}", None
+        yield f"❌ Erreur critique : {str(e)}", None
 
-# ---------------------------- # INTERFACE # ----------------------------
+# ---------------------------- # INTERFACE GRADIO # ----------------------------
 
 custom_css = """
 body { background-color: #0b0e14; }
@@ -150,11 +159,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
             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():
             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)")
+            # Le Statut est conservé ici pour informer sur l'état des fichiers audio/vidéo
+            status = gr.Markdown("### État\n*En attente de démarrage...*")
+            v_out = gr.Video(label="Vidéo finale")
 
     gr.Examples(examples=VIDEO_EXAMPLES, inputs=[v_in, m_sel], cache_examples=False)