Update app.py

app.py

@@ -1,76 +1,55 @@
 import streamlit as st
 import sympy as sp
 import re
+from sympy.ntheory.modular import solve_linear_congruence
 
-# 1. PAGE CONFIG
+# 1. SETUP
 st.set_page_config(page_title="SAAD AI ACADEMY", layout="centered")
 
-# 2. DARK UI
-st.markdown("""
-    <style>
-    .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.subheader("B.Sc. Math Engine v6.0 (Zero-Error Logic)")
-
-# 3. COMPUTATION ENGINE
-def solve_math(query):
+# 2. THE UNIVERSAL SOLVER FUNCTION
+def universal_solver(query):
     try:
-        x = sp.Symbol('x')
-        
-        # A. HANDLE LINEAR CONGRUENCE: "solve 14x = 30 mod 44"
+        # A. Check for Modular Congruence first (special case)
         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)
+                sol = solve_linear_congruence(a, b, n)
+                return f"Number Theory Result: x ≡ {sol[0]} (mod {sol[1]})" if sol else "No Solution."
 
-        # 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!"
+        # B. For everything else (Calculus, Algebra, Limits)
+        # We clean the query to remove words like "solve" or "find"
+        clean_query = query.lower().replace("solve", "").replace("find", "").replace("the", "").strip()
+        
+        # SymPy turns your text into a math expression
+        obj = sp.sympify(clean_query)
+        
+        # If it's an equation (has an '='), solve it
+        if "=" in query:
+            parts = query.split("=")
+            equation = sp.Eq(sp.sympify(parts[0]), sp.sympify(parts[1]))
+            return sp.solve(equation)
+        
+        # Otherwise, just simplify or "do" the math (like derivatives)
+        return sp.simplify(obj)
 
     except Exception as e:
-        return f"Logic Error: {e}. Please use Python syntax like x**2 for x²."
+        return f"Enter a valid math expression (e.g., x**2 + 5*x or 14x=30 mod 44). Error: {e}"
+
+# 3. UI
+st.title("📐 SAAD AI ACADEMY")
+st.write("Enter any B.Sc. Level problem (Algebra, Calculus, Number Theory)")
 
-# 4. INTERFACE
-user_input = st.text_input("Enter your problem:", placeholder="e.g., solve 14x = 30 mod 44")
+user_input = st.text_input("Input:", placeholder="e.g. diff(sin(x)*exp(x), x) or x**2 - 5*x + 6 = 0")
 
 if user_input:
-    res = solve_math(user_input)
-    st.markdown("---")
-    st.markdown("### 🎯 Result:")
-    if isinstance(res, str):
-        st.info(res)
+    result = universal_solver(user_input)
+    st.divider()
+    st.subheader("Result:")
+    if isinstance(result, (list, tuple, sp.Basic)):
+        st.latex(sp.latex(result))
     else:
-        # Display as beautiful LaTeX
-        st.latex(sp.latex(res))
+        st.write(result)
 
-# 5. SIDEBAR
-st.sidebar.markdown("### 🎓 Academic Profile")
-st.sidebar.write("**Name:** Almuyed Saad")
-st.sidebar.write("**Inst:** SUST")
-st.sidebar.info("This engine uses SymPy for symbolic math, ensuring 100% precision for B.Sc. coursework.")
+# 4. SIDEBAR
+st.sidebar.info("Universal Mode: Uses SymPy Symbolic Parsing.")