Upload app.py

app.py

@@ -2179,6 +2179,22 @@ def ask_gemini_vision(image_b64: str, mime_type: str, user_note: str) -> str:
     if not any(k for k in gemini_keys if k.strip()):
         return "⚠️ **No Gemini API key found.** Add keys in HF Space → Settings → Secrets."
 
+    prompt = (
+        "You are Saad.AI, a BSc Mathematics assistant built by Saad.\n"
+        "The student has uploaded an image containing a math problem.\n\n"
+        "YOUR TASKS:\n"
+        "1. Read and extract the math problem from the image exactly.\n"
+        "2. State what the problem is clearly.\n"
+        "3. Solve it step by step using the full math structure:\n"
+        "   🔍 **Given:** ...\n"
+        "   📌 **Method:** ...\n"
+        "   🧮 **Step 1:** ...\n"
+        "   ✅ **Final Answer:** $$\\boxed{answer}$$\n"
+        "4. ALL math must be in LaTeX — never plain text math.\n"
+        "5. If the image is unclear or not math-related, say so politely.\n\n"
+        f"Additional note from student: {user_note if user_note else 'None'}"
+    )
+
     last_error = ""
     for i, key in enumerate(gemini_keys):
         if not key.strip():
@@ -2233,22 +2249,27 @@ def handle_uploaded_file(uploaded_file, user_note: str) -> str:
     """
     import base64
 
-    # ── Validate file ────────────────────────────────────────────────
+    # ── Validate size ────────────────────────────────────────────────
     MAX_SIZE = 5 * 1024 * 1024  # 5MB
     file_bytes = uploaded_file.read()
-
+    if len(file_bytes) == 0:
+        return "⚠️ The uploaded file is empty. Please try again."
     if len(file_bytes) > MAX_SIZE:
-        return "⚠️ File too large. Please upload a file under 5MB."
+        return f"⚠️ File too large ({len(file_bytes)//1024}KB). Please upload under 5MB."
 
+    # ── Detect MIME type from Streamlit's type field, not filename ────
+    # This works even if filename has spaces, brackets, or no extension
     mime_map = {
         "jpg": "image/jpeg", "jpeg": "image/jpeg",
-        "png": "image/png", "webp": "image/webp",
+        "png": "image/png",  "webp": "image/webp",
         "pdf": "application/pdf"
     }
-    ext = uploaded_file.name.split(".")[-1].lower()
-    mime_type = mime_map.get(ext)
-
+    # Try Streamlit's type first (most reliable), fall back to extension
+    mime_type = uploaded_file.type if uploaded_file.type else None
     if not mime_type:
+        ext = uploaded_file.name.rsplit(".", 1)[-1].lower() if "." in uploaded_file.name else ""
+        mime_type = mime_map.get(ext)
+    if mime_type not in mime_map.values():
         return "⚠️ Unsupported format. Please upload JPG, PNG, WEBP or PDF."
 
     # ── Convert to base64 ────────────────────────────────────────────
@@ -2261,24 +2282,39 @@ def handle_uploaded_file(uploaded_file, user_note: str) -> str:
     if gemini_response.startswith("⚠️"):
         return gemini_response
 
-    # ── Try SymPy verification on extracted text ─────────────────────
-    # Gemini response contains the problem — try to run SymPy on it
-    sympy_result = run_sympy(gemini_response)
+    # ── SymPy verification — extract problem line first ───────────────
+    # Run SymPy on the first user-question line, not Gemini's full markdown
+    # This avoids SymPy choking on LaTeX formatting in the solution
+    extracted_problem = ""
+    for line in gemini_response.splitlines():
+        stripped = line.strip()
+        # Skip empty lines, headers, and Gemini's own solution steps
+        if (stripped and
+                not stripped.startswith("#") and
+                not stripped.startswith("🔍") and
+                not stripped.startswith("📌") and
+                not stripped.startswith("🧮") and
+                not stripped.startswith("✅") and
+                not stripped.startswith("**") and
+                len(stripped) > 5):
+            extracted_problem = stripped
+            break
+    sympy_result = run_sympy(extracted_problem) if extracted_problem else {"type": "general", "result": None, "latex": ""}
 
     if (sympy_result.get("result") and
             sympy_result["result"] not in (None, "matrix_detected", "mod_detected")):
-        # SymPy verified — strip AI answer, append verified one
         latex = sympy_result.get("latex", "")
         cleaned = re.sub(
             r'(✅\s*\*{0,2}Final\s*Answer\*{0,2}.*|✅[^\n]*$)',
             "", gemini_response,
             flags=re.DOTALL | re.IGNORECASE
         ).rstrip()
-        verified_note = "\n\n🔒 **SymPy Verified**"
-        return cleaned + verified_note + f"\n\n✅ **Final Answer:** $$\\boxed{{{latex}}}$$"
+        return cleaned + "\n\n🔒 **SymPy Verified**" + f"\n\n✅ **Final Answer:** $$\\boxed{{{latex}}}$$"
     else:
-        # No SymPy verification — return Gemini answer with warning
+        # Gemini answered, SymPy couldn't verify — single clean note
         return gemini_response + "\n\n⚠️ *AI-generated answer — not SymPy verified.*"
+
+
 def ask_ai_streaming(problem: str, sympy_info: dict, history: list) -> str:
     """Get AI response and stream it word by word using st.write_stream."""
     import time