Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
+# ── Load Model ─────────────────────────────────────────────
 print("Loading model...")
 MODEL_ID = "newtechdevng/qwen-math-tutor"
 
@@ -14,13 +15,13 @@ model = AutoModelForCausalLM.from_pretrained(
 model.eval()
 print("✅ Model loaded!")
 
+# ── Inference Function ─────────────────────────────────────
 def solve(question, history):
     messages = [
         {"role": "system", "content": "You are a helpful math tutor. Solve problems step by step, showing all working clearly."},
     ]
-    for h in history:
-        messages.append({"role": "user",      "content": h[0]})
-        messages.append({"role": "assistant", "content": h[1]})
+    for msg in history:
+        messages.append({"role": msg["role"], "content": msg["content"]})
     messages.append({"role": "user", "content": question})
 
     text = tokenizer.apply_chat_template(
@@ -41,38 +42,62 @@ def solve(question, history):
         skip_special_tokens=True
     )
 
+# ── Response Handler ───────────────────────────────────────
+def respond(question, history):
+    if not question.strip():
+        return "", history
+    answer = solve(question, history)
+    history.append({"role": "user",      "content": question})
+    history.append({"role": "assistant", "content": answer})
+    return "", history
+
+# ── UI ─────────────────────────────────────────────────────
 with gr.Blocks(title="Math Tutor AI") as demo:
     gr.Markdown("""
     # 🧮 Math Tutor AI
     ### Powered by Qwen2.5-Math — Fine-tuned for NCERT & competitive math
+    ---
     """)
 
-    chatbot = gr.Chatbot(height=400, label="Math Tutor")
+    chatbot = gr.Chatbot(
+        height = 450,
+        label  = "Math Tutor",
+        type   = "messages",
+        show_copy_button = True,
+    )
 
     with gr.Row():
         question = gr.Textbox(
             placeholder = "Type your math question here...",
             label       = "Your Question",
-            scale       = 4
+            lines       = 1,
+            scale       = 4,
         )
         submit = gr.Button("Solve →", variant="primary", scale=1)
 
+    clear = gr.Button("🗑️ Clear Chat", variant="secondary")
+
     gr.Examples(
+        label   = "Try these examples:",
         examples=[
             "Solve: 2x² - 7x + 3 = 0",
             "Find the area of a triangle with base 12 cm and height 8 cm.",
             "If sin θ = 3/5, find cos θ and tan θ.",
             "A train travels 360 km in 4 hours. Find its speed in m/s.",
+            "Find the compound interest on ₹5000 at 10% per annum for 2 years.",
+            "Prove that √2 is irrational.",
         ],
-        inputs=question
+        inputs  = question,
     )
 
-    def respond(q, history):
-        answer = solve(q, history)
-        history.append((q, answer))
-        return "", history
+    gr.Markdown("""
+    ---
+    *Model: Qwen2.5-Math-1.5B fine-tuned on 10K math problems*
+    """)
 
+    # ── Event Handlers ──────────────────────────────────────
     submit.click(respond, [question, chatbot], [question, chatbot])
     question.submit(respond, [question, chatbot], [question, chatbot])
+    clear.click(lambda: ([], ""), outputs=[chatbot, question])
 
 demo.launch(theme=gr.themes.Soft())