src/streamlit_app.py

import streamlit as st
from supabase import create_client
from datetime import datetime
from zoneinfo import ZoneInfo
from supabase_auth.errors import AuthApiError
import time
import os

mood_options = ["😡 Giận dữ", "😢 Buồn bã", "😐 Bình thường", "😊 Vui vẻ", "🤩 Hào hứng"]

# ========== CONFIG ==========
st.set_page_config(
    page_title="Personal Diary",
    page_icon="📔",
    layout="wide"
)

SUPABASE_URL = os.environ.get("SUPABASE_URL")
SUPABASE_KEY = os.environ.get("SUPABASE_KEY")

supabase = create_client(
    SUPABASE_URL,
    SUPABASE_KEY
)

# ========== STYLE ==========
st.markdown("""
<style>
.diary-card {
    padding: 1rem;
    border-radius: 12px;
    background-color: #f8f9fa;
    margin-bottom: 1rem;
}
.small-text {
    font-size: 0.8rem;
    color: #6c757d;
}
</style>
""", unsafe_allow_html=True)

def signup(email, password):
    return supabase.auth.sign_up({
        "email": email,
        "password": password
    })


# ========== HEADER ==========
st.title("📔 Personal Diary")
st.caption("Ghi chép suy nghĩ mỗi ngày – phiên bản demo")

# ========== LOGIN ==========
if "user" not in st.session_state:
    st.subheader("🔐 Tài khoản")

    tab_login, tab_signup = st.tabs(["Đăng nhập", "Đăng ký"])

    # ===== LOGIN =====
    with tab_login:
        email = st.text_input("Email", key="login_email")
        password = st.text_input("Mật khẩu", type="password", key="login_pw")

        if st.button("➡️ Đăng nhập", use_container_width=True):
            res = supabase.auth.sign_in_with_password({
                "email": email,
                "password": password
            })

            if res.user:
                st.session_state.user = res.user
                st.success("Đăng nhập thành công")
                st.rerun()
            else:
                st.error("Sai email hoặc mật khẩu")

    # ===== SIGNUP =====
    with tab_signup:
        email = st.text_input("Email đăng ký", key="signup_email")
        password = st.text_input(
            "Mật khẩu (tối thiểu 6 ký tự)",
            type="password",
            key="signup_pw"
        )

        # init cooldown
        if "signup_cooldown_until" not in st.session_state:
            st.session_state.signup_cooldown_until = 0

        now = time.time()
        cooldown_left = int(st.session_state.signup_cooldown_until - now)

        if cooldown_left > 0:
            st.warning(f"⏳ Vui lòng thử lại sau {cooldown_left} giây")

        signup_disabled = cooldown_left > 0

        if st.button(
                "🆕 Tạo tài khoản",
                use_container_width=True,
                disabled=signup_disabled
        ):
            try:
                res = signup(email, password)

                if res.user:
                    st.session_state.user = res.user
                    st.success("Đăng ký thành công 🎉")
                    st.rerun()
                else:
                    st.error("Không thể đăng ký")

            except AuthApiError as e:
                msg = str(e)

                # Bắt lỗi rate limit 60s
                if "only request this after" in msg:
                    # Mặc định 60s nếu parse không được
                    wait_seconds = 60

                    # cố gắng parse số giây từ message
                    import re

                    match = re.search(r"after (\d+) seconds", msg)
                    if match:
                        wait_seconds = int(match.group(1))

                    st.session_state.signup_cooldown_until = time.time() + wait_seconds
                    st.error(f"🚫 Bạn đã thử quá nhiều lần. Vui lòng đợi {wait_seconds} giây rồi thử lại.")
                    st.rerun()
                else:
                    st.error("Lỗi đăng ký: " + msg)


# ========== MAIN APP ==========
else:
    user = st.session_state.user

    # Top bar
    top_left, top_right = st.columns([4, 1])
    with top_left:
        st.markdown(f"👋 Xin chào **{user.email}**")
    with top_right:
        if st.button("🚪 Đăng xuất", use_container_width=True):
            st.session_state.clear()
            st.rerun()

    st.divider()

    # Main layout
    left, right = st.columns([2, 3])

    # ===== LEFT: WRITE =====
    with left:
        st.subheader("✍️ Viết nhật ký")

        # 1. Chọn Thời gian
        col_date, col_time = st.columns(2)
        with col_date:
            d = st.date_input("Chọn ngày", datetime.now())
        with col_time:
            t = st.time_input("Chọn giờ", datetime.now().time())
        
        selected_dt = combine(d, t)

        # 2. Nhập Cảm xúc và Cấp độ
        col_mood, col_level = st.columns([2, 3]) # Chia tỷ lệ cột cho cân đối
        with col_mood:
            mood = st.selectbox(
                "Cảm xúc",
                mood_options,
                index=2 # Mặc định chọn "Bình thường"
            )
        with col_level:
            level = st.select_slider(
                "Cấp độ",
                options=[1, 2, 3, 4, 5],
                value=3 # Mặc định cấp độ trung bình
            )
    
        # 3. Nội dung nhật ký
        content = st.text_area(
            "Hôm nay bạn nghĩ gì?",
            height=250,
            placeholder="Viết suy nghĩ của bạn ở đây..."
        )

        if st.button("💾 Lưu nhật ký", use_container_width=True):
            if content.strip():
                supabase.table("journals").insert({
                    "user_id": user.id,
                    "content": content,
                    "mood": mood,
                    "mood_level": level,
                    "created_at": selected_dt.isoformat()
                }).execute()

                st.success("Đã lưu nhật ký ✨")
                st.rerun()
            else:
                st.warning("Nội dung đang trống")

    # ===== RIGHT: LIST =====
    with right:
        st.subheader("📜 Nhật ký của bạn")

        data = supabase.table("journals") \
            .select("*") \
            .eq("user_id", user.id) \
            .order("created_at", desc=True) \
            .execute()

        if not data.data:
            st.info("Chưa có nhật ký nào")
        else:
            for row in data.data:
                # Format ngày cho gọn
                created_dt = datetime.fromisoformat(
                    row["created_at"].replace("Z", "+00:00")
                ).astimezone(ZoneInfo("Asia/Ho_Chi_Minh"))

                date_label = created_dt.strftime("%d/%m/%Y – %H:%M")

                # Dropdown theo ngày
                with st.expander(f"📅 {date_label}", expanded=False):

                    # Tạo 2 cột để sửa Mood và Level
                    col_m, col_l = st.columns(2)

                    try:
                        old_mood_idx = mood_options.index(row["mood"])
                    except:
                        old_mood_idx = 2 # Mặc định là Bình thường nếu có lỗi
            
                    with col_m:
                        new_mood = st.selectbox(
                            "Cảm xúc", 
                            mood_options, 
                            index=old_mood_idx, 
                            key=f"mood_{row['id']}"
                        )
                    
                    with col_l:
                        new_level = st.select_slider(
                            "Cấp độ", 
                            options=[1, 2, 3, 4, 5], 
                            value=row.get("mood_level", 3), 
                            key=f"level_{row['id']}"
                        )

                    new_content = st.text_area(
                        "Nội dung",
                        row["content"],
                        key=row["id"],
                        height=120
                    )

                    col_u, col_d = st.columns(2)

                    if col_u.button("✏️ Cập nhật", key=f"u{row['id']}"):
                        supabase.table("journals").update({
                            "content": new_content,
                            "mood": new_mood,
                            "mood_level": new_level,
                            "updated_at": datetime.utcnow().isoformat()
                        }).eq("id", row["id"]).execute()
                        st.rerun()

                    if col_d.button("🗑️ Xoá", key=f"d{row['id']}"):
                        supabase.table("journals").delete() \
                            .eq("id", row["id"]).execute()
                        st.rerun()