Update app.py

app.py

@@ -1,115 +1,27 @@
+# app.py
 import streamlit as st
-import openai
-from dotenv import load_dotenv
-import os
-from appointment_agent import book_appointment, cancel_appointment, follow_up_appointment
-from medication_agent import set_medication_reminder, mark_medication_taken, suggest_medications
-from health_tracker_agent import log_symptom, get_health_report, track_health_stat, personalized_health_advice
-from lifestyle_agent import recommend_lifestyle, get_fitness_tips, get_sleep_advice, get_diet_tips
-
-# Load environment variables
-load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-# Streamlit page configuration
-st.set_page_config(page_title="PHS - 2025 Healthcare System", layout="wide")
-st.title("PHS - 2025 Healthcare Management System")
-
-# Sidebar Navigation
-st.sidebar.title("Navigation")
-menu = ["Book Appointment", "Medication Management", "Health Tracker", "Lifestyle Recommendations"]
-choice = st.sidebar.selectbox("Select a Feature", menu)
-
-if choice == "Book Appointment":
-    st.subheader("Book an Appointment")
-    doctor = st.text_input("Enter Doctor's Name")
-    date = st.date_input("Select Appointment Date")
-    time = st.time_input("Select Appointment Time")
-
-    if st.button("Book Appointment"):
-        appointment_message = book_appointment(doctor, str(date), str(time))
-        st.success(appointment_message)
-
-    st.subheader("Follow-up on Appointment")
-    appointment_id = st.text_input("Enter Appointment ID to Follow-up")
-    if st.button("Follow-up Appointment"):
-        follow_up_message = follow_up_appointment(appointment_id)
-        st.info(follow_up_message)
-
-    st.subheader("Cancel Appointment")
-    cancel_id = st.text_input("Enter Appointment ID to Cancel")
-    if st.button("Cancel Appointment"):
-        cancel_message = cancel_appointment(cancel_id)
-        st.warning(cancel_message)
-
-elif choice == "Medication Management":
-    st.subheader("Medication Reminder Setup")
-    medicine = st.text_input("Enter Medicine Name")
-    reminder_time = st.time_input("Set Reminder Time")
-
-    if st.button("Set Medication Reminder"):
-        reminder_message = set_medication_reminder(medicine, str(reminder_time))
-        st.success(reminder_message)
-
-    st.subheader("Mark Medication as Taken")
-    reminder_id = st.text_input("Enter Medication Reminder ID to Mark Taken")
-    if st.button("Mark as Taken"):
-        taken_message = mark_medication_taken(reminder_id)
-        st.success(taken_message)
-
-    st.subheader("Medication Suggestions")
-    symptom = st.text_input("Enter Symptom (e.g., headache, fever)")
-    if st.button("Get Medication Suggestions"):
-        suggestions = suggest_medications(symptom)
-        st.info(f"Suggested Medications: {', '.join(suggestions)}")
-
-elif choice == "Health Tracker":
-    st.subheader("Log Symptom")
-    symptom = st.text_input("Enter Symptom (e.g., headache, fatigue)")
-    value = st.text_input("Enter Symptom Value (e.g., 5/10)")
-    if st.button("Log Symptom"):
-        log_message = log_symptom(symptom, value)
-        st.success(log_message)
-
-    st.subheader("Health Report")
-    if st.button("Get Health Report"):
-        health_report = get_health_report()
-        st.dataframe(health_report)
-
-    st.subheader("Track Health Stats")
-    stat_type = st.selectbox("Select Stat to Track", ["blood_pressure", "weight", "heart_rate"])
-    if st.button("Track Health Stat"):
-        stats = track_health_stat(stat_type)
-        st.write(stats)
-
-    st.subheader("Personalized Health Advice from AI")
-    if st.button("Get Personalized Health Advice"):
-        advice = personalized_health_advice(symptom, value)
-        st.info(advice)
-
-elif choice == "Lifestyle Recommendations":
-    st.subheader("Lifestyle Recommendations")
-    symptom = st.text_input("Enter Symptom (e.g., high blood pressure, diabetes)")
-    value = st.text_input("Enter Symptom Value")
-    if st.button("Get AI-based Lifestyle Recommendations"):
-        response = openai.completions.create(
-            model="gpt-3.5-turbo",
-            prompt=f"Recommend a lifestyle modification plan for someone with {symptom} and value {value}.",
-            max_tokens=150
-        )
-        st.info(response['choices'][0]['text'].strip())
-
-    st.subheader("Fitness Tips")
-    if st.button("Get Fitness Tips"):
-        fitness_tips = get_fitness_tips()
-        st.info(fitness_tips)
-
-    st.subheader("Sleep Advice")
-    if st.button("Get Sleep Advice"):
-        sleep_advice = get_sleep_advice()
-        st.info(sleep_advice)
-
-    st.subheader("Diet Tips")
-    if st.button("Get Diet Tips"):
-        diet_tips = get_diet_tips()
-        st.info(diet_tips)
+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)