From 3a759b14c4d3bed55f8be358ac46761dc6b285b1 Mon Sep 17 00:00:00 2001
From: KingSagnik <sagnikd587@gmail.com>
Date: Sat, 26 Jul 2025 00:12:22 +0530
Subject: [PATCH 1/4] Adding basic db schema

---
 TalkHeal.py                  |  56 +++++++++++-
 components/chat_interface.py | 160 ++++++++---------------------------
 core/db.py                   |  87 +++++++++++++++++++
 talkheal.db                  | Bin 0 -> 28672 bytes
 4 files changed, 173 insertions(+), 130 deletions(-)
 create mode 100644 core/db.py
 create mode 100644 talkheal.db

diff --git a/TalkHeal.py b/TalkHeal.py
index ab1d210..53dae0b 100644
--- a/TalkHeal.py
+++ b/TalkHeal.py
@@ -1,13 +1,17 @@
 import streamlit as st
 import google.generativeai as genai
-from core.utils import save_conversations, load_conversations
+from core.db import init_db, create_user, get_user_by_username, create_chat, add_message, get_messages
+
+# --- ENSURE TABLES EXIST BEFORE ANY DB USAGE ---
+init_db()
 
+from core.utils import save_conversations, load_conversations
 from core.config import configure_gemini, PAGE_CONFIG
 from core.utils import get_current_time, create_new_conversation
 from css.styles import apply_custom_css
 from components.header import render_header
 from components.sidebar import render_sidebar
-from components.chat_interface import render_chat_interface, handle_chat_input
+from components.chat_interface import render_chat_interface
 from components.emergency_page import render_emergency_page
 
 # --- 1. INITIALIZE SESSION STATE ---
@@ -28,6 +32,10 @@
         "Substance Use Disorders", "ADHD & Neurodevelopmental", "Personality Disorders",
         "Sleep Disorders"
     ]
+if "user_id" not in st.session_state:
+    st.session_state.user_id = None
+if "active_chat_id" not in st.session_state:
+    st.session_state.active_chat_id = None
 
 # --- 2. SET PAGE CONFIG ---
 st.set_page_config(
@@ -65,8 +73,48 @@
     # Render the standard chat interface
     with main_area:
         render_header()
-        render_chat_interface()
-        handle_chat_input(model)
+        render_chat_interface(model)
+
+        # User login/register
+        from core.db import create_user, get_user_by_username
+
+        if "user_id" not in st.session_state or st.session_state.user_id is None:
+            st.title("Login / Register")
+            username = st.text_input("Username")
+            email = st.text_input("Email")
+            if st.button("Register/Login"):
+                user = get_user_by_username(username)
+                # Add this block to check for existing email
+                from core.db import get_user_by_email
+                existing_email_user = get_user_by_email(email)
+                if existing_email_user and (not user or existing_email_user["id"] != (user["id"] if user else None)):
+                    st.error("This email is already registered. Please use another email or log in with the existing username.")
+                else:
+                    if not user:
+                        create_user(username, email)
+                        user = get_user_by_username(username)
+                    st.session_state.user_id = user["id"]
+                    st.success(f"Logged in as {username}")
+                    st.rerun()
+            st.stop()
+
+        # Start new chat
+        if st.button("➕ New Chat"):
+            chat_id = create_chat(st.session_state.user_id, "Untitled Chat")
+            st.session_state.active_chat_id = chat_id
+            st.experimental_rerun()
+
+        # Send message
+        user_input = st.text_input("Message", key="message_input")
+        if st.button("Send") and user_input.strip() and st.session_state.get("active_chat_id"):
+            add_message(st.session_state.active_chat_id, "user", user_input.strip())
+            st.experimental_rerun()
+
+        # Display chat history
+        if st.session_state.get("active_chat_id"):
+            messages = get_messages(st.session_state.active_chat_id)
+            for msg in messages:
+                st.write(f"{msg['sender']}: {msg['message']}")
 
 st.markdown("""
 <script>
diff --git a/components/chat_interface.py b/components/chat_interface.py
index 8581d58..e7778a9 100644
--- a/components/chat_interface.py
+++ b/components/chat_interface.py
@@ -1,138 +1,46 @@
 import streamlit as st
+from core.db import get_messages, add_message
 from core.utils import get_current_time, get_ai_response
 
-import streamlit.components.v1 as components
-import streamlit as st
-from datetime import datetime, timedelta
-import json
-from core.utils import save_conversations, load_conversations
-
-def set_user_time_in_session():
-    if "user_time_offset" not in st.session_state:
-        # Embed JS to send local timezone offset in minutes
-        components.html("""
-            <script>
-            const offset = new Date().getTimezoneOffset(); // in minutes
-            const time = new Date().toLocaleString();      // full readable local time
-            const data = {offset: offset, time: time};
-            window.parent.postMessage({type: 'USER_TIME', data: data}, '*');
-            </script>
-        """, height=0)
+def render_chat_interface(model):
+    chat_id = st.session_state.get("active_chat_id")
+    if not chat_id:
+        st.info("Start a new chat to begin.")
+        return
 
+    messages = get_messages(chat_id)
+    if not messages:
         st.markdown("""
-        <script>
-        window.addEventListener("message", (event) => {
-            if (event.data.type === "USER_TIME") {
-                const payload = JSON.stringify(event.data.data);
-                fetch("/", {
-                    method: "POST",
-                    headers: {"Content-Type": "application/json"},
-                    body: payload
-                }).then(() => location.reload());
-            }
-        });
-        </script>
+        <div class="welcome-message">
+            <strong>Hello! I'm TalkHeal, your mental health companion</strong><br>
+            How are you feeling today? You can write it down below or for a fresh start click on "➕ New Chat" 😊
+        </div>
         """, unsafe_allow_html=True)
 
-    else:
-        # Already set, do nothing
-        pass
-
-set_user_time_in_session()
-
-def render_chat_interface():    
-    if st.session_state.active_conversation >= 0:
-        active_convo = st.session_state.conversations[st.session_state.active_conversation]
-        
-        # Show welcome message if no messages exist
-        if not active_convo["messages"]:
+    for msg in messages:
+        if msg["sender"] == "user":
             st.markdown(f"""
-            <div class="welcome-message">
-                <strong>Hello! I'm TalkHeal, your mental health companion</strong><br>
-                How are you feeling today? You can write it down below or for a fresh start click on "➕ New Chat" on the left and choose a common topic😊               <div class="message-time">{get_current_time()}</div>
+            <div class="user-message">
+                {msg["message"]}
+                <div class="message-time">{msg["timestamp"]}</div>
+            </div>
+            """, unsafe_allow_html=True)
+        else:
+            st.markdown(f"""
+            <div class="bot-message">
+                {msg["message"]}
+                <div class="message-time">{msg["timestamp"]}</div>
             </div>
             """, unsafe_allow_html=True)
-        
-        # Render all messages directly without any wrapper container
-        for msg in active_convo["messages"]:
-            if msg["sender"] == "user":
-                st.markdown(f"""
-                <div class="user-message">
-                    {msg["message"]}
-                    <div class="message-time">{msg["time"]}</div>
-                </div>
-                """, unsafe_allow_html=True)
-            else:
-                st.markdown(f"""
-                <div class="bot-message">
-                    {msg["message"]}
-                    <div class="message-time">{msg["time"]}</div>
-                </div>
-                """, unsafe_allow_html=True)
-
-def handle_chat_input(model):   
-    if "pre_filled_chat_input" not in st.session_state:
-        st.session_state.pre_filled_chat_input = ""
-    initial_chat_value = st.session_state.pre_filled_chat_input
-    st.session_state.pre_filled_chat_input = "" 
 
     with st.form(key="chat_form", clear_on_submit=True):
-        col1, col2 = st.columns([5, 1])
-        
-        with col1:
-            user_input = st.text_input(
-                "Share your thoughts...", 
-                key="message_input", 
-                label_visibility="collapsed",
-                placeholder="Type your message here...",
-                value=initial_chat_value
-            )
-        
-        with col2:
-            send_pressed = st.form_submit_button("Send", use_container_width=True)
-    
-    if (send_pressed or st.session_state.get('send_chat_message', False)) and user_input.strip():
-        if 'send_chat_message' in st.session_state:
-            st.session_state.send_chat_message = False
-
-        if st.session_state.active_conversation >= 0:
-            current_time = get_current_time()
-            active_convo = st.session_state.conversations[st.session_state.active_conversation]
-            active_convo["messages"].append({
-                "sender": "user", 
-                "message": user_input.strip(), 
-                "time": current_time
-            })
-            save_conversations(st.session_state.conversations)
-            if len(active_convo["messages"]) == 1:
-                title = user_input[:30] + "..." if len(user_input) > 30 else user_input
-                active_convo["title"] = title
-            with st.spinner("TalkHeal is thinking..."):
-                try:
-                    def format_memory_for_prompt(convo_history, max_turns=10):
-                        context = ""
-                        for msg in convo_history[-max_turns*2:]:  # user+bot per turn
-                            sender = "User" if msg["sender"] == "user" else "Bot"
-                            context += f"{sender}: {msg['message']}\n"
-                        return context
-
-                    context = format_memory_for_prompt(active_convo["messages"])
-                    full_prompt = context + f"User: {user_input.strip()}\nBot:"
-
-                    ai_response = get_ai_response(full_prompt, model)
-                    
-                    active_convo["messages"].append({
-                        "sender": "bot", 
-                        "message": ai_response, 
-                        "time": get_current_time()
-                    })
-                    
-                except Exception as e:
-                    st.error(f"An error occurred: {e}")
-                    active_convo["messages"].append({
-                        "sender": "bot", 
-                        "message": "I apologise, but I'm having trouble responding right now. Please try again in a moment.", 
-                        "time": get_current_time()
-                    })
-                    save_conversations(st.session_state.conversations)
-            st.rerun()
\ No newline at end of file
+        user_input = st.text_input("Share your thoughts...", key="message_input", label_visibility="collapsed")
+        send_pressed = st.form_submit_button("Send")
+    if send_pressed and user_input.strip():
+        add_message(chat_id, "user", user_input.strip())
+        with st.spinner("TalkHeal is thinking..."):
+            context = "\n".join([f"{m['sender'].capitalize()}: {m['message']}" for m in get_messages(chat_id)])
+            prompt = context + f"\nUser: {user_input.strip()}\nBot:"
+            ai_response = get_ai_response(prompt, model)
+            add_message(chat_id, "bot", ai_response)
+        st.rerun()
\ No newline at end of file
diff --git a/core/db.py b/core/db.py
new file mode 100644
index 0000000..e388aef
--- /dev/null
+++ b/core/db.py
@@ -0,0 +1,87 @@
+import sqlite3
+from contextlib import contextmanager
+
+DB_PATH = "talkheal.db"
+
+@contextmanager
+def get_db():
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    try:
+        yield conn
+    finally:
+        conn.commit()
+        conn.close()
+
+def init_db():
+    with get_db() as db:
+        db.execute("""
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT UNIQUE,
+            email TEXT UNIQUE,
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+        )""")
+        db.execute("""
+        CREATE TABLE IF NOT EXISTS chats (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER,
+            title TEXT,
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY(user_id) REFERENCES users(id)
+        )""")
+        db.execute("""
+        CREATE TABLE IF NOT EXISTS messages (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            chat_id INTEGER,
+            sender TEXT,
+            message TEXT,
+            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY(chat_id) REFERENCES chats(id)
+        )""")
+
+def create_user(username, email):
+    with get_db() as db:
+        db.execute(
+            "INSERT INTO users (username, email) VALUES (?, ?)",
+            (username, email)
+        )
+
+def get_user_by_username(username):
+    with get_db() as db:
+        cur = db.execute(
+            "SELECT * FROM users WHERE username = ?",
+            (username,)
+        )
+        return cur.fetchone()
+
+def get_user_by_email(email):
+    with get_db() as db:
+        cur = db.execute(
+            "SELECT * FROM users WHERE email = ?",
+            (email,)
+        )
+        return cur.fetchone()
+
+def create_chat(user_id, title):
+    with get_db() as db:
+        cur = db.execute(
+            "INSERT INTO chats (user_id, title) VALUES (?, ?)",
+            (user_id, title)
+        )
+        return cur.lastrowid
+
+def add_message(chat_id, sender, message):
+    with get_db() as db:
+        db.execute(
+            "INSERT INTO messages (chat_id, sender, message) VALUES (?, ?, ?)",
+            (chat_id, sender, message)
+        )
+
+def get_messages(chat_id):
+    with get_db() as db:
+        cur = db.execute(
+            "SELECT * FROM messages WHERE chat_id = ? ORDER BY timestamp ASC",
+            (chat_id,)
+        )
+        return cur.fetchall()
\ No newline at end of file
diff --git a/talkheal.db b/talkheal.db
new file mode 100644
index 0000000000000000000000000000000000000000..0920560a573eee853112c619b18876ad3ab5ffa2
GIT binary patch
literal 28672
zcmeI)Z)?*)90%}AvbO86i_gLy3~mV8vOjd~VznTwcDGSnyDdw>;e(R3=L~B**JR)x
zI>mS38}NPj4ixs#cc8GxiVwY9Tes8>k%_qNdo-8aU4FUChfuf>ZgXSR^*MdI*J(LE
z&61l$QOF{tgpjD%Vq)tjO&pB$3u3RFIX>?&O6K0A68d)%y^$p2@AZ%RyYbz`6H!2d
z00bZa0SG_<0uX=z1pZrrqjWT8Oin7Vv%a(4<Sp)b&JOqb`&iLp1)EX3@NkvUel1Oo
z(KE?y&{D-_E6k$nR;gUDw&)|aMGIAXtyB>gD6@(^6N;!mbNt$ASh&&S?FR2qn>`+g
z>UBS)#dk%U@3fxNB{5pNRAzLEEf=b*HZ4{yOY~n0DmU#yc|F|O@|wj;E0tit$=;|^
z)MCp_1QppP4eGq4+erN!f9fa^i5Z5WygHO*@(h_pLs+uv!cqqw@0=^;yMB`o;iz}G
z<MT#M%=|9m(d#?p5!4NMtdB&E>1kz4YIyriG4HH-eE)#A>wKUX2?-4}CfiPn&xpv>
z%Wb7Ad`n(a_9-(p6fdj(ig3`gbfncvX`{-*&j@chZu9)+Khi#Zj8a<Ecwi{R6;Jz%
z9))w@@5ypa-dimrcOxnF_ovRwQ-ituXz*u~5SyyMCHe=EAwd8F5P$##AOHafKmY;|
zfB*y_@b3gnHA1GcS@DLg?e5YBKlH?w&YC%MaVL1kFjwDeWzrdQHl3Tzm~=juPtWJ)
z7cxqOOqji}Zg9iAmm3~)H@}d}FWes!ccto|iT+JwNDzPk1Rwwb2tWV=5P$##AOHaf
zTxNlBZAv+PouJ0GiFmizo&O&b{fmBlnQstdhX4d1009U<00Izz00bZa0SH_t0Yl3w
zL%tSJ4bAM$_`{zBbie;UA^KPS<T^RwULgPh2tWV=5P$##AOHafKmY<)RA5xqjJRBF
zBCcqLBG&;>gZaOre<I?C1OW&@00Izz00bZa0SG_<0uX?}B@?(U)&fxOWYub0F8<h{
qMX^BTFKY&vnPC1e7X3$p00bZa0SG_<0uX=z1Rwwb2teQp3;Y1fh+uvI

literal 0
HcmV?d00001


From ef6e1157eed6357bad2a536dbf1eccf364a95729 Mon Sep 17 00:00:00 2001
From: KingSagnik <sagnikd587@gmail.com>
Date: Sat, 26 Jul 2025 00:12:22 +0530
Subject: [PATCH 2/4] Adding basic db schema

---
 TalkHeal.py |   7 ++++-
 core/db.py  |  87 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 talkheal.db | Bin 0 -> 28672 bytes
 3 files changed, 93 insertions(+), 1 deletion(-)
 create mode 100644 core/db.py
 create mode 100644 talkheal.db

diff --git a/TalkHeal.py b/TalkHeal.py
index f99cfb6..78d6509 100644
--- a/TalkHeal.py
+++ b/TalkHeal.py
@@ -9,13 +9,18 @@
 )
 
 import google.generativeai as genai
+from core.db import init_db, create_user, get_user_by_username, create_chat, add_message, get_messages
+
+# --- ENSURE TABLES EXIST BEFORE ANY DB USAGE ---
+init_db()
+
 from core.utils import save_conversations, load_conversations
 from core.config import configure_gemini, PAGE_CONFIG
 from core.utils import get_current_time, create_new_conversation
 from css.styles import apply_custom_css
 from components.header import render_header
 from components.sidebar import render_sidebar
-from components.chat_interface import render_chat_interface, handle_chat_input
+from components.chat_interface import render_chat_interface
 from components.emergency_page import render_emergency_page
 
 
diff --git a/core/db.py b/core/db.py
new file mode 100644
index 0000000..e388aef
--- /dev/null
+++ b/core/db.py
@@ -0,0 +1,87 @@
+import sqlite3
+from contextlib import contextmanager
+
+DB_PATH = "talkheal.db"
+
+@contextmanager
+def get_db():
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    try:
+        yield conn
+    finally:
+        conn.commit()
+        conn.close()
+
+def init_db():
+    with get_db() as db:
+        db.execute("""
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT UNIQUE,
+            email TEXT UNIQUE,
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+        )""")
+        db.execute("""
+        CREATE TABLE IF NOT EXISTS chats (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER,
+            title TEXT,
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY(user_id) REFERENCES users(id)
+        )""")
+        db.execute("""
+        CREATE TABLE IF NOT EXISTS messages (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            chat_id INTEGER,
+            sender TEXT,
+            message TEXT,
+            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY(chat_id) REFERENCES chats(id)
+        )""")
+
+def create_user(username, email):
+    with get_db() as db:
+        db.execute(
+            "INSERT INTO users (username, email) VALUES (?, ?)",
+            (username, email)
+        )
+
+def get_user_by_username(username):
+    with get_db() as db:
+        cur = db.execute(
+            "SELECT * FROM users WHERE username = ?",
+            (username,)
+        )
+        return cur.fetchone()
+
+def get_user_by_email(email):
+    with get_db() as db:
+        cur = db.execute(
+            "SELECT * FROM users WHERE email = ?",
+            (email,)
+        )
+        return cur.fetchone()
+
+def create_chat(user_id, title):
+    with get_db() as db:
+        cur = db.execute(
+            "INSERT INTO chats (user_id, title) VALUES (?, ?)",
+            (user_id, title)
+        )
+        return cur.lastrowid
+
+def add_message(chat_id, sender, message):
+    with get_db() as db:
+        db.execute(
+            "INSERT INTO messages (chat_id, sender, message) VALUES (?, ?, ?)",
+            (chat_id, sender, message)
+        )
+
+def get_messages(chat_id):
+    with get_db() as db:
+        cur = db.execute(
+            "SELECT * FROM messages WHERE chat_id = ? ORDER BY timestamp ASC",
+            (chat_id,)
+        )
+        return cur.fetchall()
\ No newline at end of file
diff --git a/talkheal.db b/talkheal.db
new file mode 100644
index 0000000000000000000000000000000000000000..0920560a573eee853112c619b18876ad3ab5ffa2
GIT binary patch
literal 28672
zcmeI)Z)?*)90%}AvbO86i_gLy3~mV8vOjd~VznTwcDGSnyDdw>;e(R3=L~B**JR)x
zI>mS38}NPj4ixs#cc8GxiVwY9Tes8>k%_qNdo-8aU4FUChfuf>ZgXSR^*MdI*J(LE
z&61l$QOF{tgpjD%Vq)tjO&pB$3u3RFIX>?&O6K0A68d)%y^$p2@AZ%RyYbz`6H!2d
z00bZa0SG_<0uX=z1pZrrqjWT8Oin7Vv%a(4<Sp)b&JOqb`&iLp1)EX3@NkvUel1Oo
z(KE?y&{D-_E6k$nR;gUDw&)|aMGIAXtyB>gD6@(^6N;!mbNt$ASh&&S?FR2qn>`+g
z>UBS)#dk%U@3fxNB{5pNRAzLEEf=b*HZ4{yOY~n0DmU#yc|F|O@|wj;E0tit$=;|^
z)MCp_1QppP4eGq4+erN!f9fa^i5Z5WygHO*@(h_pLs+uv!cqqw@0=^;yMB`o;iz}G
z<MT#M%=|9m(d#?p5!4NMtdB&E>1kz4YIyriG4HH-eE)#A>wKUX2?-4}CfiPn&xpv>
z%Wb7Ad`n(a_9-(p6fdj(ig3`gbfncvX`{-*&j@chZu9)+Khi#Zj8a<Ecwi{R6;Jz%
z9))w@@5ypa-dimrcOxnF_ovRwQ-ituXz*u~5SyyMCHe=EAwd8F5P$##AOHafKmY;|
zfB*y_@b3gnHA1GcS@DLg?e5YBKlH?w&YC%MaVL1kFjwDeWzrdQHl3Tzm~=juPtWJ)
z7cxqOOqji}Zg9iAmm3~)H@}d}FWes!ccto|iT+JwNDzPk1Rwwb2tWV=5P$##AOHaf
zTxNlBZAv+PouJ0GiFmizo&O&b{fmBlnQstdhX4d1009U<00Izz00bZa0SH_t0Yl3w
zL%tSJ4bAM$_`{zBbie;UA^KPS<T^RwULgPh2tWV=5P$##AOHafKmY<)RA5xqjJRBF
zBCcqLBG&;>gZaOre<I?C1OW&@00Izz00bZa0SG_<0uX?}B@?(U)&fxOWYub0F8<h{
qMX^BTFKY&vnPC1e7X3$p00bZa0SG_<0uX=z1Rwwb2teQp3;Y1fh+uvI

literal 0
HcmV?d00001


From e915193bfb7bbe1dc9ffea2017a86e6ad3014e57 Mon Sep 17 00:00:00 2001
From: KingSagnik <sagnikd587@gmail.com>
Date: Sat, 26 Jul 2025 21:44:38 +0530
Subject: [PATCH 3/4] db setup

---
 TalkHeal.py                  | 109 ++++++++++++++++++++++++-----------
 components/chat_interface.py | 109 ++++++-----------------------------
 components/sidebar.py        |  60 ++-----------------
 talkheal.db                  | Bin 28672 -> 28672 bytes
 4 files changed, 98 insertions(+), 180 deletions(-)

diff --git a/TalkHeal.py b/TalkHeal.py
index 78d6509..56aff8e 100644
--- a/TalkHeal.py
+++ b/TalkHeal.py
@@ -9,41 +9,33 @@
 )
 
 import google.generativeai as genai
-from core.db import init_db, create_user, get_user_by_username, create_chat, add_message, get_messages
 
-# --- ENSURE TABLES EXIST BEFORE ANY DB USAGE ---
-init_db()
 
-from core.utils import save_conversations, load_conversations
+
+
 from core.config import configure_gemini, PAGE_CONFIG
-from core.utils import get_current_time, create_new_conversation
+from core.utils import get_current_time,get_ai_response
 from css.styles import apply_custom_css
 from components.header import render_header
 from components.sidebar import render_sidebar
 from components.chat_interface import render_chat_interface
 from components.emergency_page import render_emergency_page
-
+from core.db import init_db, get_messages, add_message, create_chat
+from core.db import get_user_by_username, create_user  # <-- Add this line, adjust module if needed
+# --- ENSURE TABLES EXIST BEFORE ANY DB USAGE ---
+init_db()
 
 # --- 1. INITIALIZE SESSION STATE ---
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-if "conversations" not in st.session_state:
-    st.session_state.conversations = load_conversations()
-if "active_conversation" not in st.session_state:
-    st.session_state.active_conversation = -1
 if "show_emergency_page" not in st.session_state:
     st.session_state.show_emergency_page = False
 if "sidebar_state" not in st.session_state:
     st.session_state.sidebar_state = "expanded"
-if "mental_disorders" not in st.session_state:
-    st.session_state.mental_disorders = [
-        "Depression & Mood Disorders", "Anxiety & Panic Disorders", "Bipolar Disorder",
-        "PTSD & Trauma", "OCD & Related Disorders", "Eating Disorders",
-        "Substance Use Disorders", "ADHD & Neurodevelopmental", "Personality Disorders",
-        "Sleep Disorders"
-    ]
 if "selected_tone" not in st.session_state:
     st.session_state.selected_tone = "Compassionate Listener"
+if "user_id" not in st.session_state:
+    st.session_state.user_id = None
+if "active_chat_id" not in st.session_state:
+    st.session_state.active_chat_id = None
 
 # --- 2. SET PAGE CONFIG ---
 
@@ -75,22 +67,22 @@ def get_tone_prompt():
     return TONE_OPTIONS.get(st.session_state.get("selected_tone", "Compassionate Listener"), TONE_OPTIONS["Compassionate Listener"])
 
 # --- 6. RENDER SIDEBAR ---
-render_sidebar()
+def render_sidebar():
+    from core.db import get_chats_for_user, create_chat
+    if "user_id" in st.session_state and st.session_state.user_id:
+        chats = get_chats_for_user(st.session_state.user_id)
+        for chat in chats:
+            if st.button(chat["title"], key=f"chat_{chat['id']}"):
+                st.session_state.active_chat_id = chat["id"]
+                st.rerun()
+        if st.button("➕ New Chat"):
+            chat_id = create_chat(st.session_state.user_id, "Untitled Chat")
+            st.session_state.active_chat_id = chat_id
+            st.rerun()
 
 # --- 7. PAGE ROUTING ---
 main_area = st.container()
 
-if not st.session_state.conversations:
-    saved_conversations = load_conversations()
-    if saved_conversations:
-        st.session_state.conversations = saved_conversations
-        if st.session_state.active_conversation == -1:
-            st.session_state.active_conversation = 0
-    else:
-        create_new_conversation()
-        st.session_state.active_conversation = 0
-    st.rerun()
-
 # --- 8. RENDER PAGE ---
 if st.session_state.get("show_emergency_page"):
     with main_area:
@@ -99,8 +91,57 @@ def get_tone_prompt():
     with main_area:
         render_header()
         st.subheader(f"🗣️ Current Chatbot Tone: **{st.session_state['selected_tone']}**")
-        render_chat_interface()
-        handle_chat_input(model, system_prompt=get_tone_prompt())
+        render_chat_interface(model, system_prompt=get_tone_prompt())
+
+# --- USER LOGIN/REGISTER ---e
+
+if st.session_state.user_id is None:
+    st.title("Login / Register")
+    username = st.text_input("Username")
+    email = st.text_input("Email")
+    if st.button("Register/Login"):
+        user = get_user_by_username(username)
+        if not user:
+            create_user(username, email)
+            user = get_user_by_username(username)
+        st.session_state.user_id = user["id"]
+        st.rerun()
+    st.stop()
+
+# --- CHAT CREATION ---
+if "active_chat_id" not in st.session_state:
+    st.session_state.active_chat_id = None
+
+if st.button("➕ New Chat"):
+    chat_id = create_chat(st.session_state.user_id, "Untitled Chat")
+    st.session_state.active_chat_id = chat_id
+    st.rerun()
+
+# Optionally, set a default chat on first login
+if st.session_state.active_chat_id is None:
+    chat_id = create_chat(st.session_state.user_id, "Untitled Chat")
+    st.session_state.active_chat_id = chat_id
+
+# --- CHAT INTERFACE ---
+def render_chat_interface(model, system_prompt):
+    chat_id = st.session_state.get("active_chat_id")
+    if not chat_id:
+        st.info("Start a new chat to begin.")
+        return
+
+    messages = get_messages(chat_id)
+    for msg in messages:
+        st.write(f"{msg['sender']}: {msg['message']}")
+
+    with st.form(key="chat_form", clear_on_submit=True):
+        user_input = st.text_input("Your message")
+        send_pressed = st.form_submit_button("Send")
+    if send_pressed and user_input.strip():
+        add_message(chat_id, "user", user_input.strip())
+        # Generate AI response
+        ai_response = get_ai_response(user_input, model)
+        add_message(chat_id, "bot", ai_response)
+        st.rerun()
 
 # --- 9. SCROLL SCRIPT ---
 st.markdown("""
@@ -113,4 +154,4 @@ def get_tone_prompt():
     }
     setTimeout(scrollToBottom, 100);
 </script>
-""", unsafe_allow_html=True) 
\ No newline at end of file
+""", unsafe_allow_html=True)
\ No newline at end of file
diff --git a/components/chat_interface.py b/components/chat_interface.py
index 462ad45..8b7733b 100644
--- a/components/chat_interface.py
+++ b/components/chat_interface.py
@@ -1,7 +1,8 @@
 import streamlit as st
 import streamlit.components.v1 as components
 from datetime import datetime
-from core.utils import get_current_time, get_ai_response, save_conversations
+from core.utils import get_current_time, get_ai_response
+from core.db import get_messages, add_message
 
 # Inject JS to get user's local time zone
 def set_user_time_in_session():
@@ -33,97 +34,21 @@ def set_user_time_in_session():
 set_user_time_in_session()
 
 # Display chat messages
-def render_chat_interface():    
-    if st.session_state.active_conversation >= 0:
-        active_convo = st.session_state.conversations[st.session_state.active_conversation]
+def render_chat_interface(model, system_prompt):
+    chat_id = st.session_state.get("active_chat_id")
+    if not chat_id:
+        st.info("Start a new chat to begin.")
+        return
 
-        if not active_convo["messages"]:
-            st.markdown(f"""
-            <div class="welcome-message">
-                <strong>Hello! I'm TalkHeal, your mental health companion 🤗</strong><br>
-                How are you feeling today? You can write below or start a new topic.
-                <div class="message-time">{get_current_time()}</div>
-            </div>
-            """, unsafe_allow_html=True)
-
-        for msg in active_convo["messages"]:
-            css_class = "user-message" if msg["sender"] == "user" else "bot-message"
-            st.markdown(f"""
-            <div class="{css_class}">
-                {msg["message"]}
-                <div class="message-time">{msg["time"]}</div>
-            </div>
-            """, unsafe_allow_html=True)
-
-# Handle chat input and generate AI response
-def handle_chat_input(model, system_prompt):
-    if "pre_filled_chat_input" not in st.session_state:
-        st.session_state.pre_filled_chat_input = ""
-    initial_value = st.session_state.pre_filled_chat_input
-    st.session_state.pre_filled_chat_input = ""
+    messages = get_messages(chat_id)
+    for msg in messages:
+        st.write(f"{msg['sender']}: {msg['message']}")
 
     with st.form(key="chat_form", clear_on_submit=True):
-        col1, col2 = st.columns([5, 1])
-        with col1:
-            user_input = st.text_input(
-                "Share your thoughts...",
-                key="message_input",
-                label_visibility="collapsed",
-                placeholder="Type your message here...",
-                value=initial_value
-            )
-        with col2:
-            send_pressed = st.form_submit_button("Send", use_container_width=True)
-
-    if (send_pressed or st.session_state.get("send_chat_message", False)) and user_input.strip():
-        if 'send_chat_message' in st.session_state:
-            st.session_state.send_chat_message = False
-
-        if st.session_state.active_conversation >= 0:
-            current_time = get_current_time()
-            active_convo = st.session_state.conversations[st.session_state.active_conversation]
-
-            # Save user message
-            active_convo["messages"].append({
-                "sender": "user",
-                "message": user_input.strip(),
-                "time": current_time
-            })
-
-            # Set title if it's the first message
-            if len(active_convo["messages"]) == 1:
-                title = user_input[:30] + "..." if len(user_input) > 30 else user_input
-                active_convo["title"] = title
-
-            save_conversations(st.session_state.conversations)
-
-            # Format memory
-            def format_memory(convo_history, max_turns=10):
-                context = ""
-                for msg in convo_history[-max_turns*2:]:  # user + bot per turn
-                    sender = "User" if msg["sender"] == "user" else "Bot"
-                    context += f"{sender}: {msg['message']}\n"
-                return context
-
-            try:
-                with st.spinner("TalkHeal is thinking..."):
-                    memory = format_memory(active_convo["messages"])
-                    prompt = f"{system_prompt}\n\n{memory}\nUser: {user_input.strip()}\nBot:"
-                    ai_response = get_ai_response(prompt, model)
-
-                    active_convo["messages"].append({
-                        "sender": "bot",
-                        "message": ai_response,
-                        "time": get_current_time()
-                    })
-
-            except Exception as e:
-                st.error(f"An error occurred: {e}")
-                active_convo["messages"].append({
-                    "sender": "bot",
-                    "message": "I’m having trouble responding right now. Please try again in a moment.",
-                    "time": get_current_time()
-                })
-
-            save_conversations(st.session_state.conversations)
-            st.rerun()
+        user_input = st.text_input("Your message")
+        send_pressed = st.form_submit_button("Send")
+    if send_pressed and user_input.strip():
+        add_message(chat_id, "user", user_input.strip())
+        ai_response = get_ai_response(user_input, model)
+        add_message(chat_id, "bot", ai_response)
+        st.rerun()
diff --git a/components/sidebar.py b/components/sidebar.py
index 208d066..6e7406b 100644
--- a/components/sidebar.py
+++ b/components/sidebar.py
@@ -1,7 +1,7 @@
 import streamlit as st
 import webbrowser
 from datetime import datetime
-from core.utils import create_new_conversation, get_current_time
+from core.utils import get_current_time
 from core.theme import get_current_theme, toggle_theme, set_palette, PALETTES
 
 # --- Structured Emergency Resources ---
@@ -73,6 +73,8 @@
 
 
 def render_sidebar():
+    from core.db import get_chats_for_user  # <-- move import here
+
     """Renders the left and right sidebars."""
 
     with st.sidebar:
@@ -85,7 +87,9 @@ def render_sidebar():
             st.session_state.send_chat_message = False
 
         if st.button("➕ New Chat", key="new_chat", use_container_width=True, type="primary"):
-            create_new_conversation()
+            from core.db import create_chat
+            chat_id = create_chat(st.session_state.user_id, "Untitled Chat")
+            st.session_state.active_chat_id = chat_id
             st.session_state.show_quick_start_prompts = True
             st.rerun()
         if st.session_state.show_quick_start_prompts:
@@ -108,58 +112,6 @@ def render_sidebar():
 
             st.markdown("---")
 
-        if st.session_state.conversations:
-            if "delete_candidate" not in st.session_state:
-                for i, convo in enumerate(st.session_state.conversations):
-                    is_active = i == st.session_state.active_conversation
-                    button_style_icon = "🟢" if is_active else "📝"
-
-                    col1, col2 = st.columns([5, 1])
-                    with col1:
-                        if st.button(
-                            f"{button_style_icon} {convo['title'][:22]}...",
-                            key=f"convo_{i}",
-                            help=f"Started: {convo['date']}",
-                            use_container_width=True
-                        ):
-                            st.session_state.active_conversation = i
-                            st.rerun()
-                    with col2:
-                        if st.button("🗑️", key=f"delete_{i}", type="primary"):
-                            st.session_state.delete_candidate = i
-                            st.rerun()
-
-            else:
-                st.warning(
-                    "⚠️ Are you sure you want to delete this conversation?")
-                col_confirm, col_cancel = st.columns(2)
-
-                if col_confirm.button("Yes, delete", key="confirm_delete"):
-                    del st.session_state.conversations[st.session_state.delete_candidate]
-
-                    from core.utils import save_conversations
-                    save_conversations(st.session_state.conversations)
-
-                    del st.session_state.delete_candidate
-                    st.session_state.active_conversation = -1
-                    st.rerun()
-
-                if "cancel_clicked" not in st.session_state:
-                    st.session_state.cancel_clicked = False
-
-                if col_cancel.button("Cancel", key="cancel_delete"):
-                    if not st.session_state.cancel_clicked:
-                        st.session_state.cancel_clicked = True
-                        del st.session_state.delete_candidate
-                        st.rerun()
-                else:
-                    st.session_state.cancel_clicked = False
-
-        else:
-            st.info("No conversations yet. Start a new chat!")
-
-        st.markdown("---")
-
         # --- DEDICATED EMERGENCY PAGE BUTTON ---
         if st.button("🚨 Emergency Help", use_container_width=True, type="secondary"):
             st.session_state.show_emergency_page = True
diff --git a/talkheal.db b/talkheal.db
index 0920560a573eee853112c619b18876ad3ab5ffa2..47bfb47da9a92a558a9a70781c7e2ee8040a7328 100644
GIT binary patch
delta 445
zcmZXQKTpC?6vf}`0|kP)AQDZK@Fo~Q#1_P`sBw{v#K{S2^K7f7FKKHGle#%-8oz<F
zf5OT~FzV<42^+>;XJ0YK7?+!ybAR{bo_jXJXCr*Mh~rmF+c^HvzJVA49l;}9gWgLf
z_8?g#VKiS*c2!;Px<T>pvgwFit}E60YR$jSa<MW*FoaJS_mcD(Vo@@WCp_`TD;Y8F
zBk02$yom8F%%w;HPn<UOj%r#ZjdzIuvQZJ>0Ko^m!$5#fP>B<g!5!ubn7L9^61JM5
zdTuC+MSJWr$0aiep%AnYyuv+PKwBK$gh(zFMmVn!f7Dt?1h=l&<+Yqh@<*%&qVy~N
zq+0i1Tg8Y3JaFa0QG-)mnnpNvo>j;`_Zs;Qb-Ckoi?l&C)y{iV<F?H-k2<Dqc+}=6
zCEB!@>N2VsOgo~U!RS^~(A+*S^<zi%Om3ItvRqv**Ouif{r;#@$>Ax4(;0;X`F5CP
Lae8RVv;V;_BZqg0

delta 148
zcmZp8z}WDBae_1>+e8^>Mz)O!OZYjM_@^@PpW&anSx}*qfAUoMe4xm42L5mS&w(Om
z_$LPNPkyd13lzM-!2g2(0#NV>|KtnyCP0zb4E$gDUqh5l5SHg)VixC2&PXgN=450R
lXDlsFEh=W>Wnf@n<iEkde*>(JiGgAAjd&d<1_7uxJphGfEUN$j


From fa4cef60e2b3198823da369936f23049e5f08d24 Mon Sep 17 00:00:00 2001
From: KingSagnik <sagnikd587@gmail.com>
Date: Sat, 26 Jul 2025 23:21:24 +0530
Subject: [PATCH 4/4] Database test

---
 talkheal.db | Bin 28672 -> 28672 bytes
 1 file changed, 0 insertions(+), 0 deletions(-)

diff --git a/talkheal.db b/talkheal.db
index 47bfb47da9a92a558a9a70781c7e2ee8040a7328..6d683fe9ba1ff069cd75352c5b169824bd3c2c3e 100644
GIT binary patch
delta 78
zcmZp8z}WDBae_1>??f4AM&6AHQ~c#vIT@M78B2>(i;9_f85kH?`7IduZ}4y6@8D11
ix7aKwpvKRn#yYtxUWUs|!O+ag*wV_#Y;!{V0s#OGEE6^W

delta 58
zcmZp8z}WDBae_1>&qNt#MxKocQ~YIEI2oD68B2>(i;9_f85kH?_!AiTZ}4y6@8D0^
OEGQ7bzd0d(fdBx{R1aqW