Update app.py

app.py

@@ -1,10 +1,15 @@
+# app.py
+# CLI-like Spelling Bee Tutor as a simple Gradio UI (no LLMs, all offline)
+
 from pathlib import Path
-import pandas as pd
+import os
 import csv
+import pandas as pd
+import gradio as gr
 
-# make path robust in Spaces
-WORDS_PATH = Path(__file__).parent / "words.csv"
+# ---------- Data loading (robust to messy CSVs) ----------
 
+WORDS_PATH = Path(__file__).parent / "words.csv"
 EXPECTED_COLS = ["word", "difficulty", "definition", "origin", "sentence"]
 
 def _load_words(path: Path = WORDS_PATH) -> pd.DataFrame:
@@ -28,7 +33,6 @@ def _load_words(path: Path = WORDS_PATH) -> pd.DataFrame:
                 dtype=str,
                 keep_default_na=False
             )
-            # If it parsed to 5+ columns, normalize below
             df = df_try
         except Exception:
             # fall back to manual glue if pandas fails
@@ -54,13 +58,11 @@ def _load_words(path: Path = WORDS_PATH) -> pd.DataFrame:
             df = pd.DataFrame(rows, columns=EXPECTED_COLS)
 
     # --- Normalize columns ---
-    cols = [c.strip().lower() for c in df.columns]
     mapper = {c: c.strip().lower() for c in df.columns}
     df = df.rename(columns=mapper)
 
     # If there are more than 5 columns, rebuild to our schema
     if df.shape[1] >= 5:
-        # take first four known fields if present, then glue rest into sentence
         def pick(colname, default=""):
             return df[colname] if colname in df.columns else default
         base_df = pd.DataFrame({
@@ -106,3 +108,175 @@ def _load_words(path: Path = WORDS_PATH) -> pd.DataFrame:
     # sort hardest → easiest
     df = df.sort_values("difficulty_score", ascending=False).reset_index(drop=True)
     return df
+
+DF = _load_words()
+
+# ---------- Quiz logic ----------
+
+def _summary(state):
+    if not state:
+        return "No round active."
+    n = state["n"]
+    score = state["score"]
+    return f"Round complete. Score this round: {score}/{n}."
+
+def start_quiz(n_words, state):
+    df = DF
+    if df.empty:
+        return (state,
+                "words.csv not found or empty.",
+                gr.update(value="", interactive=False),
+                "Please add a words.csv with at least one 'word' column.",
+                "Score: 0/0",
+                gr.update(value="", interactive=False))
+
+    try:
+        n = int(n_words or 5)
+    except Exception:
+        n = 5
+    n = max(1, min(n, len(df)))  # clip to available size
+
+    block = df.head(n).reset_index(drop=True)
+    s = {
+        "i": 0,
+        "n": n,
+        "score": 0,
+        "words": block["word"].astype(str).tolist(),
+        "defs": block["definition"].astype(str).tolist(),
+        "orig": block["origin"].astype(str).tolist(),
+        "sent": block["sentence"].astype(str).tolist(),
+    }
+
+    current = s["words"][0]
+    hist = f"Okay! We'll do {n} words this round, hardest → easiest.\nSpell this word: {current}"
+    status = "Type your spelling attempt, or click definition/origin/sentence."
+    return s, hist, gr.update(value=current, interactive=False), status, f"Score: 0/{n}", gr.update(value="", interactive=True)
+
+def _check_state(state):
+    return bool(state and "words" in state and state["i"] < state["n"])
+
+def check_attempt(state, attempt, history, current_word, score_md):
+    if not _check_state(state):
+        return state, history, current_word, "No round active. Click Start.", score_md, gr.update(value="")
+    attempt = (attempt or "").strip()
+    if not attempt:
+        return state, history, current_word, "Type your attempt first.", score_md, gr.update(value="")
+
+    target = state["words"][state["i"]]
+    if attempt.lower() == target.lower():
+        state["score"] += 1
+        msg = "✅ Correct!"
+    else:
+        msg = "❌ Not quite. Try again or ask for definition/origin/sentence."
+
+    history = f"{history}\nYou: {attempt}\nTutor: {msg}"
+    score_md = f"Score: {state['score']}/{state['n']}"
+    return state, history, current_word, msg, score_md, gr.update(value="")
+
+def _safe_text(val, fallback="Not available."):
+    v = (val or "").strip()
+    return v if v else fallback
+
+def show_def(state, history, current_word, score_md):
+    if not _check_state(state):
+        return state, history, current_word, "No round active. Click Start.", score_md
+    i = state["i"]
+    word = state["words"][i]
+    text = _safe_text(state["defs"][i])
+    history = f"{history}\nTutor (definition of {word}): {text}"
+    return state, history, current_word, f"Definition of {word}: {text}", score_md
+
+def show_origin(state, history, current_word, score_md):
+    if not _check_state(state):
+        return state, history, current_word, "No round active. Click Start.", score_md
+    i = state["i"]
+    word = state["words"][i]
+    text = _safe_text(state["orig"][i])
+    history = f"{history}\nTutor (origin of {word}): {text}"
+    return state, history, current_word, f"Origin of {word}: {text}", score_md
+
+def show_sentence(state, history, current_word, score_md):
+    if not _check_state(state):
+        return state, history, current_word, "No round active. Click Start.", score_md
+    i = state["i"]
+    word = state["words"][i]
+    text = _safe_text(state["sent"][i])
+    history = f"{history}\nTutor (sentence with {word}): {text}"
+    return state, history, current_word, f"Sentence: {text}", score_md
+
+def next_word(state, history, current_word, score_md):
+    if not _check_state(state):
+        return state, history, current_word, "No round active. Click Start.", score_md
+    state["i"] += 1
+    if state["i"] >= state["n"]:
+        summary = _summary(state)
+        history = f"{history}\n{summary}"
+        return {}, history, gr.update(value="", interactive=False), summary, f"Score: {state['score']}/{state['n']}"
+    w = state["words"][state["i"]]
+    history = f"{history}\nNext word: {w}"
+    status = "Type your spelling attempt, or click definition/origin/sentence."
+    return state, history, gr.update(value=w, interactive=False), status, f"Score: {state['score']}/{state['n']}"
+
+def stop_round(state, history, current_word, score_md):
+    if not state:
+        return {}, history, current_word, "Stopped.", score_md
+    summary = _summary(state)
+    history = f"{history}\n{summary}"
+    return {}, history, gr.update(value="", interactive=False), "Stopped.", f"Score: {state['score']}/{state['n']}"
+
+# ---------- UI ----------
+
+with gr.Blocks() as demo:
+    gr.Markdown("# NeMo Guardrails Demo (starter UI)\n**CLI-style spelling quiz** — works offline from `words.csv`.")
+
+    with gr.Row():
+        n_words = gr.Number(value=5, precision=0, label="Words this round")
+        start = gr.Button("Start quiz")
+
+    current_word = gr.Textbox(label="Spell this word", interactive=False)
+    attempt = gr.Textbox(label="Your attempt")
+    with gr.Row():
+        check = gr.Button("Check")
+        bdef = gr.Button("definition")
+        borg = gr.Button("origin")
+        bsent = gr.Button("sentence")
+        bnext = gr.Button("next")
+        bstop = gr.Button("stop")
+
+    status = gr.Markdown("")
+    score_md = gr.Markdown("Score: 0/0")
+    history = gr.Textbox(label="History", lines=14)
+
+    state = gr.State({})
+
+    # wire events
+    start.click(start_quiz, [n_words, state],
+                [state, history, current_word, status, score_md, attempt])
+
+    check.click(check_attempt, [state, attempt, history, current_word, score_md],
+                [state, history, current_word, status, score_md, attempt])
+
+    bdef.click(show_def, [state, history, current_word, score_md],
+               [state, history, current_word, status, score_md], queue=False)
+    borg.click(show_origin, [state, history, current_word, score_md],
+               [state, history, current_word, status, score_md], queue=False)
+    bsent.click(show_sentence, [state, history, current_word, score_md],
+                [state, history, current_word, status, score_md], queue=False)
+    bnext.click(next_word, [state, history, current_word, score_md],
+                [state, history, current_word, status, score_md], queue=False)
+    bstop.click(stop_round, [state, history, current_word, score_md],
+                [state, history, current_word, status, score_md], queue=False)
+
+# ---------- Launch (blocking on Spaces) ----------
+
+if __name__ == "__main__":
+    try:
+        demo.queue(concurrency_count=8, max_size=32)
+    except Exception:
+        pass
+
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=int(os.getenv("PORT", "7860")),
+        show_error=True,
+    )