import streamlit as st
from medication_agent import get_medication_suggestions
from health_tracker_agent import track_health_status
from lifestyle_agent import get_lifestyle_tips

st.title("Healthcare Assistant")

menu = st.sidebar.selectbox("Choose Service", ("Medication", "Health Tracker", "Lifestyle Advice"))

if menu == "Medication":
    symptom = st.text_input("Enter your symptom:")
    if st.button("Get Medication Suggestions") and symptom:
        suggestions = get_medication_suggestions(symptom)
        st.write(suggestions)

elif menu == "Health Tracker":
    health_input = st.text_area("Enter your current health status or condition:")
    if st.button("Track Health") and health_input:
        result = track_health_status(health_input)
        st.write(result)

elif menu == "Lifestyle Advice":
    lifestyle_query = st.text_input("Enter your lifestyle concern or goal:")
    if st.button("Get Advice") and lifestyle_query:
        advice = get_lifestyle_tips(lifestyle_query)
        st.write(advice)