save

app.py

@@ -1,10 +1,11 @@
 import os
 import gradio as gr
+from rapidfuzz import fuzz
 
 from audio_files import get_audio_structure
-from check import check_transcription, load_dialogue_lines
+from check import load_dialogue_lines, highlight_fuzzy_diff  # додано highlight_fuzzy_diff
 
-# === Audio structure
+# === Audio structure ===
 audio_structure = get_audio_structure()
 week_options = list(audio_structure.keys())
 default_week = week_options[0]
@@ -13,55 +14,68 @@ default_file = audio_structure[default_week][0] if audio_structure[default_week]
 with gr.Blocks(theme="soft") as demo:
     gr.Markdown("## 🎧 Modals & Conditionals Bootcamp")
 
-    # States
     current_index = gr.State(0)
     dialogue_state = gr.State([])
 
-    # Week + audio
     with gr.Row():
         week_dropdown = gr.Dropdown(label="Select a week", choices=week_options, value=default_week, scale=1)
         audio_player = gr.Audio(label="Audio preview", type="filepath", value=default_file, scale=3)
 
     gr.Markdown("### ✍️ Your Transcription")
 
-    # === Dynamic dialogue render ===
     @gr.render(inputs=[week_dropdown, current_index, dialogue_state])
     def render_transcription_inputs(week, index, inputs):
         audio_path = audio_structure[week][0]
         transcript_path = audio_path.replace(".mp3", ".txt").replace("audio", "transcripts")
         lines = load_dialogue_lines(transcript_path) if os.path.exists(transcript_path) else []
 
-        # Add next speaker
         if index < len(lines):
             next_speaker, _ = lines[index]
             inputs = [(next_speaker, "")] + inputs
             index += 1
 
-        # === UI Layout ===
         button_label = "➕ Add next speaker" if index < len(lines) else "✅ Check"
         action_btn = gr.Button(button_label)
-        textboxes = [gr.Textbox(label=speaker, value=value, lines=1, key=f"{week}-{i}") for i, (speaker, value) in enumerate(inputs)]
 
-        with gr.Accordion("📖 Show Original Text", open=False):
-            original_output = gr.HTML()
+        textboxes = []
+        feedback_outputs = []
+        diff_outputs = []
 
-        diff_output = gr.HTML(label="Comparison (highlighted)")
-        feedback_output = gr.HTML(label="Feedback and score")
+        for i, (speaker, value) in enumerate(inputs):
+            with gr.Column():
+                tb = gr.Textbox(label=speaker, value=value, lines=1, key=f"{week}-{i}")
+                diff = gr.HTML()
+                fb = gr.HTML()
+                textboxes.append(tb)
+                diff_outputs.append(diff)
+                feedback_outputs.append(fb)
 
-        # === Logic ===
-        def on_action_click(*user_inputs):
-            if index < len(lines):
-                return index, inputs
-            else:
-                user_text = "\n".join(user_inputs)
-                return check_transcription(audio_path, user_text)
+        def on_add():
+            return index, inputs
+
+        def per_line_check(*user_inputs):
+            feedback = []
+            highlights = []
+            for user_input, (_, expected) in zip(user_inputs, reversed(lines[:len(user_inputs)])):
+                score = fuzz.ratio(user_input.strip(), expected.strip())
+                if score > 90:
+                    emoji, msg = "✅", f"<b>{score}%</b> Excellent!"
+                elif score > 70:
+                    emoji, msg = "⚠️", f"<b>{score}%</b> Some mistakes"
+                else:
+                    emoji, msg = "❌", f"<b>{score}%</b> Try again"
+                feedback.append(f"<div style='margin-top: 4px;'>{emoji} {msg}</div>")
+                highlights.append(highlight_fuzzy_diff(user_input, expected))
+            return highlights + feedback  # concatenate results
 
-        # === Event binding (depending on mode)
         if index < len(lines):
-            action_btn.click(fn=lambda: (index, inputs), outputs=[current_index, dialogue_state])
+            action_btn.click(fn=on_add, outputs=[current_index, dialogue_state])
         else:
-            action_btn.click(fn=on_action_click, inputs=[tb for tb in textboxes],
-                             outputs=[audio_player, original_output, diff_output, feedback_output])
+            action_btn.click(
+                fn=per_line_check,
+                inputs=textboxes,
+                outputs=diff_outputs + feedback_outputs
+            )
 
     demo.load(lambda: None)
 

audio_files.py

@@ -4,8 +4,7 @@ import os
 def get_audio_structure(audio_dir="audio"):
     structure = {}
     for file in sorted(os.listdir(audio_dir)):
-        if file.lower().endswith(".mp3") and file.lower().startswith("week"):
-            label = file.replace(".mp3", "").replace("_", " ").title()  # e.g., "week1" -> "Week1"
+        if file.endswith(".mp3"):
+            label = os.path.splitext(file)[0].replace("_", " ").title()
             structure[label] = [os.path.join(audio_dir, file)]
     return structure
-

check.py

@@ -60,12 +60,10 @@ def check_transcription(audio_path, user_text):
 
 
 def load_dialogue_lines(path):
-    with open(path, "r") as f:
-        lines = f.readlines()
-
-    dialogue = []
-    for line in lines:
-        if ":" in line:
-            speaker, text = line.split(":", 1)
-            dialogue.append((speaker.strip(), text.strip()))
-    return dialogue
+    with open(path, "r", encoding="utf-8") as f:
+        lines = []
+        for line in f:
+            if ":" in line:
+                speaker, text = line.strip().split(":", 1)
+                lines.append((speaker.strip(), text.strip()))
+        return lines