import streamlit as st
import pandas as pd
from datetime import datetime
from database import init_db, add_medication, get_all_medications

# Initialize the database
init_db()

# Set page config
st.set_page_config(
    page_title="Medication Tracker",
    page_icon="💊",
    layout="centered",
    initial_sidebar_state="auto",
)

# Custom CSS for glass style UI
st.markdown(
    """
    <style>
    body {
        font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
        background-color: #f7f7f8;
        color: #333;
    }
    .container {
        max-width: 600px;
        margin: 0 auto;
        padding: 2rem;
        background: rgba(255, 255, 255, 0.1);
        border-radius: 16px;
        backdrop-filter: blur(10px);
        box-shadow: 0 4px 30px rgba(0, 0, 0, 0.1);
        border: 1px solid rgba(255, 255, 255, 0.3);
    }
    h1, h2, h3 {
        font-weight: 600;
    }
    .btn {
        display: inline-block;
        padding: 0.75rem 1.25rem;
        font-size: 1rem;
        font-weight: 600;
        color: #fff;
        background-color: #007aff;
        border: none;
        border-radius: 10px;
        cursor: pointer;
        text-align: center;
    }
    .btn:active {
        background-color: #005fcb;
    }
    .input-field {
        width: 100%;
        padding: 0.75rem;
        margin-bottom: 1rem;
        border: 1px solid #ddd;
        border-radius: 10px;
    }
    .log-entry {
        padding: 0.75rem;
        margin-bottom: 1rem;
        background: rgba(255, 255, 255, 0.6);
        border-radius: 10px;
        backdrop-filter: blur(5px);
        box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
    }
    </style>
    """, 
    unsafe_allow_html=True
)

st.markdown('<div class="container">', unsafe_allow_html=True)
st.markdown("<h1>Medication Tracker 💊</h1>", unsafe_allow_html=True)
st.markdown("<p>Track your prescription medications easily and get helpful insights.</p>", unsafe_allow_html=True)

# Form to log medication
st.markdown('<div class="log-form">', unsafe_allow_html=True)
st.markdown("<h2>Log Your Medication</h2>", unsafe_allow_html=True)
with st.form(key='log_medication'):
    medication_name = st.text_input("Medication Name", key='medication_name', placeholder="e.g., Aspirin")
    dosage = st.text_input("Dosage (e.g., 500 mg)", key='dosage', placeholder="e.g., 500 mg")
    time_of_day = st.selectbox("Time of Day", ["Morning", "Afternoon", "Evening", "Night"], key='time_of_day')
    date = st.date_input("Date", datetime.today(), key='date')
    submit_button = st.form_submit_button(label='Log Medication')

    if submit_button and medication_name and dosage:
        add_medication(medication_name, dosage, time_of_day, date)
        st.success(f"Logged: {medication_name} - {dosage} at {time_of_day} on {date}")

st.markdown('</div>', unsafe_allow_html=True)

# Display logged medications
st.markdown('<div class="logged-medications">', unsafe_allow_html=True)
st.markdown("<h2>Logged Medications</h2>", unsafe_allow_html=True)
medications = get_all_medications()
if not medications:
    st.markdown("No medications logged yet.", unsafe_allow_html=True)
else:
    for med in medications:
        st.markdown(f'<div class="log-entry"><strong>{med[1]}</strong><br>Dosage: {med[2]}<br>Time: {med[3]}<br>Date: {med[4]}</div>', unsafe_allow_html=True)

st.markdown('</div>', unsafe_allow_html=True)

# Provide insights
st.markdown('<div class="insights">', unsafe_allow_html=True)
st.markdown("<h2>Insights</h2>", unsafe_allow_html=True)
if medications:
    df = pd.DataFrame(medications, columns=["ID", "Medication", "Dosage", "Time", "Date"])
    most_common_med = df['Medication'].mode()[0]
    st.markdown(f"Most frequently taken medication: {most_common_med}", unsafe_allow_html=True)
    
    avg_dosage_time = df.groupby('Time').size().idxmax()
    st.markdown(f"Most common time of day for taking medications: {avg_dosage_time}", unsafe_allow_html=True)
    
    st.markdown("Take your medications consistently at the same time each day to maintain stable drug levels in your body.", unsafe_allow_html=True)
    st.markdown("If you have multiple medications, consult your healthcare provider to ensure there are no adverse interactions.", unsafe_allow_html=True)

st.markdown('</div>', unsafe_allow_html=True)

# Footer
st.markdown("<hr>", unsafe_allow_html=True)
st.markdown("<p><strong>Disclaimer:</strong> This app is for tracking purposes only. Always consult with your healthcare provider for medical advice.</p>", unsafe_allow_html=True)
st.markdown('</div>', unsafe_allow_html=True)