Update app.py

app.py

@@ -1,75 +1,90 @@
 import streamlit as st
 import requests
 
-# 1. Page Configuration
-st.set_page_config(page_title="SAAD AI ACADEMY: Math Engine", page_icon="📐", layout="centered")
+# STEP 1: PAGE CONFIG
+st.set_page_config(page_title="SAAD AI ACADEMY", page_icon="📐", layout="centered")
 
-# SECURE TOKEN ACCESS (This pulls from the Settings/Secrets you just saved)
+# STEP 2: SECURE TOKEN ACCESS
 try:
     HF_TOKEN = st.secrets["HF_TOKEN"]
 except:
-    st.error("Error: HF_TOKEN not found in Space Secrets. Please add it in Settings.")
+    st.error("Error: HF_TOKEN not found in Space Secrets.")
     st.stop()
 
 API_URL = "https://api-inference.huggingface.co/models/HuggingFaceTB/SmolLM2-1.7B-Instruct"
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
+# STEP 3: UI UPGRADE (Dark/Neon Glassmorphism)
 st.markdown("""
     <style>
-    .stApp { background-color: #f0f2f6; }
-    .stChatMessage { border-radius: 15px; border: 1px solid #d1d5db; box-shadow: 2px 2px 5px rgba(0,0,0,0.05); }
+    .stApp {
+        background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%);
+        color: #ffffff;
+    }
+    .stChatMessage {
+        background: rgba(255, 255, 255, 0.05) !important;
+        backdrop-filter: blur(10px);
+        border-radius: 15px !important;
+        border: 1px solid rgba(255, 255, 255, 0.1) !important;
+        margin-bottom: 15px;
+    }
+    section[data-testid="stSidebar"] {
+        background-color: #0f172a !important;
+    }
+    h1, h2, h3, p, span {
+        color: #ffffff !important;
+    }
+    .stCaption {
+        color: #3b82f6 !important;
+    }
     </style>
     """, unsafe_allow_html=True)
 
+# STEP 4: APP TITLES
 st.title("📐 SAAD AI ACADEMY")
-st.caption("High-Speed Mathematics Engine | SUST Department of Mathematics")
+st.caption("Universal B.Sc. Mathematics Engine | SUST Department of Mathematics")
 
-# 2. MASTER LOGIC PROMPT
+# STEP 5: UNIVERSAL LOGIC ENGINE (The New Instructions)
+# These rules apply to ANY math problem you type.
 system_instruction = (
-    "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 before applying it.\n\n"
-    "RULE 4 (FORMAT): Use LaTeX ($...$ or $$...$$) for all math. Never skip algebraic steps."
+    "You are a Senior Mathematics Professor. You must solve every problem with absolute rigor.\n\n"
+    "RULE 1 (DEFINITION): Before solving, explicitly state the mathematical domain and the key theorem or definition involved (e.g., 'In Abstract Algebra, we use the First Isomorphism Theorem...').\n"
+    "RULE 2 (VALIDATION): Check for necessary conditions or constraints before performing operations (e.g., check if a group is finite, if a function is continuous, or if a divisor is non-zero).\n"
+    "RULE 3 (DERIVATION): Show every algebraic step clearly. Do not skip intermediate simplifications. Explain the logic of each transition.\n"
+    "RULE 4 (FORMATTING): Use LaTeX ($...$ or $$...$$) for all mathematical expressions. Use bold text for final results."
 )
 
 def query_api(prompt_text):
-    full_prompt = f"<|im_start|>system\n{system_instruction}<|im_end|>\n<|im_start|>user\n{prompt_text}\nLet's think step by step.<|im_end|>\n<|im_start|>assistant\n"
-    
+    full_prompt = f"<|im_start|>system\n{system_instruction}<|im_end|>\n<|im_start|>user\n{prompt_text}\nLet's derive the solution step-by-step.<|im_end|>\n<|im_start|>assistant\n"
     payload = {
         "inputs": full_prompt,
-        "parameters": {
-            "max_new_tokens": 800,
-            "temperature": 0.1,
-            "return_full_text": False
-        }
+        "parameters": {"max_new_tokens": 1000, "temperature": 0.1, "return_full_text": False},
+        "options": {"wait_for_model": True}
     }
-    
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
-# 3. User Interface
-user_query = st.chat_input("Enter your problem (e.g., Solve 14x ≡ 30 mod 44)")
+# STEP 6: CHAT INTERFACE
+user_query = st.chat_input("Ask a B.Sc. level math question (Analysis, Algebra, Calculus...)")
 
 if user_query:
     with st.chat_message("user"):
         st.write(user_query)
 
     with st.chat_message("assistant"):
-        with st.spinner("🚀 Cloud API is solving..."):
-            try:
-                data = query_api(user_query)
-                
-                if isinstance(data, dict) and "error" in data:
-                    st.warning("Model is starting up. Please wait 10-20 seconds and try again.")
-                else:
-                    response = data[0]['generated_text']
-                    st.markdown(response)
-                    st.download_button("📩 Download Proof", response, file_name="math_solution.txt")
-            except Exception as e:
-                st.error("Connection error. Ensure your HF_TOKEN is correctly set in Secrets.")
+        with st.spinner("🚀 Cloud API is analyzing mathematical logic..."):
+            data = query_api(user_query)
+            if isinstance(data, list) and len(data) > 0:
+                response = data[0]['generated_text']
+                st.markdown(response)
+                st.download_button("📩 Download Proof", response, file_name="math_solution.txt")
+            else:
+                st.error("API error or timeout. Please check your internet or try again in 10 seconds.")
 
-st.sidebar.markdown(f"### 🎓 Student Profile\n**Name:** Almuyed Saad\n**Inst:** SUST\n**Major:** Mathematics")
-st.sidebar.info("Module: B.Sc. Math Engine V5 (API Edition)")
+# STEP 7: SIDEBAR
+st.sidebar.markdown(f"### 🎓 Student Profile")
+st.sidebar.write("**Name:** Almuyed Saad")
+st.sidebar.write("**Inst:** SUST")
+st.sidebar.write("**Major:** Mathematics")
+st.sidebar.divider()
+st.sidebar.info("Module: Universal Logic Engine V5.1")