Upload app.py

app.py

@@ -291,6 +291,57 @@ hr { border-color: #1a1a2e !important; }
 ::-webkit-scrollbar-track { background: #0f0f0f; }
 ::-webkit-scrollbar-thumb { background: #1f2937; border-radius: 2px; }
 ::-webkit-scrollbar-thumb:hover { background: #3b82f6; }
+
+/* ── File uploader — compact, attached to input bar ── */
+[data-testid="stFileUploader"] {
+    background: #111827 !important;
+    border: 1px solid #1f2937 !important;
+    border-bottom: none !important;
+    border-radius: 14px 14px 0 0 !important;
+    padding: 0.35rem 1rem !important;
+    transition: all 0.2s ease !important;
+}
+[data-testid="stFileUploader"]:hover {
+    border-color: #3b82f6 !important;
+    background: #0f172a !important;
+}
+[data-testid="stFileUploader"] section {
+    padding: 0 !important;
+    border: none !important;
+    background: transparent !important;
+}
+[data-testid="stFileUploader"] label {
+    font-size: 0.78rem !important;
+    color: #4b5563 !important;
+    transition: color 0.2s !important;
+}
+[data-testid="stFileUploader"]:hover label {
+    color: #60a5fa !important;
+}
+[data-testid="stFileUploaderDropzone"] {
+    background: transparent !important;
+    border: none !important;
+    padding: 0.1rem 0 !important;
+    min-height: 0 !important;
+    transition: all 0.2s !important;
+}
+[data-testid="stFileUploaderDropzone"]:hover {
+    background: rgba(59,130,246,0.05) !important;
+    border-radius: 8px !important;
+}
+[data-testid="stFileUploaderDropzoneInstructions"] {
+    font-size: 0.75rem !important;
+    color: #4b5563 !important;
+    padding: 0.2rem 0 !important;
+}
+/* Input bar connects seamlessly below uploader */
+[data-testid="stChatInput"] {
+    border-radius: 0 0 14px 14px !important;
+    border-top: 1px solid #1f2937 !important;
+}
+[data-testid="stChatInput"]:focus-within {
+    border-color: #3b82f6 !important;
+}
 </style>
 """, unsafe_allow_html=True)
 
@@ -2108,8 +2159,113 @@ def ask_ai(problem: str, sympy_info: dict, history: list) -> str:
 
 
 # ════════════════════════════════════════════════════════════════════
-# STREAMING WRAPPER — shows text word by word like ChatGPT
+# FILE UPLOAD — image/PDF sent to Gemini Vision, then SymPy verified
 # ════════════════════════════════════════════════════════════════════
+def ask_gemini_vision(image_b64: str, mime_type: str, user_note: str) -> str:
+    """
+    Send image or PDF to Gemini 1.5 Flash vision.
+    Returns extracted problem text + solution from Gemini.
+    Only Gemini supports vision — Groq and OpenRouter skipped.
+    """
+    gemini_keys = [
+        os.environ.get("GEMINI_API_KEY_1", ""),
+        os.environ.get("GEMINI_API_KEY_2", ""),
+    ]
+
+    prompt = (
+        "You are Saad.AI, a BSc Mathematics assistant built by Saad.\n"
+        "The student has uploaded an image or document containing a math problem.\n\n"
+        "YOUR TASKS:\n"
+        "1. Read and extract the math problem from the image/document 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'}"
+    )
+
+    for key in gemini_keys:
+        if not key:
+            continue
+        try:
+            resp = requests.post(
+                f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={key}",
+                headers={"Content-Type": "application/json"},
+                json={
+                    "contents": [{
+                        "parts": [
+                            {"inline_data": {"mime_type": mime_type, "data": image_b64}},
+                            {"text": prompt}
+                        ]
+                    }],
+                    "generationConfig": {"maxOutputTokens": 2048, "temperature": 0.15}
+                },
+                timeout=60
+            )
+            resp.raise_for_status()
+            return resp.json()["candidates"][0]["content"]["parts"][0]["text"]
+        except Exception:
+            continue
+
+    return "⚠️ Could not process the file. Please check your Gemini API keys or try again."
+
+
+def handle_uploaded_file(uploaded_file, user_note: str) -> str:
+    """
+    Process uploaded image or PDF:
+    1. Convert to base64
+    2. Send to Gemini Vision
+    3. Try SymPy verification on extracted text
+    4. Return final answer
+    """
+    import base64
+
+    # ── Validate file ────────────────────────────────────────────────
+    MAX_SIZE = 5 * 1024 * 1024  # 5MB
+    file_bytes = uploaded_file.read()
+
+    if len(file_bytes) > MAX_SIZE:
+        return "⚠️ File too large. Please upload a file under 5MB."
+
+    mime_map = {
+        "jpg": "image/jpeg", "jpeg": "image/jpeg",
+        "png": "image/png", "webp": "image/webp",
+        "pdf": "application/pdf"
+    }
+    ext = uploaded_file.name.split(".")[-1].lower()
+    mime_type = mime_map.get(ext)
+
+    if not mime_type:
+        return "⚠️ Unsupported format. Please upload JPG, PNG, WEBP or PDF."
+
+    # ── Convert to base64 ────────────────────────────────────────────
+    image_b64 = base64.b64encode(file_bytes).decode("utf-8")
+
+    # ── Send to Gemini Vision ────────────────────────────────────────
+    gemini_response = ask_gemini_vision(image_b64, mime_type, user_note)
+
+    # ── Try SymPy verification on extracted text ─────────────────────
+    # Gemini response contains the problem — try to run SymPy on it
+    sympy_result = run_sympy(gemini_response)
+
+    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}}}$$"
+    else:
+        # No SymPy verification — return Gemini answer with warning
+        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
@@ -2409,6 +2565,42 @@ for i, msg in enumerate(st.session_state.messages):
             with st.expander("📋 Copy"):
                 st.code(msg["content"], language=None)
 
+# ════════════════════════════════════════════════════════════════════
+# FILE UPLOAD — above chat input
+# ════════════════════════════════════════════════════════════════════
+uploaded_file = st.file_uploader(
+    "📎 Upload a math problem (image or PDF)",
+    type=["jpg", "jpeg", "png", "webp", "pdf"],
+    label_visibility="collapsed",
+    help="Upload a photo or PDF of your math problem — Saad.AI will read and solve it"
+)
+
+if uploaded_file and uploaded_file.name != st.session_state.get("last_uploaded_file", ""):
+    st.session_state["last_uploaded_file"] = uploaded_file.name
+    user_note = ""
+
+    # Show user message
+    with st.chat_message("user", avatar="🧑‍🎓"):
+        st.markdown(f"📎 **Uploaded:** `{uploaded_file.name}`")
+
+    # Process file
+    with st.chat_message("assistant", avatar="📐"):
+        with st.spinner("📖 Reading your file..."):
+            answer = handle_uploaded_file(uploaded_file, user_note)
+        st.markdown(answer)
+        with st.expander("📋 Copy"):
+            st.code(answer, language=None)
+
+    # Save to history
+    if not st.session_state.current_chat_id:
+        new_chat()
+    st.session_state.messages.append({
+        "role": "user",
+        "content": f"📎 Uploaded file: `{uploaded_file.name}`"
+    })
+    st.session_state.messages.append({"role": "assistant", "content": answer})
+    save_current_chat()
+
 # ════════════════════════════════════════════════════════════════════
 # INPUT — use st.chat_input (cleaner than text_input + button)
 # ════════════════════════════════════════════════════════════════════