Upload app2.py

app2.py

@@ -0,0 +1,511 @@
+# ── Imports ── (ORDER MATTERS: re must come after sympy wildcard)
+import streamlit as st
+import sympy as sp
+import anthropic
+from sympy import (
+    symbols, diff, integrate, limit, solve, simplify, expand, factor,
+    latex, sympify, oo, sin, cos, tan, exp, log, sqrt, pi, E,
+    Matrix, Symbol, Function, dsolve, Eq,
+    series, Sum, Product, Rational, Integer, Float, Abs
+)
+from sympy.parsing.sympy_parser import (
+    parse_expr,
+    standard_transformations,
+    implicit_multiplication_application
+)
+import re  # MUST be last — sympy wildcard would overwrite it otherwise
+
+# ─── Page Config ───────────────────────────────────────────────────────────────
+st.set_page_config(
+    page_title="SAAD AI ACADEMY",
+    page_icon="📐",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# ─── CSS Styling ───────────────────────────────────────────────────────────────
+st.markdown("""
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Rajdhani:wght@400;600;700&family=JetBrains+Mono:wght@400;500&family=Nunito:wght@300;400;600&display=swap');
+
+:root {
+    --bg:        #060b14;
+    --surface:   #0d1526;
+    --card:      #111d35;
+    --border:    #1e3058;
+    --accent:    #38bdf8;
+    --accent2:   #818cf8;
+    --gold:      #f59e0b;
+    --green:     #34d399;
+    --text:      #e2e8f0;
+    --muted:     #64748b;
+}
+
+html, body, [class*="css"] {
+    background-color: var(--bg);
+    color: var(--text);
+    font-family: 'Nunito', sans-serif;
+}
+
+#MainMenu, footer, header { visibility: hidden; }
+.block-container { padding: 1.5rem 2rem 2rem 2rem; max-width: 1100px; }
+
+.hero {
+    background: linear-gradient(135deg, #0d1526 0%, #0f1f40 50%, #091428 100%);
+    border: 1px solid var(--border);
+    border-radius: 16px;
+    padding: 2rem 2.5rem;
+    margin-bottom: 1.5rem;
+    position: relative;
+    overflow: hidden;
+}
+.hero::before {
+    content: "";
+    position: absolute;
+    top: -60px; right: -60px;
+    width: 220px; height: 220px;
+    background: radial-gradient(circle, rgba(56,189,248,0.12) 0%, transparent 70%);
+    border-radius: 50%;
+}
+.hero-title {
+    font-family: 'Rajdhani', sans-serif;
+    font-size: 2.4rem;
+    font-weight: 700;
+    background: linear-gradient(90deg, var(--accent), var(--accent2));
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    margin: 0 0 0.3rem 0;
+    letter-spacing: 2px;
+}
+.hero-sub {
+    color: var(--muted);
+    font-size: 0.95rem;
+    letter-spacing: 1px;
+    text-transform: uppercase;
+}
+.badge-row { display: flex; gap: 0.5rem; flex-wrap: wrap; margin-top: 1rem; }
+.badge {
+    background: rgba(56,189,248,0.08);
+    border: 1px solid rgba(56,189,248,0.25);
+    color: var(--accent);
+    padding: 3px 10px;
+    border-radius: 20px;
+    font-size: 0.72rem;
+    font-family: 'JetBrains Mono', monospace;
+    letter-spacing: 0.5px;
+}
+
+.chat-area {
+    background: var(--surface);
+    border: 1px solid var(--border);
+    border-radius: 14px;
+    padding: 1.2rem 1.5rem;
+    min-height: 420px;
+    max-height: 560px;
+    overflow-y: auto;
+    margin-bottom: 1rem;
+}
+
+.msg-user {
+    display: flex;
+    justify-content: flex-end;
+    margin: 0.8rem 0;
+}
+.msg-user .bubble {
+    background: linear-gradient(135deg, #1e3a5f, #1a2f50);
+    border: 1px solid rgba(56,189,248,0.3);
+    color: var(--text);
+    padding: 0.75rem 1.1rem;
+    border-radius: 14px 4px 14px 14px;
+    max-width: 75%;
+    font-size: 0.92rem;
+    line-height: 1.6;
+}
+.msg-ai {
+    display: flex;
+    justify-content: flex-start;
+    margin: 0.8rem 0;
+    gap: 0.6rem;
+    align-items: flex-start;
+}
+.ai-avatar {
+    width: 34px; height: 34px;
+    background: linear-gradient(135deg, var(--accent), var(--accent2));
+    border-radius: 50%;
+    display: flex; align-items: center; justify-content: center;
+    font-size: 0.9rem;
+    flex-shrink: 0;
+    color: #fff;
+}
+.msg-ai .bubble {
+    background: var(--card);
+    border: 1px solid var(--border);
+    color: var(--text);
+    padding: 0.85rem 1.2rem;
+    border-radius: 4px 14px 14px 14px;
+    max-width: 82%;
+    font-size: 0.92rem;
+    line-height: 1.7;
+}
+
+.stTextInput > div > div > input {
+    background: var(--card) !important;
+    border: 1px solid var(--border) !important;
+    border-radius: 10px !important;
+    color: var(--text) !important;
+    font-family: 'Nunito', sans-serif !important;
+    font-size: 0.95rem !important;
+    padding: 0.75rem 1rem !important;
+}
+.stTextInput > div > div > input:focus {
+    border-color: var(--accent) !important;
+    box-shadow: 0 0 0 2px rgba(56,189,248,0.15) !important;
+}
+
+.stButton > button {
+    background: linear-gradient(135deg, #1e4080, #1a3060) !important;
+    border: 1px solid var(--accent) !important;
+    color: var(--accent) !important;
+    border-radius: 10px !important;
+    font-family: 'Rajdhani', sans-serif !important;
+    font-weight: 600 !important;
+    letter-spacing: 1px !important;
+    padding: 0.5rem 1.5rem !important;
+    transition: all 0.2s !important;
+}
+.stButton > button:hover {
+    background: linear-gradient(135deg, #2a50a0, #2040a0) !important;
+    transform: translateY(-1px) !important;
+    box-shadow: 0 4px 15px rgba(56,189,248,0.2) !important;
+}
+
+[data-testid="stSidebar"] {
+    background: var(--surface) !important;
+    border-right: 1px solid var(--border) !important;
+}
+[data-testid="stSidebar"] .stMarkdown h3 {
+    font-family: 'Rajdhani', sans-serif;
+    color: var(--accent);
+    letter-spacing: 1px;
+}
+
+.topic-pill {
+    background: rgba(129,140,248,0.08);
+    border: 1px solid rgba(129,140,248,0.2);
+    color: var(--accent2);
+    padding: 4px 12px;
+    border-radius: 20px;
+    font-size: 0.78rem;
+    margin: 3px 2px;
+    display: inline-block;
+    font-family: 'JetBrains Mono', monospace;
+}
+
+::-webkit-scrollbar { width: 6px; }
+::-webkit-scrollbar-track { background: var(--surface); }
+::-webkit-scrollbar-thumb { background: var(--border); border-radius: 3px; }
+::-webkit-scrollbar-thumb:hover { background: var(--muted); }
+
+.stSelectbox > div > div {
+    background: var(--card) !important;
+    border-color: var(--border) !important;
+    color: var(--text) !important;
+    border-radius: 10px !important;
+}
+</style>
+""", unsafe_allow_html=True)
+
+# ─── Session State ──────────────────────────────────────────────────────────────
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+if "last_submitted" not in st.session_state:
+    st.session_state.last_submitted = ""  # prevents infinite rerun loop
+
+# ─── SymPy Engine ───────────────────────────────────────────────────────────────
+def run_sympy(problem: str) -> dict:
+    """
+    Safely attempt symbolic computation.
+    Uses sp.* explicitly — no wildcard collisions possible here.
+    """
+    p = problem.lower().strip()
+    x = sp.Symbol('x')
+    default = {"type": "general", "result": None, "latex": ""}
+
+    try:
+        tfms = standard_transformations + (implicit_multiplication_application,)
+
+        def clean(s):
+            """Normalise expression string for sympy parsing."""
+            s = re.sub(r"\s+", "", s)
+            s = re.sub(r"\^", "**", s)
+            return s
+
+        # ── Derivative ──
+        if any(k in p for k in ["derivative", "differentiate", "d/dx", "diff"]):
+            raw = p
+            for kw in ["derivative of", "differentiate", "diff of", "d/dx of", "d/dx"]:
+                if kw in p:
+                    raw = p.split(kw, 1)[-1].strip()
+                    break
+            raw = re.sub(r"\s*(dx|with\s*respect\s*to\s*x).*$", "", raw).strip()
+            expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+            result = sp.diff(expr, x)
+            return {"type": "Derivative", "result": str(result), "latex": sp.latex(result)}
+
+        # ── Integral ──
+        elif any(k in p for k in ["integral", "integrate", "antiderivative"]):
+            raw = p
+            for kw in ["integral of", "integrate", "antiderivative of"]:
+                if kw in p:
+                    raw = p.split(kw, 1)[-1].strip()
+                    break
+            raw = re.sub(r"\s*dx.*$", "", raw).strip()
+            expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+            result = sp.integrate(expr, x)
+            return {"type": "Integral", "result": str(result), "latex": sp.latex(result)}
+
+        # ── Limit ──
+        elif "limit" in p:
+            match = re.search(
+                r"limit\s+of\s+([\w\s\(\)\+\-\*/\^\.\,]+?)\s+as\s+x\s*(?:->|→|approaches)\s*([\w\.\+\-]+)",
+                p
+            )
+            if match:
+                raw_expr = clean(match.group(1))
+                point_str = match.group(2).strip()
+                point = sp.oo if point_str in ("inf", "infinity", "oo") else sp.sympify(point_str)
+                expr = parse_expr(raw_expr, transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+                result = sp.limit(expr, x, point)
+                return {"type": "Limit", "result": str(result), "latex": sp.latex(result)}
+
+        # ── Solve equation ──
+        elif any(k in p for k in ["solve", "roots", "find x"]):
+            raw = re.sub(r"(solve|find x|roots of|roots|the equation)", "", p).strip().strip(": ")
+            if "=" in raw:
+                lhs_s, rhs_s = raw.split("=", 1)
+                lhs_e = parse_expr(clean(lhs_s), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+                rhs_e = parse_expr(clean(rhs_s), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+                expr = lhs_e - rhs_e
+            else:
+                expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+            if x in expr.free_symbols:
+                sol = sp.solve(expr, x)
+                sol_latex = ", ".join([sp.latex(s) for s in sol])
+                return {"type": "Equation", "result": str(sol), "latex": f"x \\in \\{{{sol_latex}\\}}"}
+
+        # ── Matrix / eigenvalues — delegate to Claude ──
+        elif any(k in p for k in ["matrix", "determinant", "eigenvalue", "eigenvector", "det("]):
+            return {"type": "Matrix", "result": "matrix_detected", "latex": ""}
+
+        # ── Modular arithmetic — delegate to Claude ──
+        elif "mod" in p or "congruence" in p:
+            return {"type": "NumberTheory", "result": "mod_detected", "latex": ""}
+
+    except Exception:
+        pass  # fall back gracefully; Claude handles it without sympy hint
+
+    return default
+
+
+# ─── Claude API Call ────────────────────────────────────────────────────────────
+def ask_claude(problem: str, sympy_info: dict, history: list) -> str:
+    sympy_context = ""
+    if (sympy_info.get("result")
+            and sympy_info["result"] not in (None, "matrix_detected", "mod_detected")):
+        sympy_context = f"""
+SYMPY VERIFIED RESULT (mathematically exact — your final answer MUST match this):
+- Operation : {sympy_info.get('type', 'N/A')}
+- Result    : {sympy_info.get('result', 'N/A')}
+- LaTeX     : {sympy_info.get('latex', 'N/A')}
+"""
+
+    system_prompt = f"""You are SAAD AI — an elite BSc Mathematics tutor.
+{sympy_context}
+RULES:
+1. Label every step: "Step 1:", "Step 2:", etc. — no steps skipped.
+2. Use $...$ for inline math and $$...$$ for display equations.
+3. State the method/theorem at the start.
+4. Show ALL intermediate algebra.
+5. End with: **Final Answer:** $$...$$
+6. If a SymPy verified result is shown above, your final answer MUST match it exactly.
+7. NEVER make arithmetic errors.
+
+TOPICS: Calculus | Linear Algebra | Number Theory | ODEs | Numerical Methods |
+Differential Geometry | Hydro Mechanics | Theory of Numbers | Real Analysis II | General Math
+"""
+
+    messages = []
+    for msg in history[-8:]:
+        messages.append({"role": msg["role"], "content": msg["content"]})
+    messages.append({"role": "user", "content": problem})
+
+    client = anthropic.Anthropic()
+    response = client.messages.create(
+        model="claude-sonnet-4-5",
+        max_tokens=3000,
+        system=system_prompt,
+        messages=messages
+    )
+    return response.content[0].text
+
+
+# ─── Sidebar ────────────────────────────────────────────────────────────────────
+with st.sidebar:
+    st.markdown("### 📚 SAAD AI ACADEMY")
+    st.markdown("---")
+
+    st.markdown("**🎯 Topics Covered**")
+    topics = [
+        "Calculus", "Linear Algebra", "Number Theory",
+        "Differential Equations", "Numerical Methods",
+        "Differential Geometry", "Hydro Mechanics",
+        "Theory of Numbers", "Real Analysis II", "General Math"
+    ]
+    pills_html = "".join([f'<span class="topic-pill">{t}</span>' for t in topics])
+    st.markdown(pills_html, unsafe_allow_html=True)
+
+    st.markdown("---")
+    st.markdown("**⚡ Quick Examples**")
+
+    examples = {
+        "-- Choose --": "",
+        "🔢 Derivative": "Find the derivative of x^3 + 5x^2 - 3x + 7",
+        "∫ Integral": "Integrate sin(x) * e^x dx",
+        "📊 Eigenvalue": "Find eigenvalues of matrix [[3,1],[1,3]]",
+        "🔁 Congruence": "Solve 14x ≡ 30 (mod 44)",
+        "🌊 ODE": "Solve dy/dx + 2y = e^x",
+        "📐 Limit": "Find limit of sin(x)/x as x -> 0",
+        "🔍 Series": "Test convergence of sum 1/n^2 from n=1 to infinity",
+        "📏 Newton-Raphson": "Apply Newton-Raphson to find root of x^3 - 2x - 5 = 0",
+    }
+
+    selected_example = st.selectbox(
+        "Load an example:",
+        list(examples.keys()),
+        key="example_select"
+    )
+
+    st.markdown("---")
+    st.markdown("**🔧 Engine**")
+    st.markdown("""
+    <div style="font-size:0.8rem; color:#64748b; line-height:1.8;">
+    🤖 <b style="color:#38bdf8">Claude AI</b> — reasoning & steps<br>
+    🧮 <b style="color:#34d399">SymPy</b> — verified computation<br>
+    ✅ Zero arithmetic errors
+    </div>
+    """, unsafe_allow_html=True)
+
+    st.markdown("---")
+    if st.button("🗑️ Clear Chat"):
+        st.session_state.messages = []
+        st.session_state.last_submitted = ""
+        st.rerun()
+
+
+# ─── Hero Banner ────────────────────────────────────────────────────────────────
+st.markdown("""
+<div class="hero">
+    <div class="hero-title">📐 SAAD AI ACADEMY</div>
+    <div class="hero-sub">BSc Mathematics — Computational Engine</div>
+    <div class="badge-row">
+        <span class="badge">Claude AI</span>
+        <span class="badge">SymPy Verified</span>
+        <span class="badge">Step-by-Step</span>
+        <span class="badge">Zero Errors</span>
+        <span class="badge">LaTeX Output</span>
+    </div>
+</div>
+""", unsafe_allow_html=True)
+
+
+# ─── Chat Display ────────────────────────────────────────────────────────────────
+chat_html = '<div class="chat-area" id="chatArea">'
+
+if not st.session_state.messages:
+    chat_html += """
+    <div style="display:flex;flex-direction:column;align-items:center;justify-content:center;
+                height:300px;color:#334155;text-align:center;">
+        <div style="font-size:3rem;margin-bottom:1rem;">📐</div>
+        <div style="font-family:'Rajdhani',sans-serif;font-size:1.2rem;color:#475569;letter-spacing:1px;">
+            Ask any BSc Mathematics problem
+        </div>
+        <div style="font-size:0.82rem;color:#334155;margin-top:0.5rem;">
+            Calculus · Linear Algebra · Number Theory · ODEs · Real Analysis · and more
+        </div>
+    </div>
+    """
+else:
+    for msg in st.session_state.messages:
+        if msg["role"] == "user":
+            # Escape HTML to prevent rendering issues with math symbols
+            safe = (msg['content']
+                    .replace("&", "&amp;")
+                    .replace("<", "&lt;")
+                    .replace(">", "&gt;"))
+            chat_html += f'<div class="msg-user"><div class="bubble">🧑‍🎓 {safe}</div></div>'
+        else:
+            content = msg['content']
+            # Escape first, then apply safe bold conversion
+            content = (content
+                       .replace("&", "&amp;")
+                       .replace("<", "&lt;")
+                       .replace(">", "&gt;"))
+            content = re.sub(r'\*\*(.*?)\*\*', r'<b>\1</b>', content)
+            content = content.replace('\n', '<br>')
+            chat_html += (
+                f'<div class="msg-ai">'
+                f'<div class="ai-avatar">∑</div>'
+                f'<div class="bubble">{content}</div>'
+                f'</div>'
+            )
+
+chat_html += '</div>'
+st.markdown(chat_html, unsafe_allow_html=True)
+
+st.markdown("""
+<script>
+    const c = document.getElementById('chatArea');
+    if (c) c.scrollTop = c.scrollHeight;
+</script>
+""", unsafe_allow_html=True)
+
+
+# ─── Input Row ──────────────────────────────────────────────────────────────────
+col1, col2 = st.columns([5, 1])
+
+with col1:
+    prefill = examples.get(selected_example, "") if selected_example != "-- Choose --" else ""
+    user_input = st.text_input(
+        label="problem_input",
+        value=prefill,
+        placeholder="Type a BSc math problem... e.g. 'Solve d²y/dx² + 4y = cos(2x)'",
+        label_visibility="collapsed",
+        key="user_input_field"
+    )
+
+with col2:
+    send_btn = st.button("SOLVE →", use_container_width=True)
+
+
+# ─── Process Input ───────────────────────────────────────────────────────────────
+# Only fires when SOLVE is clicked, input is non-empty, AND it's a new problem
+# This prevents the infinite rerun loop caused by value="" pre-fill on every cycle
+problem = user_input.strip() if user_input else ""
+
+if send_btn and problem and problem != st.session_state.last_submitted:
+    st.session_state.last_submitted = problem
+
+    with st.spinner("🧮 Running SymPy verification + Claude reasoning..."):
+        sympy_result = run_sympy(problem)
+        try:
+            answer = ask_claude(problem, sympy_result, st.session_state.messages)
+        except Exception as e:
+            answer = (
+                f"⚠️ **API Error:** {str(e)}\n\n"
+                "Please ensure **ANTHROPIC_API_KEY** is added in your Hugging Face Space → "
+                "Settings → Variables and Secrets."
+            )
+
+    st.session_state.messages.append({"role": "user", "content": problem})
+    st.session_state.messages.append({"role": "assistant", "content": answer})
+    st.rerun()