Update app.py

app.py

@@ -1,6 +1,10 @@
 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(
@@ -10,7 +14,7 @@ st.set_page_config(
     initial_sidebar_state="auto",
 )
 
-# Custom CSS for iOS-like UI
+# Custom CSS for glass style UI
 st.markdown(
     """
     <style>
@@ -23,9 +27,11 @@ st.markdown(
         max-width: 600px;
         margin: 0 auto;
         padding: 2rem;
-        background-color: #fff;
-        border-radius: 12px;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        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;
@@ -55,8 +61,10 @@ st.markdown(
     .log-entry {
         padding: 0.75rem;
         margin-bottom: 1rem;
-        background-color: #f1f1f1;
+        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>
     """, 
@@ -67,10 +75,6 @@ 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)
 
-# Initialize session state to store medication data
-if 'medications' not in st.session_state:
-    st.session_state.medications = pd.DataFrame(columns=["Medication", "Dosage", "Time", "Date"])
-
 # 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)
@@ -82,8 +86,7 @@ with st.form(key='log_medication'):
     submit_button = st.form_submit_button(label='Log Medication')
 
     if submit_button and medication_name and dosage:
-        new_entry = pd.DataFrame({"Medication": [medication_name], "Dosage": [dosage], "Time": [time_of_day], "Date": [date]})
-        st.session_state.medications = pd.concat([st.session_state.medications, new_entry], ignore_index=True)
+        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)
@@ -91,22 +94,24 @@ 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)
-if st.session_state.medications.empty:
+medications = get_all_medications()
+if not medications:
     st.markdown("No medications logged yet.", unsafe_allow_html=True)
 else:
-    for index, row in st.session_state.medications.iterrows():
-        st.markdown(f'<div class="log-entry"><strong>{row["Medication"]}</strong><br>Dosage: {row["Dosage"]}<br>Time: {row["Time"]}<br>Date: {row["Date"]}</div>', unsafe_allow_html=True)
+    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 not st.session_state.medications.empty:
-    most_common_med = st.session_state.medications['Medication'].mode()[0]
+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 = st.session_state.medications.groupby('Time').size().idxmax()
+    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)