Upload app.py

app.py

@@ -7,6 +7,7 @@
 import streamlit as st
 import os
 import requests
+import json as _json
 import sympy as sp
 from sympy import (
     symbols, diff, integrate, limit, solve,
@@ -192,10 +193,111 @@ hr { border-color: #1a1a1a !important; }
 """, unsafe_allow_html=True)
 
 # ── Session state ────────────────────────────────────────────────────
+import datetime as _dt
+
+# ── Session state ────────────────────────────────────────────────
 if "messages" not in st.session_state:
     st.session_state.messages = []
 if "last_submitted" not in st.session_state:
     st.session_state.last_submitted = ""
+if "chats" not in st.session_state:
+    # Load from Supabase on first load
+    st.session_state.chats = supa_load_all_chats()
+if "current_chat_id" not in st.session_state:
+    st.session_state.current_chat_id = None
+
+# ════════════════════════════════════════════════════════════════════
+# SUPABASE — Persistent chat history
+# ════════════════════════════════════════════════════════════════════
+_SUPA_URL = os.environ.get("SUPABASE_URL", "")
+_SUPA_KEY = os.environ.get("SUPABASE_KEY", "")
+
+def _supa_headers():
+    return {
+        "apikey": _SUPA_KEY,
+        "Authorization": f"Bearer {_SUPA_KEY}",
+        "Content-Type": "application/json",
+        "Prefer": "return=minimal"
+    }
+
+def supa_load_all_chats():
+    """Load all chats from Supabase."""
+    if not _SUPA_URL or not _SUPA_KEY:
+        return {}
+    try:
+        resp = requests.get(
+            f"{_SUPA_URL}/rest/v1/chats?select=*&order=created_at.desc",
+            headers=_supa_headers(), timeout=5
+        )
+        if resp.status_code == 200:
+            rows = resp.json()
+            return {r["id"]: {
+                "title": r["title"],
+                "messages": r["messages"],
+                "created": r["created_at"][:16].replace("T"," ")
+            } for r in rows}
+    except Exception:
+        pass
+    return {}
+
+def supa_save_chat(chat_id, title, messages):
+    """Save or update a chat in Supabase."""
+    if not _SUPA_URL or not _SUPA_KEY or not messages:
+        return
+    try:
+        requests.post(
+            f"{_SUPA_URL}/rest/v1/chats",
+            headers={**_supa_headers(), "Prefer": "resolution=merge-duplicates"},
+            data=_json.dumps({
+                "id": chat_id,
+                "title": title,
+                "messages": messages
+            }), timeout=5
+        )
+    except Exception:
+        pass
+
+def supa_delete_chat(chat_id):
+    """Delete a chat from Supabase."""
+    if not _SUPA_URL or not _SUPA_KEY:
+        return
+    try:
+        requests.delete(
+            f"{_SUPA_URL}/rest/v1/chats?id=eq.{chat_id}",
+            headers=_supa_headers(), timeout=5
+        )
+    except Exception:
+        pass
+
+def new_chat():
+    """Start a fresh chat session."""
+    chat_id = f"chat_{_dt.datetime.now().strftime('%Y%m%d_%H%M%S')}"
+    st.session_state.current_chat_id = chat_id
+    st.session_state.messages = []
+    st.session_state.last_submitted = ""
+
+def save_current_chat():
+    """Save current messages to session state and Supabase."""
+    cid = st.session_state.current_chat_id
+    if not cid or not st.session_state.messages:
+        return
+    # Auto-title from first user message
+    first_user = next((m["content"] for m in st.session_state.messages if m["role"]=="user"), "New Chat")
+    title = first_user[:35] + "..." if len(first_user) > 35 else first_user
+    st.session_state.chats[cid] = {
+        "title": title,
+        "messages": list(st.session_state.messages),
+        "created": _dt.datetime.now().strftime("%d %b %H:%M")
+    }
+    # Save to Supabase (persistent)
+    supa_save_chat(cid, title, list(st.session_state.messages))
+
+def load_chat(chat_id):
+    """Load a previous chat."""
+    if chat_id in st.session_state.chats:
+        st.session_state.current_chat_id = chat_id
+        st.session_state.messages = list(st.session_state.chats[chat_id]["messages"])
+        st.session_state.last_submitted = ""
 
 
 # ════════════════════════════════════════════════════════════════════
@@ -1635,12 +1737,16 @@ def ask_ai_streaming(problem: str, sympy_info: dict, history: list) -> str:
     # Get full response first
     full_response = ask_ai(problem, sympy_info, history)
 
-    # Stream it word by word
+    # Stream chunk by chunk (3-4 words at a time) — natural reading pace
     def word_generator():
         words = full_response.split(" ")
-        for i, word in enumerate(words):
-            yield word + (" " if i < len(words)-1 else "")
-            time.sleep(0.015)  # 15ms per word — smooth streaming
+        chunk_size = 3  # 3 words at a time
+        for i in range(0, len(words), chunk_size):
+            chunk = " ".join(words[i:i+chunk_size])
+            if i + chunk_size < len(words):
+                chunk += " "
+            yield chunk
+            time.sleep(0.08)  # 80ms per chunk — comfortable reading pace
 
     # Use st.write_stream for streaming display
     streamed = st.write_stream(word_generator())
@@ -1651,10 +1757,46 @@ def ask_ai_streaming(problem: str, sympy_info: dict, history: list) -> str:
 # SIDEBAR
 # ════════════════════════════════════════════════════════════════════
 with st.sidebar:
-    st.markdown("### 📐 Saad.AI")
+    st.markdown("### 🧠 Saad.AI")
     st.caption("BSc Mathematics Engine")
     st.divider()
 
+    # ── New Chat Button ──────────────────────────────────────────
+    if st.button("➕  New Chat", use_container_width=True):
+        save_current_chat()
+        new_chat()
+        st.rerun()
+
+    st.divider()
+
+    # ── Chat History ─────────────────────────────────────────────
+    if st.session_state.chats:
+        st.markdown("**💬 Chat History**")
+        # Show most recent first
+        sorted_chats = sorted(
+            st.session_state.chats.items(),
+            key=lambda x: x[1]["created"],
+            reverse=True
+        )
+        for chat_id, chat_data in sorted_chats:
+            col1, col2 = st.columns([4,1])
+            with col1:
+                # Highlight current chat
+                is_current = chat_id == st.session_state.current_chat_id
+                label = ("▶ " if is_current else "") + chat_data["title"]
+                if st.button(label, key=f"load_{chat_id}", use_container_width=True):
+                    save_current_chat()
+                    load_chat(chat_id)
+                    st.rerun()
+            with col2:
+                if st.button("🗑", key=f"del_{chat_id}"):
+                    del st.session_state.chats[chat_id]
+                    supa_delete_chat(chat_id)  # delete from Supabase too
+                    if chat_id == st.session_state.current_chat_id:
+                        new_chat()
+                    st.rerun()
+        st.divider()
+
     st.markdown("**🎯 Topics**")
     topics = [
         "📈 Calculus", "🔢 Linear Algebra", "🔁 Number Theory",
@@ -1735,10 +1877,18 @@ if not st.session_state.messages:
     """, unsafe_allow_html=True)
 
 # Render chat history
-for msg in st.session_state.messages:
+for i, msg in enumerate(st.session_state.messages):
     avatar = "🧑‍🎓" if msg["role"] == "user" else "📐"
     with st.chat_message(msg["role"], avatar=avatar):
         st.markdown(msg["content"])
+        # Copy button on AI responses
+        if msg["role"] == "assistant":
+            if st.button("📋 Copy", key=f"copy_{i}", help="Copy response"):
+                st.write(
+                    f'<script>navigator.clipboard.writeText({repr(msg["content"])})</script>',
+                    unsafe_allow_html=True
+                )
+                st.toast("✅ Copied!", icon="📋")
 
 # ════════════════════════════════════════════════════════════════════
 # INPUT — use st.chat_input (cleaner than text_input + button)
@@ -1782,3 +1932,7 @@ if problem and problem != st.session_state.last_submitted:
     # Save to history
     st.session_state.messages.append({"role": "user", "content": problem})
     st.session_state.messages.append({"role": "assistant", "content": answer})
+    # Auto-save to chat history
+    if not st.session_state.current_chat_id:
+        new_chat()
+    save_current_chat()