Update app.py

app.py

@@ -1,9 +1,12 @@
 """
-Estella Explainer — HuggingFace Spaces Demo
+Estella Explainer — HuggingFace Spaces / Lightning AI Demo
 A math reading coach for grades 3–8, based on Usable Math (usablemath.org)
-Model: sgattup/EstellaExplainerLLM (Gemma 4 E4B fine-tuned)
 
-Deploy to a ZeroGPU Space for free GPU inference.
+Developed by Sai Gattupalli, PhD
+College of Education, University of Massachusetts Amherst
+Principal Scientist, Society and AI Research Group
+
+Model: sgattup/EstellaExplainerLLM (Gemma 4 E4B fine-tuned)
 """
 
 import gradio as gr
@@ -35,7 +38,6 @@ EXAMPLES = [
     "The perimeter of a square is 36 inches. What is the length of one side?",
 ]
 
-# Load model once at startup
 print(f"Loading {MODEL_ID}...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 model = AutoModelForCausalLM.from_pretrained(
@@ -44,15 +46,16 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 model.eval()
-print("Model ready.")
+print("Estella is ready.")
 
 
-def ask_estella(problem: str) -> str:
-    if not problem.strip():
-        return "Please type a math word problem above."
+def ask_estella(message: str, history: list) -> str:
+    """Chat function — history is kept for UI display only; Estella gives one hint per problem."""
+    if not message.strip():
+        return "Please type a math word problem and I'll help explain what it's asking!"
 
     messages = [
-        {"role": "user", "content": SYSTEM_PROMPT + f"\n\n### Math Problem:\n{problem.strip()}\n\n### Estella's Hint:"}
+        {"role": "user", "content": SYSTEM_PROMPT + f"\n\n### Math Problem:\n{message.strip()}\n\n### Estella's Hint:"}
     ]
     text = tokenizer.apply_chat_template(
         messages,
@@ -76,50 +79,18 @@ def ask_estella(problem: str) -> str:
     return hint
 
 
-# Gradio UI
-with gr.Blocks(title="Estella Explainer — Math Reading Coach") as demo:
-    gr.Markdown("""
-# 📚 Estella Explainer
-### A Math Reading Coach for Grades 3–8
-
-*"My job is to explain math questions clearly so you know what you are supposed to do to solve the problem."*
-
-Estella helps students **understand** what a math problem is asking — without solving it.
-Based on [Usable Math](https://usablemath.org), an open educational resource from UMass Amherst.
-
----
-    """)
-
-    with gr.Row():
-        with gr.Column(scale=2):
-            problem_input = gr.Textbox(
-                label="Math Word Problem",
-                placeholder="Type or paste a math problem here...",
-                lines=4
-            )
-            submit_btn = gr.Button("Ask Estella", variant="primary")
-
-        with gr.Column(scale=2):
-            hint_output = gr.Textbox(
-                label="Estella's Hint",
-                lines=4,
-                interactive=False
-            )
-
-    gr.Examples(
-        examples=EXAMPLES,
-        inputs=problem_input,
-        label="Try these examples"
-    )
+# Gradio ChatInterface
+demo = gr.ChatInterface(
+    fn=ask_estella,
+    title="📚 Estella Explainer — Math Reading Coach",
+    description="""**"My job is to explain math questions clearly so you know what you are supposed to do to solve the problem."**
 
-    gr.Markdown("""
----
-**Model:** [sgattup/EstellaExplainerLLM](https://huggingface.co/sgattup/EstellaExplainerLLM) · Fine-tuned Gemma 4 E4B via QLoRA
-**Project:** [Usable Math](https://usablemath.org) · UMass Amherst College of Education
-**Note:** Estella explains problems — she does not solve them or show answers.
-    """)
+Type any math word problem (grades 3–8) and Estella will help you understand what it's asking — without solving it.
 
-    submit_btn.click(fn=ask_estella, inputs=problem_input, outputs=hint_output)
-    problem_input.submit(fn=ask_estella, inputs=problem_input, outputs=hint_output)
+*Developed by [Sai Gattupalli, PhD](https://huggingface.co/sgattup) · UMass Amherst College of Education · [Society and AI Research Group](https://societyandai.org) · Based on [Usable Math](https://usablemath.org)*""",
+    examples=EXAMPLES,
+    cache_examples=False,
+    type="messages",
+)
 
 demo.launch()