src/streamlit_app.py

import streamlit as st
import sqlite3
from datetime import datetime, timedelta
import os
import json
import requests

# -----------------------------
# CONFIG
# -----------------------------
st.set_page_config(
    page_title="SmartDay.AI",
    page_icon="🌸",
    layout="wide",
    initial_sidebar_state="expanded"
)

# Theme Colors
PRIMARY_COLOR = "#FF6F61"  # Coral
BG_COLOR = "#FFFFFF"       # White

# Database setup
DB_FILE = "smartday_ai.db"

def init_db():
    conn = sqlite3.connect(DB_FILE)
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS users
                 (username TEXT PRIMARY KEY, password TEXT)''')
    c.execute('''CREATE TABLE IF NOT EXISTS journal
                 (username TEXT, date TEXT, entry TEXT)''')
    c.execute('''CREATE TABLE IF NOT EXISTS planner
                 (username TEXT, date TEXT, task TEXT, status TEXT)''')
    conn.commit()
    conn.close()

init_db()

# -----------------------------
# AUTHENTICATION
# -----------------------------
def register_user(username, password):
    conn = sqlite3.connect(DB_FILE)
    c = conn.cursor()
    try:
        c.execute("INSERT INTO users VALUES (?,?)", (username, password))
        conn.commit()
        return True
    except:
        return False
    finally:
        conn.close()

def login_user(username, password):
    conn = sqlite3.connect(DB_FILE)
    c = conn.cursor()
    c.execute("SELECT * FROM users WHERE username=? AND password=?", (username, password))
    result = c.fetchone()
    conn.close()
    return result is not None

# -----------------------------
# GROQ AI INTEGRATION
# -----------------------------
def groq_chat(prompt):
    """Call Groq API - key saved in Hugging Face Secrets"""
    api_key = os.environ.get("GROQ_API_KEY")
    if not api_key:
        return "⚠️ No API key found. Please set GROQ_API_KEY in Secrets."
    
    url = "https://api.groq.com/openai/v1/chat/completions"
    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
    data = {
        "model": "llama-3.1-8b-instant",
        "messages": [{"role": "user", "content": prompt}],
        "temperature": 0.5
    }

    response = requests.post(url, headers=headers, json=data)
    if response.status_code == 200:
        return response.json()["choices"][0]["message"]["content"]
    else:
        return f"Error: {response.text}"

# -----------------------------
# DASHBOARD COMPONENTS
# -----------------------------
def learning_journal(username):
    st.header("📖 Learning Journal")
    today = datetime.today().strftime("%Y-%m-%d")
    conn = sqlite3.connect(DB_FILE)
    c = conn.cursor()

    entry = st.text_area("Write today’s reflection or learning:", height=150)
    if st.button("Save Entry"):
        c.execute("INSERT INTO journal VALUES (?,?,?)", (username, today, entry))
        conn.commit()
        st.success("Journal saved!")

    st.subheader("Past Entries")
    c.execute("SELECT date, entry FROM journal WHERE username=? ORDER BY date DESC", (username,))
    rows = c.fetchall()
    for r in rows:
        st.markdown(f"**{r[0]}** — {r[1]}")
    conn.close()

def smart_day_planner(username):
    st.header("🧠 Smart Day Planner")
    conn = sqlite3.connect(DB_FILE)
    c = conn.cursor()

    task = st.text_input("Add a new task:")
    if st.button("Add Task"):
        c.execute("INSERT INTO planner VALUES (?,?,?,?)", (username, datetime.today().strftime("%Y-%m-%d"), task, "Pending"))
        conn.commit()
        st.success("Task added!")

    st.subheader("Today’s Tasks")
    c.execute("SELECT task, status FROM planner WHERE username=? AND date=?", (username, datetime.today().strftime("%Y-%m-%d"),))
    tasks = c.fetchall()
    for t in tasks:
        if st.checkbox(f"{t[0]} — {t[1]}", value=(t[1]=="Done")):
            c.execute("UPDATE planner SET status='Done' WHERE username=? AND task=?", (username, t[0]))
            conn.commit()
    conn.close()

def weekly_summary(username):
    st.header("📅 Weekly Summary & Insights")
    conn = sqlite3.connect(DB_FILE)
    c = conn.cursor()
    week_start = (datetime.today() - timedelta(days=7)).strftime("%Y-%m-%d")
    c.execute("SELECT date, entry FROM journal WHERE username=? AND date>=?", (username, week_start))
    data = c.fetchall()

    if not data:
        st.info("No entries found for this week.")
        return

    combined_text = "\n".join([f"{d[0]}: {d[1]}" for d in data])
    st.write("Here’s your AI-powered reflection summary 👇")
    summary = groq_chat(f"Summarize and analyze this weekly journal:\n{combined_text}")
    st.markdown(summary)
    conn.close()

def ai_assistant():
    st.header("🤖 Ask SmartDay.AI")
    query = st.text_area("What would you like help with?", height=100)
    if st.button("Ask AI"):
        response = groq_chat(query)
        st.markdown(f"**Answer:** {response}")

# -----------------------------
# MAIN APP
# -----------------------------
def main():
    st.markdown(
        f"""
        <style>
        body {{
            background-color: {BG_COLOR};
            color: #333;
        }}
        .stButton>button {{
            background-color: {PRIMARY_COLOR};
            color: white;
            border-radius: 8px;
            padding: 0.6em 1.2em;
        }}
        </style>
        """,
        unsafe_allow_html=True
    )

    st.sidebar.title("🌸 SmartDay.AI")
    menu = ["Login", "Register"]
    choice = st.sidebar.selectbox("Menu", menu)

    if choice == "Register":
        st.subheader("Create Account")
        new_user = st.text_input("Username")
        new_pass = st.text_input("Password", type="password")
        if st.button("Register"):
            if register_user(new_user, new_pass):
                st.success("Account created! Please go to Login.")
            else:
                st.error("User already exists.")

    elif choice == "Login":
        st.subheader("User Login")
        username = st.text_input("Username")
        password = st.text_input("Password", type="password")
        if st.button("Login"):
            if login_user(username, password):
                st.session_state["user"] = username
                st.success(f"Welcome, {username}!")
            else:
                st.error("Invalid credentials.")

    if "user" in st.session_state:
        username = st.session_state["user"]
        st.sidebar.success(f"Logged in as {username}")

        page = st.sidebar.radio("Navigate", ["Learning Journal", "Smart Day Planner", "Weekly Summary", "AI Assistant"])
        if page == "Learning Journal":
            learning_journal(username)
        elif page == "Smart Day Planner":
            smart_day_planner(username)
        elif page == "Weekly Summary":
            weekly_summary(username)
        elif page == "AI Assistant":
            ai_assistant()

if __name__ == "__main__":
    main()