save

listen_transcribe/highlighter.py

@@ -7,29 +7,29 @@ def highlight_fuzzy_diff(user_text, original_text, threshold=85):
 
     highlighted = []
     user_index = 0
+    correct_count = 0
 
     for orig_word in original_words:
         if user_index < len(user_words):
             user_word = user_words[user_index]
             score = fuzz.ratio(orig_word, user_word)
             if score >= threshold:
-                # ✅ Correct word in correct place
+                correct_count += 1
                 highlighted.append(
                     f"<span style='background-color:#e6ffe6; color:#006600; padding:4px 10px; "
                     f"margin:4px; border-radius:999px; font-weight:500; display:inline-block;'>{orig_word}</span>"
                 )
                 user_index += 1
             else:
-                # ❌ Word mismatch, consider as missing
                 highlighted.append(
                     f"<span style='background-color:#ffe6e6; color:#990000; padding:4px 10px; "
                     f"margin:4px; border-radius:999px; font-weight:500; display:inline-block;'>...</span>"
                 )
         else:
-            # ❌ No more user input, word is missing
             highlighted.append(
                 f"<span style='background-color:#ffe6e6; color:#990000; padding:4px 10px; "
                 f"margin:4px; border-radius:999px; font-weight:500; display:inline-block;'>...</span>"
             )
 
-    return " ".join(highlighted)
+    score_percent = round((correct_count / len(original_words)) * 100) if original_words else 0
+    return " ".join(highlighted), score_percent

listen_transcribe/listen_transcribe_ui.py

@@ -1,7 +1,6 @@
 import os
 
 import gradio as gr
-from rapidfuzz import fuzz
 
 from listen_transcribe.audio_files import get_audio_structure
 from listen_transcribe.dialogue import load_dialogue_lines
@@ -96,12 +95,10 @@ def transcribe_ui():
         def on_check(*user_inputs):
             feedback = []
             highlights = []
-            scores = []
 
             for user_input, (line_index, _, _) in zip(user_inputs, inputs):
                 expected = lines[line_index][1]
-                score = round(fuzz.ratio(user_input.strip(), expected.strip()))
-                scores.append(score)
+                diff_html, score = highlight_fuzzy_diff(user_input, expected)
 
                 if score > 90:
                     emoji, msg, color = "✅", "Excellent!", "#2ecc71"
@@ -111,8 +108,8 @@ def transcribe_ui():
                     emoji, msg, color = "❌", "Try again", "#e74c3c"
 
                 fb_html = f"<div style='color:{color}; font-size: 14px;'>{emoji} <b>{score}%</b> — {msg}</div>"
+                highlights.append(diff_html)
                 feedback.append(fb_html)
-                highlights.append(highlight_fuzzy_diff(user_input, expected))
 
             return highlights + feedback