Upload app.py

app.py

@@ -1807,11 +1807,24 @@ def ask_ai(problem: str, sympy_info: dict, history: list) -> str:
         )
 
     system_prompt = (
-        "You are Saad.AI, a precise BSc Mathematics tutor built by Saad. "
+        "You are Saad.AI, a BSc Mathematics assistant built by Saad. "
+        "You are friendly, helpful, and professional — like ChatGPT or Claude. "
         "If anyone asks who made you, who built you, who created you, or who invented you, "
         "always say: I was built by Saad, a passionate developer who created me from scratch "
         "to help BSc Mathematics students. Never mention Groq, Meta, Gemini, or any AI company as your creator.\n\n"
 
+        "=== CONVERSATION MODE ===\n"
+        "You have TWO modes:\n\n"
+        "MODE A — CASUAL (when sympy type is 'casual' or message is a greeting/small talk):\n"
+        "  → Respond naturally like ChatGPT or Claude — warm, friendly, conversational.\n"
+        "  → NO math structure. NO steps. NO boxed answers. NO LaTeX.\n"
+        "  → Just reply naturally in 1-3 sentences.\n"
+        "  → Examples: 'Hi!' → 'Hey! How can I help you today?'\n"
+        "              'How are you?' → 'Doing great! Ready to tackle some math. What would you like to solve?'\n"
+        "              'What can you do?' → Briefly explain you solve BSc Math problems step by step.\n\n"
+        "MODE B — MATH (when sympy type is anything else — actual math problem):\n"
+        "  → Use full math structure below.\n\n"
+
         "═══════════════════════════════════════════════════════\n"
         "CORE FORMATTING RULES — follow every rule without exception\n"
         "═══════════════════════════════════════════════════════\n\n"
@@ -2219,28 +2232,17 @@ with st.sidebar:
 
 # Header
 st.markdown("""
-<div style="text-align:center; padding:1.4rem 0 0.8rem 0; margin-bottom:0.5rem;">
+<div style="text-align:center; padding:2rem 0 1rem 0;">
     <div style="
-        font-size:2.2rem;
+        font-size:2.4rem;
         font-weight:700;
-        background: linear-gradient(135deg, #3b82f6 0%, #a78bfa 50%, #60a5fa 100%);
-        -webkit-background-clip: text;
-        -webkit-text-fill-color: transparent;
-        background-clip: text;
-        color: #3b82f6;
-        letter-spacing:1px;
-        margin-bottom:0.3rem;
-        font-family: 'Inter', sans-serif;
-    ">📐 Saad.AI</div>
-    <div style="font-size:0.8rem; color:#6b7280; letter-spacing:0.3px; margin-bottom:0.7rem;">
-        BSc Mathematics &nbsp;·&nbsp; Step-by-Step Solver &nbsp;·&nbsp; SymPy Verified
-    </div>
-    <div style="display:flex; gap:0.4rem; justify-content:center; flex-wrap:wrap;">
-        <span style="background:#0f172a; border:1px solid #1e3a5f; color:#60a5fa; padding:3px 10px; border-radius:20px; font-size:0.68rem; font-family:monospace;">⚡ Groq Free</span>
-        <span style="background:#0f172a; border:1px solid #1e3a5f; color:#60a5fa; padding:3px 10px; border-radius:20px; font-size:0.68rem; font-family:monospace;">🧮 SymPy Verified</span>
-        <span style="background:#0f172a; border:1px solid #1e3a5f; color:#60a5fa; padding:3px 10px; border-radius:20px; font-size:0.68rem; font-family:monospace;">📊 Step-by-Step</span>
-        <span style="background:#0f172a; border:1px solid #1e3a5f; color:#60a5fa; padding:3px 10px; border-radius:20px; font-size:0.68rem; font-family:monospace;">∑ LaTeX</span>
-        <span style="background:#0f172a; border:1px solid #1e3a5f; color:#60a5fa; padding:3px 10px; border-radius:20px; font-size:0.68rem; font-family:monospace;">📈 Graphs</span>
+        color:#ffffff;
+        letter-spacing:0.5px;
+        margin-bottom:0.4rem;
+        font-family:'Inter',sans-serif;
+    ">Saad.AI</div>
+    <div style="font-size:0.85rem; color:#6b7280;">
+        Your BSc Mathematics Assistant
     </div>
 </div>
 """, unsafe_allow_html=True)
@@ -2248,43 +2250,45 @@ st.markdown("""
 # Welcome message if no chat
 if not st.session_state.messages:
     st.markdown("""
-    <div style="text-align:center; padding:1.5rem 0 0.5rem 0;">
-        <div style="font-size:0.85rem; color:#6b7280; margin-bottom:1.2rem;">
-            Ask any BSc Mathematics problem — I'll solve it step by step
+    <div style="text-align:center; padding:3rem 0 2rem 0;">
+        <div style="
+            font-size:3rem;
+            margin-bottom:0.8rem;
+            animation: pulse 2s infinite;
+        ">🧮</div>
+        <div style="font-size:1.1rem; color:#d1d5db; font-weight:500; margin-bottom:0.5rem;">
+            How can I help you today?
         </div>
-        <div style="display:grid; grid-template-columns:repeat(3,1fr); gap:0.5rem; margin:0 0 1rem 0;">
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">📈</div>Calculus
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">🔢</div>Linear Algebra
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">📉</div>ODEs
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">🧮</div>Numerical Methods
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">🔍</div>Number Theory
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">📐</div>Diff. Geometry
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">🌊</div>Hydro Mechanics
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">📊</div>Real Analysis II
-            </div>
-            <div style="background:#111827; border:1px solid #1f2937; border-radius:10px; padding:0.7rem 0.5rem; text-align:center; font-size:0.75rem; color:#9ca3af;">
-                <div style="font-size:1.3rem; margin-bottom:0.2rem;">📈</div>Graph Plotting
-            </div>
+        <div style="font-size:0.82rem; color:#4b5563; margin-bottom:2rem;">
+            Ask me anything — math problems, explanations, or just say hi
         </div>
-        <div style="font-size:0.72rem; color:#374151; margin-top:0.5rem;">
-            💡 Tip: Select an example from the sidebar or type your problem below
+        <div style="display:flex; flex-wrap:wrap; gap:0.5rem; justify-content:center; max-width:600px; margin:0 auto;">
+            <span style="background:#111827; border:1px solid #1f2937; color:#9ca3af; padding:0.5rem 1rem; border-radius:20px; font-size:0.8rem; cursor:pointer;">
+                📈 Differentiate x³ + 2x
+            </span>
+            <span style="background:#111827; border:1px solid #1f2937; color:#9ca3af; padding:0.5rem 1rem; border-radius:20px; font-size:0.8rem; cursor:pointer;">
+                ∫ Integrate sin(x)·eˣ
+            </span>
+            <span style="background:#111827; border:1px solid #1f2937; color:#9ca3af; padding:0.5rem 1rem; border-radius:20px; font-size:0.8rem; cursor:pointer;">
+                🔢 Solve ODE dy/dx + 2y = e⁻ˣ
+            </span>
+            <span style="background:#111827; border:1px solid #1f2937; color:#9ca3af; padding:0.5rem 1rem; border-radius:20px; font-size:0.8rem; cursor:pointer;">
+                🧮 Newton-Raphson x³−2x−5=0
+            </span>
+            <span style="background:#111827; border:1px solid #1f2937; color:#9ca3af; padding:0.5rem 1rem; border-radius:20px; font-size:0.8rem; cursor:pointer;">
+                📐 Find curvature of y=x²
+            </span>
+            <span style="background:#111827; border:1px solid #1f2937; color:#9ca3af; padding:0.5rem 1rem; border-radius:20px; font-size:0.8rem; cursor:pointer;">
+                📊 Plot y = x² − 4
+            </span>
         </div>
     </div>
+    <style>
+    @keyframes pulse {
+        0%, 100% { opacity: 1; }
+        50% { opacity: 0.6; }
+    }
+    </style>
     """, unsafe_allow_html=True)
 
 # Render chat history
@@ -2325,30 +2329,53 @@ else:
 if problem and problem != st.session_state.last_submitted:
     st.session_state.last_submitted = problem
 
+    # ── Detect casual / non-math messages ───────────────────────────
+    p_lower = problem.lower().strip()
+    casual_keywords = [
+        "hi", "hello", "hey", "how are you", "how r u", "what's up",
+        "whats up", "good morning", "good evening", "good night",
+        "who are you", "what are you", "what can you do", "help",
+        "thanks", "thank you", "bye", "goodbye", "ok", "okay",
+        "what is your name", "your name", "who made you", "who built you",
+        "how do you work", "what do you do", "sup", "hlo", "hlw",
+        "what u doing", "what are you doing", "hows it going",
+    ]
+    is_casual = (
+        any(p_lower == kw for kw in casual_keywords) or
+        any(p_lower.startswith(kw) for kw in casual_keywords) or
+        (len(p_lower.split()) <= 4 and not any(c in p_lower for c in [
+            "=", "+", "-", "*", "/", "^", "∫", "∑", "√", "dx", "dy",
+            "sin", "cos", "tan", "log", "lim", "diff", "solve", "find",
+            "calculate", "compute", "prove", "matrix", "eigen", "gcd",
+            "integral", "derivative", "equation", "theorem"
+        ]))
+    )
+
     # Show user message immediately
     with st.chat_message("user", avatar="🧑‍🎓"):
         st.markdown(problem)
 
-    # Show AI response with streaming
+    # Show AI response
     with st.chat_message("assistant", avatar="📐"):
-        # Step 1: SymPy computation (instant)
-        import random
-        spinner_msgs = [
-            "🧮 Computing with SymPy...",
-            "∫ Integrating the solution...",
-            "∑ Summing it all up...",
-            "📐 Applying the formula...",
-            "🔢 Crunching the numbers...",
-            "📊 Verifying the answer...",
-        ]
-        with st.spinner(random.choice(spinner_msgs)):
-            sympy_result = run_sympy(problem)
-
-        # Step 2: Stream AI response word by word
-        answer = ask_ai_streaming(problem, sympy_result, st.session_state.messages)
-
-        # Step 3: Plot graph if user asked for it
-        plot_graph(problem, sympy_result)
+        if is_casual:
+            # ── Casual message — no SymPy, no math structure ─────────
+            casual_sympy = {"type": "casual", "result": None, "latex": ""}
+            answer = ask_ai_streaming(problem, casual_sympy, st.session_state.messages)
+        else:
+            # ── Math message — full SymPy + structured response ───────
+            import random
+            spinner_msgs = [
+                "🧮 Computing with SymPy...",
+                "∫ Integrating the solution...",
+                "∑ Summing it all up...",
+                "📐 Applying the formula...",
+                "🔢 Crunching the numbers...",
+                "📊 Verifying the answer...",
+            ]
+            with st.spinner(random.choice(spinner_msgs)):
+                sympy_result = run_sympy(problem)
+            answer = ask_ai_streaming(problem, sympy_result, st.session_state.messages)
+            plot_graph(problem, sympy_result)
 
     # Save to history
     st.session_state.messages.append({"role": "user", "content": problem})