Update app.py

app.py

@@ -2,144 +2,100 @@ import streamlit as st
 from dotenv import load_dotenv
 from langchain_groq import ChatGroq
 import os
-import pandas as pd
-import altair as alt
-import datetime
-import random
+import importlib.util
 
-# Set page configuration for enhanced UI
+# Set page configuration
 st.set_page_config(
-    page_title="AI Health & Wellness Planner",
+    page_title="Wellness Planner",
     page_icon="🌿",
     layout="wide",
     initial_sidebar_state="expanded"
 )
 
-# Apply custom CSS styling
+# Custom CSS for better styling
 st.markdown("""
 <style>
     .main {
+        padding: 2rem;
+    }
+    .stApp {
         background-color: #f8f9fa;
     }
+    .css-18e3th9 {
+        padding-top: 2rem;
+    }
     .stTabs [data-baseweb="tab-list"] {
-        gap: 8px;
+        gap: 2px;
     }
     .stTabs [data-baseweb="tab"] {
         height: 50px;
         white-space: pre-wrap;
-        border-radius: 4px 4px 0px 0px;
-        padding: 10px 16px;
-        font-weight: 600;
+        background-color: #f0f2f6;
+        border-radius: 4px 4px 0 0;
+        gap: 1px;
+        padding-top: 10px;
+        padding-bottom: 10px;
     }
     .stTabs [aria-selected="true"] {
-        background-color: #f0f7ff;
-        border-bottom: 2px solid #4361ee;
+        background-color: #4CAF50 !important;
+        color: white !important;
     }
-    .stButton>button {
-        width: 100%;
-        border-radius: 8px;
-        height: 3em;
-        background-color: #4361ee;
+    div.stButton > button:first-child {
+        background-color: #4CAF50;
         color: white;
-        font-weight: 600;
+        font-weight: bold;
+        padding: 0.5rem 1rem;
+        border-radius: 0.5rem;
+        border: none;
     }
-    .stButton>button:hover {
-        background-color: #3a56d4;
-        color: white;
+    div.stButton > button:hover {
+        background-color: #45a049;
+        border: none;
     }
-    .metric-card {
+    .result-container {
         background-color: white;
-        border-radius: 8px;
-        padding: 20px;
+        padding: 2rem;
+        border-radius: 0.5rem;
         box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-        text-align: center;
-    }
-    .input-section {
-        background-color: white;
-        border-radius: 8px;
-        padding: 20px;
-        box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
-        margin-bottom: 20px;
     }
-    .badge {
-        display: inline-block;
-        padding: 5px 10px;
-        border-radius: 15px;
-        font-size: 12px;
-        font-weight: 600;
-        margin-right: 5px;
+    .info-box {
+        background-color: #e8f4f8;
+        padding: 1rem;
+        border-radius: 0.5rem;
+        margin-bottom: 1rem;
     }
-    .badge-green {
-        background-color: #d1fae5;
-        color: #065f46;
-    }
-    .badge-yellow {
-        background-color: #fef3c7;
-        color: #92400e;
-    }
-    .badge-red {
-        background-color: #fee2e2;
-        color: #b91c1c;
-    }
-    .badge-blue {
-        background-color: #dbeafe;
-        color: #1e40af;
+    h1, h2, h3 {
+        color: #2E7D32;
     }
 </style>
 """, unsafe_allow_html=True)
 
-# Check for team_info module
+# Check if team_info module exists and import it
 try:
     import team_info
     has_team_info = True
 except ImportError:
     has_team_info = False
 
-# Load environment variables and initialize model
+# Load environment variables
 load_dotenv()
+
+# Initialize the model
 model = ChatGroq(model="llama3-8b-8192")
 
-# Function to calculate daily calorie requirements with activity level
-def calculate_calorie_requirements(age, gender, weight, height, fitness_goal, activity_level):
+# Function to calculate daily calorie requirements
+def calculate_calorie_requirements(age, gender, weight, height, fitness_goal):
     if gender == "Male":
         bmr = 10 * weight + 6.25 * height - 5 * age + 5
     else:
         bmr = 10 * weight + 6.25 * height - 5 * age - 161
 
-    # Activity level multipliers
-    activity_multipliers = {
-        "Sedentary (little or no exercise)": 1.2,
-        "Lightly active (light exercise/sports 1-3 days/week)": 1.375,
-        "Moderately active (moderate exercise/sports 3-5 days/week)": 1.55,
-        "Very active (hard exercise/sports 6-7 days/week)": 1.725,
-        "Super active (very hard exercise & physical job or 2x training)": 1.9
-    }
-    
-    # Goal adjustments
-    goal_adjustments = {
-        "Weight Loss": 0.8,  # 20% calorie deficit
-        "Weight Gain": 1.15,  # 15% calorie surplus
-        "Maintenance": 1.0    # No adjustment
-    }
-    
-    return bmr * activity_multipliers[activity_level] * goal_adjustments[fitness_goal]
-
-# Function to determine BMI category with detailed breakdown
-def get_bmi_category(bmi):
-    if bmi < 16:
-        return "Severely Underweight", "red"
-    elif 16 <= bmi < 18.5:
-        return "Underweight", "yellow"
-    elif 18.5 <= bmi < 25:
-        return "Normal weight", "green"
-    elif 25 <= bmi < 30:
-        return "Overweight", "yellow"
-    elif 30 <= bmi < 35:
-        return "Obesity Class I", "red"
-    elif 35 <= bmi < 40:
-        return "Obesity Class II", "red"
+    if fitness_goal == "Weight Loss":
+        return bmr * 1.2
+    elif fitness_goal == "Weight Gain":
+        return bmr * 1.5
     else:
-        return "Obesity Class III", "red"
+        return bmr * 1.375
 
 # Function to generate the plan
 def generate_plan_with_prompt(metrics, prompt_template):
@@ -147,57 +103,22 @@ def generate_plan_with_prompt(metrics, prompt_template):
     response = model.invoke(prompt)
     return response
 
-# Function to format the response with enhanced visual styling
+# Function to format the response neatly
 def format_plan(response):
     try:
         content = response.content
-        days = content.split("Day: ")
-        formatted_content = ""
-        for i, day in enumerate(days):
-            if i == 0:  # Skip the first split which might be empty or contain introduction
-                if day.strip():
-                    formatted_content += f"<div class='plan-intro'>{day.strip()}</div>"
-                continue
-            # Create a card for each day
-            formatted_content += f"""
-            <div style='background-color: white; border-radius: 8px; padding: 15px; margin-bottom: 15px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);'>
-                <h3 style='color: #4361ee; border-bottom: 1px solid #e5e7eb; padding-bottom: 10px;'>Day: {day.split('\n', 1)[0].strip()}</h3>
-                <div style='padding: 10px 0;'>
-            """
-            # Process each line of the day's content
-            sections = day.split('\n', 1)[1].split('\n  - ')
-            for section in sections:
-                if not section.strip():
-                    continue
-                # Add special formatting for each type of entry
-                if "Breakfast:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🍳 Breakfast:</strong> {section.replace('Breakfast:', '').strip()}</div>"
-                elif "Lunch:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🍲 Lunch:</strong> {section.replace('Lunch:', '').strip()}</div>"
-                elif "Snacks:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🥨 Snacks:</strong> {section.replace('Snacks:', '').strip()}</div>"
-                elif "Dinner:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🍽️ Dinner:</strong> {section.replace('Dinner:', '').strip()}</div>"
-                elif "Exercise:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🏋️ Exercise:</strong> {section.replace('Exercise:', '').strip()}</div>"
-                elif "Wearable Tracking:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>⌚ Wearable Tracking:</strong> {section.replace('Wearable Tracking:', '').strip()}</div>"
-                elif "Progress Monitoring:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>📊 Progress Monitoring:</strong> {section.replace('Progress Monitoring:', '').strip()}</div>"
-                elif "Food Delivery:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🚚 Food Delivery:</strong> {section.replace('Food Delivery:', '').strip()}</div>"
-                elif "Ayurvedic Tips:" in section:
-                    formatted_content += f"<div style='margin-bottom: 10px;'><strong style='color: #4361ee;'>🌿 Ayurvedic Tips:</strong> {section.replace('Ayurvedic Tips:', '').strip()}</div>"
-                else:
-                    formatted_content += f"<div style='margin-bottom: 10px;'>{section.strip()}</div>"
-            formatted_content += "</div></div>"
-        return formatted_content
+        sections = content.split("\n\n")
+        formatted = ""
+        for section in sections:
+            formatted += f"{section.strip()}\n\n"
+        return formatted
     except Exception as e:
-        return f"<div class='error'>Error formatting plan: {e}</div>"
+        return f"Error formatting plan: {e}"
 
-# Prompt templates (same as original)
+# Ayurvedic prompt template
 ayurveda_prompt_template = """
 You are a health expert specialized in both modern medicine and Ayurveda. Generate a personalized weekly diet and exercise plan for {name}, a {age}-year-old {gender} with a BMI of {bmi} ({health_status}).
+
 Fitness Goal: {fitness_goal}.
 Daily Calorie Requirement: {daily_calories} kcal.
 Dietary Preference: {dietary_preference}.
@@ -205,6 +126,7 @@ Food Allergies: {food_allergies}.
 Local Cuisine: {local_cuisine}.
 Month: {month}.
 Ayurvedic Consideration: True
+
 Plan should include:
 1. A daily diet plan with meal timings, calorie details, and meal alternatives.
 2. Exercise routines based on goals, incorporating cardio, strength, and flexibility.
@@ -224,7 +146,9 @@ Plan should include:
    - Seasonal herbs and spices for balance.
    - Daily routines (Dinacharya) for optimal health.
    - Natural remedies complementing modern medicine.
+
 Provide a detailed plan for each weekday: Monday, Tuesday, Wednesday, Thursday, Friday, Saturday, Sunday.
+
 Return output as:
 Day: {{weekday}}
   - Breakfast: Time, Description, Calories, Ayurvedic Properties
@@ -238,14 +162,17 @@ Day: {{weekday}}
   - Ayurvedic Tips: Daily wellness practices based on dosha.
 """
 
+# Regular prompt template
 regular_prompt_template = """
 You are a health expert. Generate a personalized weekly diet and exercise plan for {name}, a {age}-year-old {gender} with a BMI of {bmi} ({health_status}).
+
 Fitness Goal: {fitness_goal}.
 Daily Calorie Requirement: {daily_calories} kcal.
 Dietary Preference: {dietary_preference}.
 Food Allergies: {food_allergies}.
 Local Cuisine: {local_cuisine}.
 Month: {month}.
+
 Plan should include:
 1. A daily diet plan with meal timings, calorie details, and meal alternatives.
 2. Exercise routines based on goals, incorporating cardio, strength, and flexibility.
@@ -259,7 +186,9 @@ Plan should include:
    - Multi-restaurant meal aggregation for complete diet fulfillment.
    - Location-based meal recommendations.
    - Customizable meal delivery schedules.
+
 Provide a detailed plan for each weekday: Monday, Tuesday, Wednesday, Thursday, Friday, Saturday, Sunday.
+
 Return output as:
 Day: {{weekday}}
   - Breakfast: Time, Description, Calories
@@ -272,367 +201,208 @@ Day: {{weekday}}
   - Food Delivery: Suggested meal items and delivery options.
 """
 
-# Initialize session state variables
-if 'bmi' not in st.session_state:
-    st.session_state.bmi = 0
-if 'daily_calories' not in st.session_state:
-    st.session_state.daily_calories = 0
-if 'health_status' not in st.session_state:
-    st.session_state.health_status = ""
-if 'status_color' not in st.session_state:
-    st.session_state.status_color = ""
-if 'metrics' not in st.session_state:
-    st.session_state.metrics = {}
-if 'plan_generated' not in st.session_state:
-    st.session_state.plan_generated = False
-if 'plan_content' not in st.session_state:
-    st.session_state.plan_content = ""
-
-# Generate sample progress data for demo purposes
-def generate_sample_data(days=30):
-    start_date = datetime.datetime.now() - datetime.timedelta(days=days)
-    dates = [start_date + datetime.timedelta(days=i) for i in range(days)]
-    # Generate weight data with a slight downward trend
-    initial_weight = 75
-    weights = [initial_weight - 0.1 * i + random.uniform(-0.5, 0.5) for i in range(days)]
-    # Generate calorie data with weekly patterns
-    calories = []
-    for i in range(days):
-        # Weekends tend to have higher calorie intake
-        if i % 7 >= 5:  # Saturday and Sunday
-            calories.append(2200 + random.uniform(-200, 300))
-        else:
-            calories.append(1800 + random.uniform(-100, 200))
-    # Generate steps data
-    steps = [random.randint(6000, 12000) for _ in range(days)]
-    # Create a DataFrame
-    data = pd.DataFrame({
-        'date': dates,
-        'weight': weights,
-        'calories': calories,
-        'steps': steps
-    })
-    return data
-
-# Create a responsive header
-st.markdown("""
-<div class="header-section">
-    <h1 style="color: #4361ee;">AI-Based Personalized Health & Wellness Planner</h1>
-    <p style="font-size: 18px;">Integrating modern science with Ayurvedic wisdom for holistic health</p>
-</div>
-""", unsafe_allow_html=True)
-
-# Create tabs for different sections of the app
-tabs = st.tabs(["📋 Profile & Plan", "📊 Analytics", "ℹ️ About"])
-
-with tabs[0]:
-    # Create a two-column layout for input form and results
-    col1, col2 = st.columns([3, 2])
+# Main App Layout
+def main():
+    # App Header
+    col1, col2 = st.columns([1, 3])
     with col1:
-        st.markdown("""<div class="input-section">""", unsafe_allow_html=True)
-        st.subheader("Personal Details")
-        # Create multiple columns for compact form layout
-        detail_col1, detail_col2 = st.columns(2)
-        with detail_col1:
-            name = st.text_input("Your Name", placeholder="Enter your name")
+        st.image("https://img.icons8.com/color/96/000000/yoga.png", width=80)
+    with col2:
+        st.title("AI-Based Wellness Planner")
+        st.subheader("Personalized Diet & Exercise Plan with Ayurvedic Insights")
+    
+    st.markdown("---")
+    
+    # Sidebar content
+    with st.sidebar:
+        st.image("https://img.icons8.com/color/96/000000/natural-food.png", width=50)
+        st.header("About this Project")
+        st.markdown("""
+        <div class="info-box">
+        This app creates personalized wellness plans by integrating Ayurveda with modern medical science for comprehensive health insights.
+        </div>
+        """, unsafe_allow_html=True)
+        
+        if has_team_info:
+            st.subheader("���� Team Members")
+            team_members = [
+                ("Aditi Soni", "https://www.linkedin.com/in/aditi-soni-259813285/"),
+                ("Bhumika Patel", "https://www.linkedin.com/in/bhumika-patel-ml/"),
+                ("Aditi Lakhera", "https://www.linkedin.com/in/aditi-lakhera-b628802bb/"),
+                ("Anushri Tiwari", "https://www.linkedin.com/in/anushri-tiwari-916494300")
+            ]
+            
+            for name, link in team_members:
+                st.markdown(f"- [{name}]({link})")
+            
+            st.subheader("🔗 Project Links")
+            st.markdown("[GitHub Repository](https://github.com/Abs6187/AI_Health_v2)")
+            st.markdown("[Presentation](https://github.com/Abs6187/AI_Health_v2/blob/main/HackGirl_PPT_HackSRIT.pptx)")
+            st.markdown("[Hackathon](https://unstop.com/hackathons/hacksrit-shri-ram-group-of-institutions-jabalpur-1471613)")
+        
+        st.markdown("---")
+        st.caption("© 2023 Wellness Planner")
+
+    # Main content
+    tab1, tab2 = st.tabs(["Create Your Plan", "About Ayurveda"])
+    
+    with tab1:
+        # Input Form in two columns
+        st.header("Enter Your Details")
+        
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            st.subheader("Personal Information")
+            name = st.text_input("Name", placeholder="Enter your name")
             age = st.number_input("Age", min_value=1, max_value=120, value=25)
             gender = st.selectbox("Gender", options=["Male", "Female", "Other"])
-        with detail_col2:
-            weight = st.number_input("Weight (kg)", min_value=1.0, max_value=300.0, value=70.0)
-            height = st.number_input("Height (cm)", min_value=50.0, max_value=250.0, value=170.0)
-        st.subheader("Health Goals")
-        goal_col1, goal_col2 = st.columns(2)
-        with goal_col1:
+            
+            st.subheader("Physical Metrics")
+            weight = st.number_input("Weight (kg)", min_value=1.0, max_value=300.0, value=70.0, format="%.1f")
+            height = st.number_input("Height (cm)", min_value=1, max_value=250, value=170)
+            
+        with col2:
+            st.subheader("Goals & Preferences")
             fitness_goal = st.selectbox("Fitness Goal", options=["Weight Loss", "Weight Gain", "Maintenance"])
-            dietary_preference = st.selectbox("Dietary Preference", options=["None", "Vegetarian", "Vegan", "Keto", "Halal"])
-        with goal_col2:
-            activity_level = st.selectbox("Activity Level", options=[
-                "Sedentary (little or no exercise)",
-                "Lightly active (light exercise/sports 1-3 days/week)",
-                "Moderately active (moderate exercise/sports 3-5 days/week)",
-                "Very active (hard exercise/sports 6-7 days/week)",
-                "Super active (very hard exercise & physical job or 2x training)"
-            ])
+            dietary_preference = st.selectbox("Dietary Preference", 
+                                         options=["No Restrictions", "Vegetarian", "Vegan", "Keto", "Paleo", "Halal", "Gluten-Free"])
+            food_allergies = st.text_input("Food Allergies (if any)", placeholder="e.g., nuts, dairy, shellfish")
+            local_cuisine = st.text_input("Preferred Local Cuisine", placeholder="e.g., Indian, Italian, Chinese")
+            month = st.selectbox("Current Month", 
+                            options=["January", "February", "March", "April", "May", "June", 
+                                     "July", "August", "September", "October", "November", "December"])
+            
             include_ayurveda = st.checkbox("Include Ayurvedic wellness insights", value=True)
-        with st.expander("Additional Preferences", expanded=False):
-            food_allergies = st.text_input("Food Allergies (if any)", placeholder="e.g., peanuts, shellfish")
-            local_cuisine = st.text_input("Preferred Local Cuisine", value="Indian", placeholder="e.g., Indian, Italian, Chinese")
-            month = st.selectbox("Select Month", options=[
-                "January", "February", "March", "April", "May", "June", 
-                "July", "August", "September", "October", "November", "December"
-            ], index=datetime.datetime.now().month - 1)
-        # Calculate BMI and calorie requirements
-        if st.button("Calculate Health Metrics", use_container_width=True):
-            st.session_state.bmi = round(weight / (height / 100) ** 2, 2)
-            st.session_state.health_status, st.session_state.status_color = get_bmi_category(st.session_state.bmi)
-            st.session_state.daily_calories = int(calculate_calorie_requirements(age, gender, weight, height, fitness_goal, activity_level))
-            # Store metrics in session state
-            st.session_state.metrics = {
-                "name": name,
-                "age": age,
-                "gender": gender,
-                "bmi": st.session_state.bmi,
-                "health_status": st.session_state.health_status,
-                "fitness_goal": fitness_goal,
-                "dietary_preference": dietary_preference,
-                "food_allergies": food_allergies,
-                "daily_calories": st.session_state.daily_calories,
-                "local_cuisine": local_cuisine,
-                "weekdays": ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"],
-                "month": month,
-                "activity_level": activity_level
-            }
-            # Rerun to update the UI
-            st.rerun()
-        # Generate plan button (only enabled if metrics are calculated)
-        if st.session_state.bmi > 0:
-            if st.button("Generate Personalized Plan", type="primary", use_container_width=True):
-                with st.spinner("Creating your personalized health plan..."):
+        
+        # Calculate metrics
+        bmi = round(weight / (height / 100) ** 2, 2)
+        
+        if bmi < 18.5:
+            health_status = "Underweight"
+            bmi_color = "orange"
+        elif bmi <= 24.9:
+            health_status = "Normal weight"
+            bmi_color = "green"
+        elif bmi <= 29.9:
+            health_status = "Overweight"
+            bmi_color = "orange"
+        else:
+            health_status = "Obese"
+            bmi_color = "red"
+            
+        daily_calories = int(calculate_calorie_requirements(age, gender, weight, height, fitness_goal))
+        
+        # Display metrics in a nice formatted box
+        st.markdown("### Your Health Metrics")
+        metrics_col1, metrics_col2 = st.columns(2)
+        
+        with metrics_col1:
+            st.markdown(f"""
+            <div style="background-color: #e8f4f8; padding: 20px; border-radius: 10px; margin-bottom: 20px;">
+                <h4 style="color: #1E88E5;">BMI Analysis</h4>
+                <p>Your BMI is <span style="font-weight: bold; color: {bmi_color};">{bmi}</span></p>
+                <p>Status: <span style="font-weight: bold; color: {bmi_color};">{health_status}</span></p>
+            </div>
+            """, unsafe_allow_html=True)
+            
+        with metrics_col2:
+            st.markdown(f"""
+            <div style="background-color: #e8f4f8; padding: 20px; border-radius: 10px; margin-bottom: 20px;">
+                <h4 style="color: #1E88E5;">Calorie Requirements</h4>
+                <p>Daily requirements: <span style="font-weight: bold;">{daily_calories} kcal</span></p>
+                <p>Goal: <span style="font-weight: bold;">{fitness_goal}</span></p>
+            </div>
+            """, unsafe_allow_html=True)
+        
+        # User metrics dictionary
+        metrics = {
+            "name": name,
+            "age": age,
+            "gender": gender,
+            "bmi": bmi,
+            "health_status": health_status,
+            "fitness_goal": fitness_goal,
+            "dietary_preference": dietary_preference,
+            "food_allergies": food_allergies,
+            "daily_calories": daily_calories,
+            "local_cuisine": local_cuisine,
+            "weekdays": ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"],
+            "month": month,
+        }
+        
+        # Generate button
+        st.markdown("---")
+        generate_col1, generate_col2, generate_col3 = st.columns([1, 2, 1])
+        with generate_col2:
+            generate_plan = st.button("Generate Your Personalized Wellness Plan", use_container_width=True)
+        
+        # Generate and display plan
+        if generate_plan:
+            if not name:
+                st.error("Please enter your name to generate a plan")
+            else:
+                with st.spinner("Crafting your personalized wellness plan..."):
                     try:
                         # Choose the appropriate prompt based on the Ayurveda option
                         selected_prompt = ayurveda_prompt_template if include_ayurveda else regular_prompt_template
-                        plan = generate_plan_with_prompt(st.session_state.metrics, selected_prompt)
-                        st.session_state.plan_content = format_plan(plan)
-                        st.session_state.plan_generated = True
-                        # Show success message
-                        st.success("Your personalized health plan has been generated!")
-                        st.balloons()
-                        # Rerun to update the UI
-                        st.rerun()
+                        plan = generate_plan_with_prompt(metrics, selected_prompt)
+                        formatted_plan = format_plan(plan)
+                        
+                        plan_type = "Integrated Ayurvedic & Modern Wellness Plan" if include_ayurveda else "Modern Wellness Plan"
+                        
+                        st.success(f"Your personalized plan is ready!")
+                        
+                        st.markdown(f"""
+                        <div class="result-container">
+                            <h2 style="text-align: center; color: #2E7D32;">{plan_type} for {name}</h2>
+                            <h4 style="text-align: center; color: #666;">Month: {month}</h4>
+                            <hr>
+                            {formatted_plan.replace('\n', '<br>')}
+                        </div>
+                        """, unsafe_allow_html=True)
+                        
+                        # Download option
+                        plan_text = f"{plan_type} for {name}\nMonth: {month}\n\n{plan.content}"
+                        st.download_button(
+                            label="Download Your Plan",
+                            data=plan_text,
+                            file_name=f"{name}_wellness_plan.txt",
+                            mime="text/plain"
+                        )
+                        
                     except Exception as e:
                         st.error(f"Error generating the plan: {e}")
-        st.markdown("""</div>""", unsafe_allow_html=True)
-    with col2:
-        # Display health metrics if they have been calculated
-        if st.session_state.bmi > 0:
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1); text-align: center; margin-bottom: 20px;">
-                <h3 style="margin-bottom: 20px;">Your Health Metrics</h3>
-            """, unsafe_allow_html=True)
-            metric_col1, metric_col2 = st.columns(2)
-            with metric_col1:
-                st.markdown(f"""
-                <div class="metric-card">
-                    <h4>BMI</h4>
-                    <h2 style="font-size: 2.5rem; margin: 10px 0;">{st.session_state.bmi}</h2>
-                    <span class="badge badge-{st.session_state.status_color}">{st.session_state.health_status}</span>
-                </div>
-                """, unsafe_allow_html=True)
-            with metric_col2:
-                st.markdown(f"""
-                <div class="metric-card">
-                    <h4>Daily Calories</h4>
-                    <h2 style="font-size: 2.5rem; margin: 10px 0;">{st.session_state.daily_calories}</h2>
-                    <span class="badge badge-blue">kcal</span>
-                </div>
-                """, unsafe_allow_html=True)
-            st.markdown("""</div>""", unsafe_allow_html=True)
-            # Display recommendations based on health status
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1); margin-bottom: 20px;">
-                <h3>Health Recommendations</h3>
-            """, unsafe_allow_html=True)
-            if st.session_state.status_color == "green":
-                st.markdown("""
-                <p>✅ Your BMI is in the healthy range!</p>
-                <p>🔍 Focus on maintaining your current weight with a balanced diet and regular exercise.</p>
-                <p>💪 Strength training 2-3 times per week is recommended for muscle maintenance.</p>
-                <p>🥗 Continue eating a variety of nutritious foods to maintain your overall health.</p>
-                """, unsafe_allow_html=True)
-            elif st.session_state.status_color == "yellow":
-                st.markdown("""
-                <p>⚠️ Your BMI indicates you may benefit from some adjustments to your lifestyle.</p>
-                <p>🏃‍♂️ Regular cardio exercise (30-45 minutes, 4-5 days per week) is recommended.</p>
-                <p>🍽️ Focus on portion control and balanced nutrition.</p>
-                <p>📱 Consider tracking your food intake and exercise for better awareness.</p>
-                """, unsafe_allow_html=True)
-            elif st.session_state.status_color == "red":
-                st.markdown("""
-                <p>❗ Your BMI suggests you may benefit from consulting with a healthcare professional.</p>
-                <p>👨‍⚕️ Consider seeking advice from a physician or registered dietitian.</p>
-                <p>👣 Start with gentle, low-impact exercise like walking or swimming.</p>
-                <p>🥦 Focus on whole, unprocessed foods rich in nutrients.</p>
-                <p>🧘‍♀️ Stress management techniques may also support your health journey.</p>
-                """, unsafe_allow_html=True)
-            st.markdown("""</div>""", unsafe_allow_html=True)
-            # Display tips based on fitness goal
-            st.markdown(f"""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);">
-                <h3>{st.session_state.metrics.get('fitness_goal', '')} Tips</h3>
-            """, unsafe_allow_html=True)
-            if st.session_state.metrics.get('fitness_goal') == "Weight Loss":
-                st.markdown("""
-                <p>🔻 Focus on creating a moderate calorie deficit (500-750 kcal/day).</p>
-                <p>⏱️ Consider intermittent fasting methods like 16:8 that may support your goals.</p>
-                <p>🚶‍♀️ Increase NEAT (Non-Exercise Activity Thermogenesis) through daily movement.</p>
-                <p>🧠 Manage stress as it can contribute to weight gain.</p>
-                """, unsafe_allow_html=True)
-            elif st.session_state.metrics.get('fitness_goal') == "Weight Gain":
-                st.markdown("""
-                <p>🔺 Aim for a calorie surplus of 300-500 kcal/day.</p>
-                <p>🏋️ Focus on progressive overload in strength training to build muscle.</p>
-                <p>🥛 Include protein-rich foods and consider protein supplementation if needed.</p>
-                <p>😴 Prioritize quality sleep as it's essential for muscle recovery and growth.</p>
-                """, unsafe_allow_html=True)
-            elif st.session_state.metrics.get('fitness_goal') == "Maintenance":
-                st.markdown("""
-                <p>⚖️ Monitor your weight weekly to ensure you're maintaining.</p>
-                <p>🏃‍♂️ Mix up your exercise routine to prevent plateaus and maintain interest.</p>
-                <p>🥗 Practice the 80/20 rule: 80% nutritious foods, 20% treats in moderation.</p>
-                <p>📊 Adjust calorie intake based on activity levels throughout the week.</p>
-                """, unsafe_allow_html=True)
-            st.markdown("""</div>""", unsafe_allow_html=True)
-        else:
-            # Display a welcome message if no metrics have been calculated yet
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 30px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1); text-align: center; height: 100%;">
-                <img src="https://www.svgrepo.com/show/513354/yoga.svg" width="100">
-                <h2 style="margin: 20px 0;">Welcome to Your Health Journey</h2>
-                <p style="font-size: 16px; margin-bottom: 20px;">Fill in your details and click 'Calculate Health Metrics' to start your personalized wellness plan.</p>
-                <p style="font-style: italic;">"The greatest wealth is health." - Virgil</p>
-            </div>
-            """, unsafe_allow_html=True)
-    # Display the generated plan if available
-    if st.session_state.plan_generated:
-        st.markdown("""
-        <div class="plan-section">
-            <h2 style="text-align: center; margin-bottom: 20px;">Your Personalized Health Plan</h2>
-        """, unsafe_allow_html=True)
-        st.markdown(st.session_state.plan_content, unsafe_allow_html=True)
-        st.markdown("""</div>""", unsafe_allow_html=True)
-
-with tabs[1]:
-    st.markdown("""
-    <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); margin-bottom: 20px;">
-        <h2 style="text-align: center;">Health Progress Analytics</h2>
-    </div>
-    """, unsafe_allow_html=True)
-    # Generate sample data for demo purposes
-    sample_data = generate_sample_data()
-    # Choose analysis timeframe
-    timeframe = st.radio("Select Timeframe", options=["7 Days", "14 Days", "30 Days"], horizontal=True)
-    days = int(timeframe.split()[0])
-    filtered_data = sample_data.iloc[-days:]
-    # Create multiple charts
-    chart_col1, chart_col2 = st.columns(2)
-    with chart_col1:
-        st.subheader("Weight Progress")
-        weight_chart = alt.Chart(filtered_data).mark_line(point=True).encode(
-            x=alt.X('date:T', title='Date'),
-            y=alt.Y('weight:Q', title='Weight (kg)', scale=alt.Scale(zero=False)),
-            tooltip=['date:T', 'weight:Q']
-        ).properties(height=300).interactive()
-        st.altair_chart(weight_chart, use_container_width=True)
-    with chart_col2:
-        st.subheader("Daily Calorie Intake")
-        calorie_chart = alt.Chart(filtered_data).mark_bar().encode(
-            x=alt.X('date:T', title='Date'),
-            y=alt.Y('calories:Q', title='Calories (kcal)'),
-            color=alt.condition(
-                alt.datum.calories > 2000,
-                alt.value('#ffaa00'),
-                alt.value('#4361ee')
-            ),
-            tooltip=['date:T', 'calories:Q']
-        ).properties(height=300).interactive()
-        st.altair_chart(calorie_chart, use_container_width=True)
-    st.subheader("Daily Steps")
-    steps_chart = alt.Chart(filtered_data).mark_area(
-        line={'color':'#4361ee'},
-        color=alt.Gradient(
-            gradient='linear',
-            stops=[alt.GradientStop(color='white', offset=0),
-                   alt.GradientStop(color='#dbeafe', offset=1)],
-            x1=1,
-            x2=1,
-            y1=1,
-            y2=0
-        )
-    ).encode(
-        x=alt.X('date:T', title='Date'),
-        y=alt.Y('steps:Q', title='Steps'),
-        tooltip=['date:T', 'steps:Q']
-    ).properties(height=250).interactive()
-    st.altair_chart(steps_chart, use_container_width=True)
-    # Display key metrics
-    metric_cols = st.columns(4)
-    with metric_cols[0]:
-        st.metric("Average Weight", f"{filtered_data['weight'].mean():.1f} kg", f"{filtered_data['weight'].iloc[-1] - filtered_data['weight'].iloc[0]:.1f} kg")
-    with metric_cols[1]:
-        st.metric("Average Calories", f"{int(filtered_data['calories'].mean())} kcal")
-    with metric_cols[2]:
-        st.metric("Average Steps", f"{int(filtered_data['steps'].mean())}", f"{int(filtered_data['steps'].iloc[-1] - filtered_data['steps'].iloc[0])}")
-    with metric_cols[3]:
-        if st.session_state.daily_calories > 0:
-            calorie_diff = int(filtered_data['calories'].mean() - st.session_state.daily_calories)
-            st.metric("vs. Target Calories", f"{int(st.session_state.daily_calories)} kcal", f"{calorie_diff} kcal")
-        else:
-            st.metric("Calorie Goal Status", "Not set")
-
-with tabs[2]:
-    st.markdown("""
-    <div style="background-color: white; border-radius: 8px; padding: 30px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); margin-bottom: 20px;">
-        <h2 style="text-align: center;">About This Project</h2>
-        <p style="font-size: 16px; margin: 20px 0;">
-            The AI-Based Personalized Health & Wellness Planner is a cutting-edge application that integrates modern scientific knowledge with traditional Ayurvedic wisdom to create comprehensive, personalized wellness plans.
-        </p>
-        <ul style="margin-left: 20px; font-size: 16px;">
-            <li>Personalized diet and exercise recommendations based on your unique profile</li>
-            <li>Optional Ayurvedic insights tailored to your body type (dosha)</li>
-            <li>Integration with food delivery platforms for convenient meal ordering</li>
-            <li>Wearable device tracking to monitor your progress</li>
-            <li>Adaptive recommendations based on seasonal factors</li>
-        </ul>
-    </div>
-    """, unsafe_allow_html=True)
-    # Display team information if available
-    if has_team_info:
+    
+    with tab2:
+        st.header("Ayurveda: Ancient Wisdom for Modern Wellness")
+        
         st.markdown("""
-        <div style="background-color: white; border-radius: 8px; padding: 30px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);">
-            <h2 style="text-align: center;">Our Team</h2>
+        <div style="background-color: white; padding: 20px; border-radius: 10px; margin-bottom: 20px;">
+            <p>Ayurveda is a traditional Indian system of medicine that dates back over 5,000 years. The term Ayurveda is derived from the Sanskrit words "ayur" (life) and "veda" (knowledge).</p>
+            
+            <h3>The Three Doshas</h3>
+            <p>Ayurveda believes that each person has a unique mix of three energies or doshas:</p>
+            <ul>
+                <li><strong>Vata</strong> (Air & Space) - Controls movement and is associated with creativity and flexibility</li>
+                <li><strong>Pitta</strong> (Fire & Water) - Controls metabolism and is associated with intelligence and determination</li>
+                <li><strong>Kapha</strong> (Earth & Water) - Controls structure and is associated with stability and compassion</li>
+            </ul>
+            
+            <h3>Ayurvedic Approach to Wellness</h3>
+            <p>Ayurveda takes a holistic approach to health that includes:</p>
+            <ul>
+                <li>Personalized nutrition based on your dosha</li>
+                <li>Lifestyle recommendations aligned with natural cycles</li>
+                <li>Herbal medicines and therapies</li>
+                <li>Mental wellness practices</li>
+                <li>Seasonal adjustments to maintain balance</li>
+            </ul>
+            
+            <p>Our wellness planner integrates these ancient principles with modern health science to provide you with a truly holistic approach to health and wellness.</p>
         </div>
         """, unsafe_allow_html=True)
-        # Display team members in a grid
-        team_col1, team_col2 = st.columns(2)
-        with team_col1:
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); margin: 10px 0; text-align: center;">
-                <img src="https://www.svgrepo.com/show/382106/female-avatar-girl-face-woman-user-5.svg" width="100" style="border-radius: 50%;">
-                <h3>Aditi Soni</h3>
-                <p>Project Lead & AI Integration</p>
-                <a href="https://www.linkedin.com/in/aditi-soni-259813285/" target="_blank">LinkedIn Profile</a>
-            </div>
-            """, unsafe_allow_html=True)
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); margin: 10px 0; text-align: center;">
-                <img src="https://www.svgrepo.com/show/382105/female-avatar-girl-face-woman-user-4.svg" width="100" style="border-radius: 50%;">
-                <h3>Aditi Lakhera</h3>
-                <p>Nutrition Science Expert</p>
-                <a href="https://www.linkedin.com/in/aditi-lakhera-b628802bb/" target="_blank">LinkedIn Profile</a>
-            </div>
-            """, unsafe_allow_html=True)
-        with team_col2:
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); margin: 10px 0; text-align: center;">
-                <img src="https://www.svgrepo.com/show/382101/female-avatar-girl-face-woman-user.svg" width="100" style="border-radius: 50%;">
-                <h3>Bhumika Patel</h3>
-                <p>Machine Learning Engineer</p>
-                <a href="https://www.linkedin.com/in/bhumika-patel-ml/" target="_blank">LinkedIn Profile</a>
-            </div>
-            """, unsafe_allow_html=True)
-            st.markdown("""
-            <div style="background-color: white; border-radius: 8px; padding: 20px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05); margin: 10px 0; text-align: center;">
-                <img src="https://www.svgrepo.com/show/382103/female-avatar-girl-face-woman-user-2.svg" width="100" style="border-radius: 50%;">
-                <h3>Anushri Tiwari</h3>
-                <p>UI/UX Designer</p>
-                <a href="https://www.linkedin.com/in/anushri-tiwari-916494300" target="_blank">LinkedIn Profile</a>
-            </div>
-            """, unsafe_allow_html=True)
 
-# Add a modern footer
-st.markdown("""
-<div style="text-align: center; margin-top: 30px; padding: 20px; font-size: 14px; color: #6b7280;">
-    <p>© 2025 AI Health & Wellness Planner. Current version: 2.1.0</p>
-    <p>Last updated: May 10, 2025</p>
-</div>
-""", unsafe_allow_html=True)
+if __name__ == "__main__":
+    main()
+