Update app.py

app.py

@@ -1,51 +1,54 @@
 import streamlit as st
 from transformers import pipeline
+import torch
 
-# 1. Page Configuration for a high-end UI
-st.set_page_config(page_title="SAAD AI ACADEMY | Math Engine", page_icon="🎓", layout="centered")
+# Professional Academic Layout
+st.set_page_config(page_title="B.Sc. Math Engine", page_icon="📐", layout="centered")
 
-st.markdown("""
-    <style>
-    .stApp { background: linear-gradient(to bottom, #0f172a, #000000); color: white; }
-    .stButton>button { width: 100%; border-radius: 20px; background: #3b82f6; color: white; }
-    </style>
-    """, unsafe_allow_html=True)
+st.title("📐 B.Sc. Mathematics Engine")
+st.write("Advanced Proof & Logic Solver | SUST Dept. of Mathematics")
 
 @st.cache_resource
 def load_engine():
-    # Fast 360M model optimized for CPU
-    return pipeline("text-generation", model="HuggingFaceTB/SmolLM2-360M-Instruct", device=-1)
+    # Switching to Llama-1B: Much smarter logic for Group Theory/Analysis
+    model_id = "meta-llama/Llama-3.2-1B-Instruct"
+    return pipeline(
+        "text-generation", 
+        model=model_id, 
+        device_map="auto", 
+        torch_dtype=torch.float32
+    )
 
 pipe = load_engine()
 
-st.title("🎓 SAAD AI ACADEMY")
-st.subheader("B.Sc. Mathematics Engine V4")
-st.caption("Optimized for Number Theory, Calculus & Linear Algebra")
-
-# 2. Professor Mode Instructions
+# Strict Academic System Prompt
 system_instruction = (
-    "You are a Senior Mathematics Professor. Solve the problem with extreme rigor. "
-    "Use LaTeX for all mathematical notation. If it's a divisibility problem, "
-    "always consider Fermat's Little Theorem or Induction."
+    "You are a rigorous Mathematics Professor. "
+    "1. Always use LaTeX for equations. "
+    "2. A homomorphism must satisfy f(ab) = f(a)f(b). "
+    "3. The kernel is the set {x in G : f(x) = e}, where e is the identity. "
+    "4. Verify every logical step before outputting."
 )
 
-user_query = st.chat_input("Enter your SUST Math problem...")
+user_query = st.chat_input("Enter your problem (e.g., Prove the kernel of z^2 is {1, -1})")
 
 if user_query:
     with st.chat_message("user"):
         st.write(user_query)
 
     with st.chat_message("assistant"):
-        with st.spinner("⏳ Analyzing mathematical structures..."):
-            prompt = f"<|im_start|>system\n{system_instruction}<|im_end|>\n<|im_start|>user\n{user_query}<|im_end|>\n<|im_start|>assistant\n"
+        with st.spinner("Calculating logic... (May take 30-60s on Free Tier)"):
+            # Formatting for Llama-3.2
+            messages = [
+                {"role": "system", "content": system_instruction},
+                {"role": "user", "content": user_query}
+            ]
             
-            # Increased tokens for longer proofs
-            result = pipe(prompt, max_new_tokens=800, temperature=0.1, do_sample=False)
-            response = result[0]['generated_text'].split("<|im_start|>assistant\n")[-1]
+            # do_sample=False (Temperature 0) for mathematical consistency
+            result = pipe(messages, max_new_tokens=512, do_sample=False)
+            response = result[0]['generated_text'][-1]['content']
             
             st.markdown(response)
-            
-            # 3. Add a simple Copy/Download button
-            st.download_button("Download Solution", response, file_name="solution.txt")
 
-st.sidebar.info("Tip: Use standard notation like x^2, integrals, or matrices.")
+st.sidebar.markdown("### SUST Math Module")
+st.sidebar.info("This engine uses a 1-Billion parameter logic model optimized for CPU execution.")