Format plain output with in-syllable quantity markers

app.py

@@ -7,7 +7,7 @@ import torch
 import torch.nn.functional as F
 from transformers import AutoModelForTokenClassification, AutoTokenizer
 
-from grc_utils import lower_grc, normalize_word, heavy
+from grc_utils import lower_grc, normalize_word, heavy, vowel
 
 from syllabify import syllabify_joined
 from preprocess import process_word
@@ -122,6 +122,20 @@ def _syllable_chip(syllable: str, label_id: int) -> str:
         return f'<span class="chip clear">{escaped}</span>'
 
 
+def _mark_syllable_plain(syllable: str, label_id: int) -> str:
+    if label_id not in (1, 2):
+        return syllable
+
+    marker = "_" if label_id == 1 else "^"
+    chars = list(syllable)
+
+    for i in range(len(chars) - 1, -1, -1):
+        if vowel(chars[i]):
+            return "".join(chars[: i + 1]) + marker + "".join(chars[i + 1 :])
+
+    return syllable + marker
+
+
 def render_results(text: str, model_label: str):
     lines = [line.strip() for line in text.splitlines() if line.strip()]
     if not lines:
@@ -135,6 +149,7 @@ def render_results(text: str, model_label: str):
     for idx, line in enumerate(lines, start=1):
         aligned = classify_line(line, model_id)
         chips = "".join(_syllable_chip(syl, label) for syl, label in aligned)
+        plain_marked = [_mark_syllable_plain(syl, label) for syl, label in aligned]
 
         cards.append(
             f"""
@@ -147,9 +162,7 @@ def render_results(text: str, model_label: str):
         )
 
         export_lines.append(f"Line {idx}: {line}")
-        for syl, label in aligned:
-            tag = "long" if label == 1 else "short" if label == 2 else "clear"
-            export_lines.append(f"  - {syl}: {tag}")
+        export_lines.append("  " + " ".join(plain_marked) if plain_marked else "  (no syllables found)")
 
     html_result = (
         "<div class='legend'><span class='dot long'></span>Long"
@@ -421,7 +434,7 @@ body.dark-mode {
 """
 
 
-with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
+with gr.Blocks() as demo:
         gr.HTML("""
         <script>
         // Detect system dark mode preference and apply on load
@@ -499,4 +512,4 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
 
 
 if __name__ == "__main__":
-        demo.launch()
+    demo.launch(css=CSS)