Update app.py

app.py

@@ -1,62 +1,76 @@
 import streamlit as st
-from transformers import pipeline
-import torch
+import sympy as sp
+import re
 
 # 1. PAGE CONFIG
-st.set_page_config(page_title="SAAD AI ACADEMY", page_icon="📐", layout="centered")
+st.set_page_config(page_title="SAAD AI ACADEMY", layout="centered")
 
-# 2. DARK MODE UI
+# 2. DARK UI
 st.markdown("""
     <style>
-    .stApp { background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%); color: #ffffff; }
-    .stChatMessage { background: rgba(255, 255, 255, 0.05) !important; border-radius: 15px !important; border: 1px solid rgba(255, 255, 255, 0.1) !important; }
-    h1, h2, h3, p, span { color: #ffffff !important; }
-    .stCaption { color: #3b82f6 !important; }
+    .stApp { background: #0f172a; color: white; }
+    h1, h2, h3 { color: #3b82f6 !important; }
+    .stTextInput > div > div > input { background-color: #1e293b; color: white; border: 1px solid #3b82f6; }
     </style>
     """, unsafe_allow_html=True)
 
 st.title("📐 SAAD AI ACADEMY")
-st.caption("Stable Mathematics Engine | SUST Logic System")
-
-# 3. STABLE LOCAL LOADING
-@st.cache_resource
-def load_engine():
-    model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
-    # Loading locally to ensure 100% uptime and no API timeouts
-    return pipeline("text-generation", model=model_id, device=-1, torch_dtype=torch.float32)
-
-with st.spinner("⏳ Restoring the Stable Engine..."):
-    pipe = load_engine()
-
-# 4. FIXED LOGIC INSTRUCTIONS
-# Added a check so it doesn't give random math for greetings
-system_instruction = (
-    "You are a helpful assistant and a Senior Mathematics Professor. "
-    "If the user greets you or asks a general question, respond normally. "
-    "ONLY if the user provides a math problem, follow these rules:\n"
-    "1. State the math domain.\n"
-    "2. Show all steps using LaTeX ($...$).\n"
-    "3. Bold the final result."
-)
-
-# 5. INTERFACE
-user_query = st.chat_input("Enter your query...")
-
-if user_query:
-    with st.chat_message("user"):
-        st.write(user_query)
-
-    with st.chat_message("assistant"):
-        with st.spinner("🧠 Processing..."):
-            prompt = f"<|im_start|>system\n{system_instruction}<|im_end|>\n<|im_start|>user\n{user_query}<|im_end|>\n<|im_start|>assistant\n"
-            
-            # Fixed generation settings to prevent hallucinations
-            result = pipe(prompt, max_new_tokens=800, temperature=0.7, do_sample=True)
-            response = result[0]['generated_text'].split("<|im_start|>assistant\n")[-1]
-            st.markdown(response)
-
-# 6. SIDEBAR
-st.sidebar.markdown(f"### 🎓 Student Profile")
+st.subheader("B.Sc. Math Engine v6.0 (Zero-Error Logic)")
+
+# 3. COMPUTATION ENGINE
+def solve_math(query):
+    try:
+        x = sp.Symbol('x')
+        
+        # A. HANDLE LINEAR CONGRUENCE: "solve 14x = 30 mod 44"
+        if "mod" in query.lower():
+            # Extract numbers using Regex
+            nums = re.findall(r'\d+', query)
+            if len(nums) >= 3:
+                a, b, n = map(int, nums[:3])
+                # SymPy's solver for ax ≡ b (mod n)
+                solutions = sp.solve_linear_congruence(a, b, n)
+                return f"The solutions are: x \equiv {solutions} \pmod{{{n}}}"
+            return "Format error. Try: 'solve 14x = 30 mod 44'"
+
+        # B. HANDLE DERIVATIVES: "derivative of x**2 + 5*x"
+        if "derivative" in query.lower() or "diff" in query.lower():
+            expr_str = query.split("of")[-1].strip()
+            expr = sp.sympify(expr_str)
+            return sp.diff(expr, x)
+
+        # C. HANDLE INTEGRALS: "integrate x**2"
+        if "integrate" in query.lower():
+            expr_str = query.split("integrate")[-1].strip()
+            expr = sp.sympify(expr_str)
+            return sp.integrate(expr, x)
+
+        # D. GENERAL SOLVE: "solve x**2 - 4"
+        if "solve" in query.lower() and "mod" not in query.lower():
+            expr_str = query.split("solve")[-1].strip()
+            expr = sp.sympify(expr_str)
+            return sp.solve(expr, x)
+
+        return "I can solve Congruences, Derivatives, Integrals, and Equations. Please be specific!"
+
+    except Exception as e:
+        return f"Logic Error: {e}. Please use Python syntax like x**2 for x²."
+
+# 4. INTERFACE
+user_input = st.text_input("Enter your problem:", placeholder="e.g., solve 14x = 30 mod 44")
+
+if user_input:
+    res = solve_math(user_input)
+    st.markdown("---")
+    st.markdown("### 🎯 Result:")
+    if isinstance(res, str):
+        st.info(res)
+    else:
+        # Display as beautiful LaTeX
+        st.latex(sp.latex(res))
+
+# 5. SIDEBAR
+st.sidebar.markdown("### 🎓 Academic Profile")
 st.sidebar.write("**Name:** Almuyed Saad")
 st.sidebar.write("**Inst:** SUST")
-st.sidebar.write("**Major:** Mathematics")
+st.sidebar.info("This engine uses SymPy for symbolic math, ensuring 100% precision for B.Sc. coursework.")