Update app.py

app.py

@@ -83,6 +83,11 @@ def generate_audio_name():
 
 
 def process_audio(audio):
+    """Convertit un audio numpy en fichier WAV temporaire.
+    
+    FIX : Normalisation correcte — si le pic est > 1.0, on divise par le pic
+    (et non par 32768.0) pour éviter de rendre l'audio quasi-silencieux.
+    """
     if audio is None:
         return None
     if isinstance(audio, tuple):
@@ -90,15 +95,21 @@ def process_audio(audio):
         if data.ndim > 1:
             data = data.mean(axis=1)
         data = data.astype(np.float32)
-        if np.abs(data).max() > 1.0:
-            data = data / 32768.0
+        # FIX : vraie normalisation par le pic réel
+        peak = np.abs(data).max()
+        if peak > 1.0:
+            data = data / peak
         tmp = tempfile.NamedTemporaryFile(suffix=".wav", delete=False)
         sf.write(tmp.name, data, sample_rate)
         return tmp.name
     return audio
 
 
-def append_and_upload(feedback_file, entry, repo_id):
+def append_and_upload(entry, repo_id):
+    """Ajoute une entrée au feedback.jsonl local puis l'upload vers HF.
+    
+    FIX : suppression du paramètre feedback_file inutilisé.
+    """
     api = HfApi(token=HF_TOKEN)
     local = f"/tmp/{repo_id.replace('/', '')}.jsonl"
     try:
@@ -146,8 +157,14 @@ def build_prompt(transcription: str) -> str:
     )
 
 
-# ── Filtre caractères non-arabes ──────────────────────────────────
+# ── Filtre caractères ─────────────────────────────────────────────
 def filter_arabic_only(text: str) -> str:
+    """Garde les caractères arabes + ponctuation + chiffres + lettres latines.
+    
+    FIX : ajout des plages \u0030-\u0039 (chiffres 0-9) et
+    \u0041-\u005A / \u0061-\u007A (lettres latines) pour préserver
+    les numéros de téléphone, codes, et mots techniques (modem, fiber...).
+    """
     allowed = re.compile(
         r'[^\u0600-\u06FF\u0750-\u077F\u08A0-\u08FF'
         r'\uFB50-\uFDFF\uFE70-\uFEFF'
@@ -155,6 +172,8 @@ def filter_arabic_only(text: str) -> str:
         r'\u060C\u061B\u061F\u06D4'
         r'\u0021-\u002F\u003A-\u0040'
         r'\u005B-\u0060\u007B-\u007E'
+        r'\u0030-\u0039'              # Chiffres 0-9
+        r'\u0041-\u005A\u0061-\u007A' # Lettres latines A-Z a-z
         r'\n]'
     )
     return allowed.sub('', text).strip()
@@ -177,7 +196,8 @@ def extract_reply(text: str) -> str:
         next_parts = []
         for j in range(i + 1, len(lines)):
             nl = lines[j].strip().strip('"\'\u201c\u201d\u2018\u2019\u00ab\u00bb')
-            nl = re.sub(r'[*_`#*]+', '', nl).strip()
+            # FIX : regex nettoyée (le * était en double)
+            nl = re.sub(r'[*_`#]+', '', nl).strip()
             if nl:
                 next_parts.append(nl)
         if next_parts:
@@ -259,7 +279,6 @@ def generate_gemma_response(transcription: str, progress=None):
 
 # ── ÉTAPE 1 : Transcription ───────────────────────────────────────
 def transcribe_only(audio, progress=gr.Progress()):
-    # FIX : reset TOUJOURS en premier — évite d'utiliser une transcription précédente
     current_exchange["transcription"] = None
     current_exchange["reponse"] = None
     stop_generation["value"] = False
@@ -271,7 +290,7 @@ def transcribe_only(audio, progress=gr.Progress()):
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(interactive=True),
-            False,  # flag succès → False
+            False,
         )
 
     try:
@@ -290,7 +309,7 @@ def transcribe_only(audio, progress=gr.Progress()):
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(interactive=False),
-            True,   # flag succès → True
+            True,
         )
 
     except Exception as e:
@@ -299,13 +318,13 @@ def transcribe_only(audio, progress=gr.Progress()):
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(interactive=True),
-            False,  # flag succès → False
+            False,
         )
 
 
 # ── ÉTAPE 2 : Réponse Gemma ───────────────────────────────────────
 def respond_gemma(ok, progress=gr.Progress()):
-    # FIX : ne rien faire si la transcription a échoué ou était vide
+    """FIX : yield le texte PARTIEL à chaque itération pour un vrai streaming."""
     if not ok:
         yield (
             "",
@@ -331,8 +350,9 @@ def respond_gemma(ok, progress=gr.Progress()):
         last_partial = ""
         for partial in generate_gemma_response(transcription, progress=progress):
             last_partial = partial
+            # FIX : on yield le texte partiel (pas un message figé)
             yield (
-                "⏳ جاري التفكير...",
+                partial if partial else "⏳ جاري التفكير...",
                 gr.update(visible=False),
                 gr.update(visible=False),
                 gr.update(interactive=False),
@@ -359,9 +379,10 @@ def respond_gemma(ok, progress=gr.Progress()):
 
 # ── Régénération Gemma seule ──────────────────────────────────────
 def regenerate_gemma_only(progress=gr.Progress()):
+    """FIX : yield le streaming ET gère les boutons pour éviter les conflits."""
     transcription = current_exchange.get("transcription")
     if not transcription:
-        yield "", gr.update(visible=False)
+        yield "", gr.update(visible=False), gr.update(interactive=True), gr.update(interactive=True)
         return
 
     try:
@@ -370,14 +391,30 @@ def regenerate_gemma_only(progress=gr.Progress()):
         last_partial = ""
         for partial in generate_gemma_response(transcription, progress=progress):
             last_partial = partial
-            yield "⏳ جاري التفكير...", gr.update(visible=False)
+            # FIX : yield le texte partiel + boutons désactivés
+            yield (
+                partial if partial else "⏳ جاري التفكير...",
+                gr.update(visible=False),
+                gr.update(interactive=False),
+                gr.update(interactive=False),
+            )
 
         current_exchange["reponse"] = last_partial
         progress(1.0, desc="✅ Prêt")
-        yield last_partial, gr.update(visible=True)
+        yield (
+            last_partial,
+            gr.update(visible=True),
+            gr.update(interactive=True),
+            gr.update(interactive=True),
+        )
 
     except Exception as e:
-        yield f"Erreur : {str(e)}", gr.update(visible=False)
+        yield (
+            f"Erreur : {str(e)}",
+            gr.update(visible=False),
+            gr.update(interactive=True),
+            gr.update(interactive=True),
+        )
 
 
 # ── Reset ─────────────────────────────────────────────────────────
@@ -392,7 +429,7 @@ def reset_all():
         gr.update(visible=False), gr.update(visible=False),
         None, None, "", "",
         gr.update(interactive=True), gr.update(interactive=True),
-        False,  # reset du state flag
+        False,
     )
 
 
@@ -410,7 +447,8 @@ def save_feedback_whisper(transcription, note):
             "transcription": transcription,
             "note": int(note),
         }
-        append_and_upload("feedback.jsonl", entry, FEEDBACK_REPO)
+        # FIX : appel sans le paramètre feedback_file fantôme
+        append_and_upload(entry, FEEDBACK_REPO)
         if current_audio["path"] and os.path.exists(current_audio["path"]):
             api.upload_file(
                 path_or_fileobj=current_audio["path"],
@@ -436,7 +474,8 @@ def save_feedback_gemma(note):
             "reponse": r,
             "note": int(note),
         }
-        append_and_upload("feedback.jsonl", entry, FEEDBACK2_REPO)
+        # FIX : appel sans le paramètre feedback_file fantôme
+        append_and_upload(entry, FEEDBACK2_REPO)
         return f"✅ Feedback enregistré — {'⭐' * int(note)}"
     except Exception as e:
         return f"❌ Erreur : {str(e)}"
@@ -456,7 +495,6 @@ body, .gradio-container {
     padding: 2rem 1.5rem !important;
 }
 
-/* ── Header ── */
 #app-header {
     text-align: center;
     margin-bottom: 2rem;
@@ -466,7 +504,6 @@ body, .gradio-container {
 #app-header h1 { font-size: 2rem; font-weight: 600; color: #f0f0f5; margin: 0 0 0.4rem; }
 #app-header p  { font-size: 0.9rem; color: #6b6b80; margin: 0; }
 
-/* ── Section labels ── */
 .section-label {
     font-size: 0.72rem;
     font-weight: 600;
@@ -476,7 +513,6 @@ body, .gradio-container {
     margin-bottom: 0.6rem;
 }
 
-/* ── Bouton Transcrire voix (mic) ── */
 #mic-btn {
     background: linear-gradient(135deg, #22c97a 0%, #16a85f 100%) !important;
     color: #fff !important;
@@ -500,7 +536,6 @@ body, .gradio-container {
     box-shadow: none !important;
 }
 
-/* ── Bouton Transcrire fichier ── */
 #file-btn {
     background: linear-gradient(135deg, #5a5aff 0%, #4040d9 100%) !important;
     color: #fff !important;
@@ -524,7 +559,6 @@ body, .gradio-container {
     box-shadow: none !important;
 }
 
-/* ── Bouton Reset ── */
 button.danger {
     background: #dc2626 !important;
     color: #fff !important;
@@ -535,7 +569,6 @@ button.danger {
 }
 button.danger:hover { background: #b91c1c !important; }
 
-/* ── Bouton Régénérer ── */
 button.regenerate {
     background: #7c3aed !important;
     color: #fff !important;
@@ -546,7 +579,6 @@ button.regenerate {
 }
 button.regenerate:hover { background: #6d28d9 !important; }
 
-/* ── Textboxes ── */
 #transcription-box textarea {
     background: #0f0f13 !important;
     border: 1.5px solid #1e1e2a !important;
@@ -580,7 +612,6 @@ button.regenerate:hover { background: #6d28d9 !important; }
     box-shadow: 0 0 0 3px rgba(168,85,247,0.12) !important;
 }
 
-/* ── Panneau feedback ── */
 .feedback-panel {
     background: #13131c;
     border: 1px solid #222230;
@@ -589,7 +620,6 @@ button.regenerate:hover { background: #6d28d9 !important; }
     margin-top: 0.8rem;
 }
 
-/* ── Stars radio (feedback) ── */
 .feedback-panel .wrap {
     gap: 6px !important;
 }
@@ -624,7 +654,6 @@ button.regenerate:hover { background: #6d28d9 !important; }
     color: #f59e0b !important;
 }
 
-/* ── Bouton Envoyer feedback ── */
 .feedback-panel button {
     background: linear-gradient(135deg, #f59e0b 0%, #d97706 100%) !important;
     color: #1a1000 !important;
@@ -649,10 +678,8 @@ button.regenerate:hover { background: #6d28d9 !important; }
     box-shadow: 0 2px 8px rgba(245, 158, 11, 0.3) !important;
 }
 
-/* ── Status feedback ── */
 .feedback-panel .prose { color: #6b6b80 !important; font-size: 0.85rem !important; }
 
-/* ── Mobile fixes ── */
 @media (max-width: 768px) {
     .gr-audio canvas,
     [data-testid="audio"] canvas,
@@ -694,7 +721,6 @@ function() {
 # ── Interface ─────────────────────────────────────────────────────
 with gr.Blocks(title="Whisper + Gemma · Darija", css=css, js=js) as demo:
 
-    # FIX : state qui transmet le résultat de transcription à respond_gemma
     transcription_ok = gr.State(False)
 
     gr.HTML(
@@ -760,7 +786,7 @@ with gr.Blocks(title="Whisper + Gemma · Darija", css=css, js=js) as demo:
         outputs=[whisper_out, fb_whisper, fb_gemma, mic_btn, transcription_ok],
     ).then(
         fn=respond_gemma,
-        inputs=[transcription_ok],   # FIX : conditionne Gemma au succès Whisper
+        inputs=[transcription_ok],
         outputs=[gemma_out, fb_whisper, fb_gemma, mic_btn],
     )
 
@@ -770,14 +796,15 @@ with gr.Blocks(title="Whisper + Gemma · Darija", css=css, js=js) as demo:
         outputs=[whisper_out, fb_whisper, fb_gemma, file_btn, transcription_ok],
     ).then(
         fn=respond_gemma,
-        inputs=[transcription_ok],   # FIX : conditionne Gemma au succès Whisper
+        inputs=[transcription_ok],
         outputs=[gemma_out, fb_whisper, fb_gemma, file_btn],
     )
 
+    # FIX : la régénération gère maintenant aussi mic_btn et 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(
@@ -798,7 +825,7 @@ with gr.Blocks(title="Whisper + Gemma · Darija", css=css, js=js) as demo:
             fb_whisper, fb_gemma,
             note_whisper, note_gemma, status_whisper, status_gemma,
             mic_btn, file_btn,
-            transcription_ok,   # FIX : reset du flag aussi
+            transcription_ok,
         ],
     )