Update app.py

app.py

@@ -3,19 +3,19 @@ from transformers import pipeline
 import torch
 
 # Professional Academic Layout
-st.set_page_config(page_title="B.Sc. Math Engine", page_icon="📐", layout="centered")
+st.set_page_config(page_title="B.Sc. Math Engine", page_icon="📐")
 
 st.title("📐 B.Sc. Mathematics Engine")
 st.write("Advanced Proof & Logic Solver | SUST Dept. of Mathematics")
 
 @st.cache_resource
 def load_engine():
-    # Switching to Llama-1B: Much smarter logic for Group Theory/Analysis
-    model_id = "meta-llama/Llama-3.2-1B-Instruct"
+    # SmolLM2-1.7B: The best middle-ground for logic vs speed on free CPU
+    model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
     return pipeline(
         "text-generation", 
         model=model_id, 
-        device_map="auto", 
+        device=-1, # Force CPU usage
         torch_dtype=torch.float32
     )
 
@@ -24,31 +24,28 @@ pipe = load_engine()
 # Strict Academic System Prompt
 system_instruction = (
     "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."
+    "Instructions: "
+    "1. Use LaTeX for all mathematical notation. "
+    "2. For group theory, a homomorphism phi satisfies phi(ab) = phi(a)phi(b). "
+    "3. The kernel is the set of elements mapping to the identity element (1 for multiplication). "
+    "4. Provide clear, logical steps for the proof."
 )
 
-user_query = st.chat_input("Enter your problem (e.g., Prove the kernel of z^2 is {1, -1})")
+user_query = st.chat_input("Enter your problem...")
 
 if user_query:
     with st.chat_message("user"):
         st.write(user_query)
 
     with st.chat_message("assistant"):
-        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}
-            ]
+        with st.spinner("Processing logic... (Usually takes 45-90s on CPU)"):
+            # Format specifically for SmolLM2
+            prompt = f"<|im_start|>system\n{system_instruction}<|im_end|>\n<|im_start|>user\n{user_query}<|im_end|>\n<|im_start|>assistant\n"
             
-            # 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']
+            # do_sample=False ensures the math is consistent (Temperature 0)
+            result = pipe(prompt, max_new_tokens=600, do_sample=False)
+            response = result[0]['generated_text'].split("<|im_start|>assistant\n")[-1]
             
             st.markdown(response)
 
-st.sidebar.markdown("### SUST Math Module")
-st.sidebar.info("This engine uses a 1-Billion parameter logic model optimized for CPU execution.")
+st.sidebar.info("Model: SmolLM2-1.7B | Mode: Exact Logic")