Update app.py

app.py

@@ -2,61 +2,62 @@ import streamlit as st
 from transformers import pipeline
 import torch
 
-# Professional Academic Layout
-st.set_page_config(page_title="B.Sc. Math Engine", page_icon="📐")
+# 1. Page Configuration for a Professional Academic Interface
+st.set_page_config(page_title="B.Sc. Math Engine V5", page_icon="📐", layout="centered")
+
+st.markdown("""
+    <style>
+    .stApp { background-color: #f0f2f6; }
+    .stChatMessage { border-radius: 15px; border: 1px solid #d1d5db; }
+    </style>
+    """, unsafe_allow_html=True)
+
 st.title("📐 B.Sc. Mathematics Engine")
-st.write("Advanced Proof & Logic Solver | SUST Dept. of Mathematics")
+st.caption("Advanced Logic & Proof System | Optimized for SUST 3rd Year")
 
 @st.cache_resource
 def load_engine():
-    # SmolLM2-1.7B: The best middle-ground for logic vs speed on free CPU
+    # SmolLM2-1.7B is the best logic engine for Free CPU spaces
     model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
     return pipeline(
         "text-generation", 
         model=model_id, 
-        device=-1, # Force CPU usage
+        device=-1, 
         torch_dtype=torch.float32
     )
 
 pipe = load_engine()
 
-# Strict Academic System Prompt
+# 2. THE LOGIC CONTROLLER (Prevents common B.Sc. Math errors)
 system_instruction = (
-    "You are a Senior Mathematics Professor at a top-tier university. "
-    "Follow these strict logical rules for all solutions:\n\n"
-    
-    "1. STRUCTURE: Always start with 'Definitions & Prerequisites', followed by 'Step-by-Step Proof', and end with 'Conclusion'.\n"
-    
-    "2. GROUP THEORY: To prove a subgroup H is normal, you must show gH = Hg for all g in G. "
-    "If the index [G:H]=2, use the partition G = H U (G-H) to show left and right cosets are identical.\n"
-    
-    "3. NUMBER THEORY: For divisibility (e.g., n^k - n), use Fermat's Little Theorem or Induction. "
-    "For congruences ax ≡ b (mod n), always calculate d = gcd(a, n) first. If d does not divide b, state 'No Solution'.\n"
-    
-    "4. REAL ANALYSIS: For limits, use the epsilon-delta or epsilon-N definition. "
-    "For convergence, state which test you are using (Ratio, Root, or Comparison).\n"
-    
-    "5. COMPLEX ANALYSIS: To check differentiability, verify Cauchy-Riemann equations (ux = vy and uy = -vx). "
-    "For integrals, identify poles and use the Residue Theorem.\n"
-    
-    "6. FORMATTING: Use LaTeX for EVERY mathematical symbol. Use '$$' for centered equations and '$' for inline math. "
-    "Never say 'it is obvious'; explain every logical jump."
+    "You are a Senior Mathematics Professor. You must follow a 'Verify-then-Solve' protocol.\n\n"
+    "RULE 1 (DIVISIBILITY): For ax ≡ b (mod n), you MUST check if d = gcd(a, n) divides b. "
+    "If it divides, divide the entire congruence by d: (a/d)x ≡ (b/d) (mod n/d) BEFORE finding the inverse.\n\n"
+    "RULE 2 (GROUPS): To prove H is normal where [G:H]=2, use the fact that gH and Hg are both "
+    "the complement G\\H when g is not in H. Do not assume gh=hg.\n\n"
+    "RULE 3 (REASONING): Think step-by-step. State the theorem you are using (e.g., Lagrange, Fermat, Cauchy-Riemann) "
+    "before applying it.\n\n"
+    "RULE 4 (FORMAT): Use LaTeX ($...$ or $$...$$) for all math. Never skip algebraic steps."
 )
 
-user_query = st.chat_input("Enter your problem...")
+# 3. User Interface
+user_query = st.chat_input("Enter your problem (e.g., Prove H is normal if [G:H]=2)")
 
 if user_query:
     with st.chat_message("user"):
         st.write(user_query)
+
     with st.chat_message("assistant"):
-        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"
+        with st.spinner("⏳ Analyzing mathematical structures and verifying logic..."):
+            # We add "Let's think step by step" to the prompt to trigger Chain-of-Thought
+            prompt = f"<|im_start|>system\n{system_instruction}<|im_end|>\n<|im_start|>user\n{user_query}\nLet's think step by step.<|im_end|>\n<|im_start|>assistant\n"
             
-            # do_sample=False ensures the math is consistent (Temperature 0)
-            result = pipe(prompt, max_new_tokens=600, do_sample=False)
+            # Increased max_new_tokens for longer, detailed proofs
+            result = pipe(prompt, max_new_tokens=900, temperature=0.1, do_sample=True, top_p=0.9)
             response = result[0]['generated_text'].split("<|im_start|>assistant\n")[-1]
             
             st.markdown(response)
+            st.download_button("📩 Download Proof", response, file_name="math_solution.txt")
 
-st.sidebar.info("Model: SmolLM2-1.7B | Mode: Exact Logic")
+st.sidebar.markdown("### 🎓 Academic Modules")
+st.sidebar.info("• Abstract Algebra\n• Real Analysis\n• Number Theory\n• Complex Variables")