Update app.py

app.py

@@ -75,6 +75,29 @@ print("Gemma pret !")
 current_audio    = {"path": None, "filename": None}
 current_exchange = {"transcription": None, "reponse": None}
 stop_generation = {"value": False}
+audio_state = {"mic_ready": False, "file_ready": False}
+
+
+def btn_states():
+    """
+    Retourne les mises à jour (interactive=True/False) pour mic_btn et file_btn
+    en fonction de la présence d'un audio micro / fichier, à utiliser une fois
+    qu'un traitement (transcription/génération) est terminé.
+    """
+    mic_label = "🎤 Transcrire la voix"
+    file_label = "📂 Transcrire le fichier"
+    return (
+        gr.update(interactive=audio_state["mic_ready"], value=mic_label),
+        gr.update(interactive=audio_state["file_ready"], value=file_label),
+    )
+
+
+def btn_states_disabled():
+    """Désactive les deux boutons de transcription (traitement en cours)."""
+    return (
+        gr.update(interactive=False, value="🎤 Transcrire la voix"),
+        gr.update(interactive=False, value="📂 Transcrire le fichier"),
+    )
 
 # ── Helpers ───────────────────────────────────────────────────────
 def generate_audio_name():
@@ -262,12 +285,25 @@ def generate_gemma_response(transcription: str, progress=None):
 def transcribe_only(audio, progress=gr.Progress()):
     path = process_audio(audio)
     if path is None:
-        return (
+        mic_u, file_u = btn_states()
+        yield (
             "Aucun audio recu.",
             gr.update(visible=False),
             gr.update(visible=False),
-            gr.update(interactive=False),
+            mic_u,
+            file_u,
         )
+        return
+
+    # Désactive les deux boutons pendant la transcription + génération Gemma
+    mic_u, file_u = btn_states_disabled()
+    yield (
+        gr.update(),
+        gr.update(visible=False),
+        gr.update(visible=False),
+        mic_u,
+        file_u,
+    )
 
     try:
         progress(0.3, desc="🎙 Transcription audio...")
@@ -281,19 +317,22 @@ def transcribe_only(audio, progress=gr.Progress()):
 
         progress(1.0, desc="✅ Transcription terminée")
 
-        return (
+        yield (
             text,
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(interactive=False),
+            gr.update(interactive=False),
         )
 
     except Exception as e:
-        return (
+        mic_u, file_u = btn_states()
+        yield (
             f"Erreur Whisper : {str(e)}",
             gr.update(visible=False),
             gr.update(visible=False),
-            gr.update(interactive=True),
+            mic_u,
+            file_u,
         )
 
 
@@ -301,11 +340,13 @@ def transcribe_only(audio, progress=gr.Progress()):
 def respond_gemma(progress=gr.Progress()):
     transcription = current_exchange.get("transcription")
     if not transcription:
+        mic_u, file_u = btn_states()
         yield (
             "⚠️ Pas de transcription disponible.",
             gr.update(visible=False),
             gr.update(visible=False),
-            gr.update(interactive=True),
+            mic_u,
+            file_u,
         )
         return
 
@@ -322,6 +363,7 @@ def respond_gemma(progress=gr.Progress()):
                 gr.update(visible=False),
                 gr.update(visible=False),
                 gr.update(interactive=False),
+                gr.update(interactive=False),
             )
 
         if stop_generation["value"]:
@@ -330,19 +372,23 @@ def respond_gemma(progress=gr.Progress()):
         current_exchange["reponse"] = last_partial
         progress(1.0, desc="✅ Terminé")
 
+        mic_u, file_u = btn_states()
         yield (
             last_partial,
             gr.update(visible=True),
             gr.update(visible=True),
-            gr.update(interactive=True),
+            mic_u,
+            file_u,
         )
 
     except Exception as e:
+        mic_u, file_u = btn_states()
         yield (
             f"Erreur Gemma : {str(e)}",
             gr.update(visible=False),
             gr.update(visible=False),
-            gr.update(interactive=True),
+            mic_u,
+            file_u,
         )
 
 
@@ -350,9 +396,14 @@ def respond_gemma(progress=gr.Progress()):
 def regenerate_gemma_only(progress=gr.Progress()):
     transcription = current_exchange.get("transcription")
     if not transcription:
-        yield "", gr.update(visible=False)
+        mic_u, file_u = btn_states()
+        yield "", gr.update(visible=False), mic_u, file_u
         return
 
+    # Désactive les boutons de transcription pendant la régénération
+    mic_u, file_u = btn_states_disabled()
+    yield gr.update(), gr.update(visible=False), mic_u, file_u
+
     try:
         progress(0.3, desc="🔄 Régénération...")
 
@@ -361,17 +412,19 @@ def regenerate_gemma_only(progress=gr.Progress()):
             if stop_generation["value"]:
                 return
             last_partial = partial
-            yield partial, gr.update(visible=False)
+            yield partial, gr.update(visible=False), gr.update(interactive=False), gr.update(interactive=False)
 
         if stop_generation["value"]:
             return
 
         current_exchange["reponse"] = last_partial
         progress(1.0, desc="✅ Prêt")
-        yield last_partial, gr.update(visible=True)
+        mic_u, file_u = btn_states()
+        yield last_partial, gr.update(visible=True), mic_u, file_u
 
     except Exception as e:
-        yield f"Erreur : {str(e)}", gr.update(visible=False)
+        mic_u, file_u = btn_states()
+        yield f"Erreur : {str(e)}", gr.update(visible=False), mic_u, file_u
 
 
 # ── Gestion de l'enregistrement micro (attente de chargement) ─────
@@ -383,12 +436,15 @@ def on_audio_recorded(audio):
     réactive le bouton (uniquement si un audio est bien présent).
     """
     if audio is None:
+        audio_state["mic_ready"] = False
         # Audio supprimé -> bouton inactif
         return gr.update(interactive=False, value="🎤 Transcrire la voix")
 
+    audio_state["mic_ready"] = False
     # On affiche un état "chargement" pendant l'attente
     yield gr.update(interactive=False, value="⏳ Chargement de l'audio...")
     time.sleep(2)
+    audio_state["mic_ready"] = True
     yield gr.update(interactive=True, value="🎤 Transcrire la voix")
 
 
@@ -400,10 +456,13 @@ def on_file_uploaded(audio):
     de se charger côté navigateur.
     """
     if audio is None:
+        audio_state["file_ready"] = False
         return gr.update(interactive=False, value="📂 Transcrire le fichier")
 
+    audio_state["file_ready"] = False
     yield gr.update(interactive=False, value="⏳ Chargement du fichier...")
     time.sleep(2)
+    audio_state["file_ready"] = True
     yield gr.update(interactive=True, value="📂 Transcrire le fichier")
 
 
@@ -414,6 +473,8 @@ def reset_all():
     current_audio["filename"] = None
     current_exchange["transcription"] = None
     current_exchange["reponse"] = None
+    audio_state["mic_ready"] = False
+    audio_state["file_ready"] = False
     return (
         None, None, "", "",
         gr.update(visible=False), gr.update(visible=False),
@@ -651,27 +712,27 @@ with gr.Blocks(title="Whisper + Gemma · Darija", css=css, js=js) as demo:
     mic_btn.click(
         fn=transcribe_only,
         inputs=[mic_input],
-        outputs=[whisper_out, fb_whisper, fb_gemma, mic_btn],
+        outputs=[whisper_out, fb_whisper, fb_gemma, mic_btn, file_btn],
     ).then(
         fn=respond_gemma,
         inputs=[],
-        outputs=[gemma_out, fb_whisper, fb_gemma, mic_btn],
+        outputs=[gemma_out, fb_whisper, fb_gemma, mic_btn, file_btn],
     )
 
     file_btn.click(
         fn=transcribe_only,
         inputs=[file_input],
-        outputs=[whisper_out, fb_whisper, fb_gemma, file_btn],
+        outputs=[whisper_out, fb_whisper, fb_gemma, mic_btn, file_btn],
     ).then(
         fn=respond_gemma,
         inputs=[],
-        outputs=[gemma_out, fb_whisper, fb_gemma, file_btn],
+        outputs=[gemma_out, fb_whisper, fb_gemma, mic_btn, file_btn],
     )
 
     regenerate_btn.click(
         fn=regenerate_gemma_only,
         inputs=[],
-        outputs=[gemma_out, fb_gemma],
+        outputs=[gemma_out, fb_gemma, mic_btn, file_btn],
     )
 
     btn_whisper.click(fn=save_feedback_whisper, inputs=[whisper_out, note_whisper], outputs=[status_whisper])