Code Editor for app.py

import streamlit as st
import pandas as pd
from sqlalchemy import create_engine, Column, Integer, String, DateTime, func
from sqlalchemy.orm import Session
from models import Activity, Base

engine = create_engine('sqlite:///newborn_tracker.sqlite')
Base.metadata.create_all(bind=engine)

st.title('Newborn Tracker')

activity_type = st.selectbox('Select Activity', ['Sleeping', 'Eating', 'Diaper Change'])
if st.button('Log Activity'):
    with Session(engine) as session:
        activity = Activity(activity_type=activity_type)
        session.add(activity)
        session.commit()
        st.success(f'{activity_type} logged successfully!')

if st.button('Show Activity Log'):
    with Session(engine) as session:
        activities = session.query(Activity).all()
        if activities:
            st.table([[activity.id, activity.activity_type, activity.timestamp] for activity in activities])
        else:
            st.write('No activities logged yet.')