Update app.py

app.py

@@ -1,207 +1,218 @@
-import streamlit as st
+import os
+import random
 import pandas as pd
+import streamlit as st
+from groq import Groq
 import gdown
-from PIL import Image
-
-# Function to load data from Google Drive
-def load_data(file_url):
-    file_path = "menu_data.xlsx"
-    gdown.download(file_url, file_path, quiet=False)
-    return pd.read_excel(file_path, engine='openpyxl')
-
-# Function to generate the weekly menu
-def generate_menu(data, season, sub_category_pref, outings, budget):
-    season_data = data[(data['Season'] == season) | (data['Season'] == 'All Seasons')]
-
-    meals_needed = 14 - outings  # Main Courses (14) minus outings
-
-    menu = []
-    for sub_category, freq in sub_category_pref.items():
-        sub_category_dishes = season_data[season_data['Sub-Category'] == sub_category]
-        menu.extend(sub_category_dishes.sample(min(freq, len(sub_category_dishes))).to_dict('records'))
-
-    menu = menu[:meals_needed]
-
-    outing_dishes = data[data['Home Made / Outing'] == 'Outing'].sample(outings).to_dict('records')
-    menu.extend(outing_dishes)
-
-    side_dishes = data[data['Course'] == 'Side Dish'].sample(2).to_dict('records')
-    menu.extend(side_dishes)
-
-    menu_df = pd.DataFrame(menu)
-
-    lunch_menu = menu_df.iloc[:len(menu_df)//2]
-    dinner_menu = menu_df.iloc[len(menu_df)//2:]
-
-    total_cost = menu_df["Cost per 4 persons"].sum()
-
-    if total_cost > budget:
-        st.warning(f"The total cost of the menu exceeds your budget of PKR {budget}! Please adjust your preferences.")
-        return None, None, None
-
-    lunch_menu = lunch_menu[["Name"]]
-    dinner_menu = dinner_menu[["Name"]]
-
-    return lunch_menu, dinner_menu, total_cost
-
-# Function to calculate BMI and recommend diet for adults
-def bmi_insights(weight, height_cm):
-    height_m = height_cm / 100
-    bmi = weight / (height_m ** 2)
-
-    if bmi < 18.5:
-        return f"Your BMI is {bmi:.2f}. You are underweight. A diet rich in proteins and healthy fats would be beneficial."
-    elif 18.5 <= bmi < 24.9:
-        return f"Your BMI is {bmi:.2f}. You are in the normal weight range. Maintain a balanced diet."
-    elif 25 <= bmi < 29.9:
-        return f"Your BMI is {bmi:.2f}. You are overweight. A diet with fewer carbs and more lean proteins can help manage your weight."
-    else:
-        return f"Your BMI is {bmi:.2f}. You are obese. A calorie-restricted diet with a focus on whole foods and lean proteins is recommended."
-
-# Function to provide general diet recommendation for infants
-def infant_diet_recommendation():
-    return (
-        "For infants (up to 12 months), breast milk or formula should be the primary source of nutrition. "
-        "Around 6 months, you can start introducing solid foods, including pureed fruits, vegetables, and cereals. "
-        "Consult with your pediatrician for more personalized advice."
-    )
-
-# Function to provide dietary recommendations based on medical conditions
-def medical_condition_recommendation(condition):
-    recommendations = {
-        "Diabetes": "A low-carb diet is essential. Focus on fiber-rich vegetables, lean proteins, and healthy fats. Avoid sugary foods.",
-        "Celiac Disease": "Avoid gluten-containing foods like wheat, barley, and rye. Focus on gluten-free grains, fruits, vegetables, and lean proteins.",
-        "Hypertension": "Limit salt intake. Incorporate potassium-rich foods such as bananas, spinach, and potatoes. A heart-healthy diet with lean proteins is ideal.",
-        "Heart Disease": "Consume a heart-healthy diet rich in omega-3 fatty acids, vegetables, and whole grains. Limit saturated fats and processed foods.",
-        "Obesity": "Focus on a calorie-restricted diet, prioritizing vegetables, lean meats, and whole grains. Avoid sugary snacks and drinks.",
-        "Cholesterol Disorders": "Eat foods rich in fiber, such as oats, fruits, and vegetables. Limit saturated fats and trans fats.",
-        "IBS": "Incorporate fiber-rich foods like oats and vegetables. Avoid foods that cause bloating, such as beans, cabbage, and onions.",
-        "GERD": "Avoid acidic and spicy foods, caffeine, and alcohol. Focus on bland, easy-to-digest meals like rice, oatmeal, and non-citrus fruits.",
-        "Peptic Ulcers": "Avoid spicy, fatty, and acidic foods. Focus on bland foods like rice, oatmeal, and non-citrus fruits.",
-        "Gout": "Limit foods high in purines, such as red meat, shellfish, and alcohol. Focus on low-fat dairy, fruits, and vegetables.",
-        "Lactose Intolerance": "Avoid dairy products or choose lactose-free alternatives. Incorporate plant-based sources of calcium.",
-        "Kidney Disease": "Reduce sodium, potassium, and phosphorus intake. Focus on lean proteins and avoid processed foods.",
-        "Hypothyroidism": "Increase iodine-rich foods like seaweed, eggs, and dairy. Focus on vegetables and lean proteins.",
-        "Anemia": "Focus on iron-rich foods such as red meat, spinach, and beans. Pair with vitamin C-rich foods for better absorption.",
-        "Food Allergies": "Avoid allergenic foods like peanuts, tree nuts, dairy, eggs, shellfish, and others. Focus on a balanced, allergy-free diet.",
-        "Acid Reflux": "Avoid acidic and spicy foods. Incorporate alkaline foods like bananas, melons, and oatmeal.",
-        "Crohn's Disease": "Focus on low-fiber foods and avoid spicy, fatty, and acidic foods. Prioritize lean meats and easy-to-digest grains.",
-        "Osteoporosis": "Ensure adequate calcium and vitamin D intake through dairy or fortified alternatives. Incorporate weight-bearing exercises.",
-        "Autoimmune Diseases": "A diet rich in anti-inflammatory foods like leafy greens, berries, nuts, and fatty fish is ideal.",
-        "Migraines": "Avoid foods that trigger migraines, such as chocolate, caffeine, and aged cheeses. Focus on magnesium-rich foods.",
-        "PCOS": "Focus on a low-glycemic index diet with plenty of fruits, vegetables, and lean proteins. Limit processed sugars.",
-        "CFS": "A balanced diet with adequate calories and nutrients is essential. Focus on anti-inflammatory foods and avoid refined sugars.",
-        "Mental Health": "Focus on omega-3 fatty acids, whole grains, fruits, and vegetables. Avoid processed foods and sugar."
-    }
-    return recommendations.get(condition, "No specific recommendations available.")
-
-# Main workflow
-def main():
-    # Set the background color and text color for the app
-    st.markdown("""
-        <style>
-            body {
-                background-color: navy;
-                color: white;
-            }
-        </style>
-    """, unsafe_allow_html=True)
-
-    st.title("Roz Roz Ka Masla..... Aaj Kya Pakayen!!! ")
-    st.title("Now generate weekly Menu with BMI Insights")
-
-    # Add food image at the top right of the app
-    image = Image.open("food_image.jpg")  # Ensure you have an image in the same folder
-    st.sidebar.image(image, use_container_width=True)
-
-    # User input for BMI insights or Infant's weekly diet
-    st.sidebar.header("Select your category:")
-    category = st.sidebar.radio("Are you an Adult or Infant?", ["Adult", "Infant"])
-
-    if category == "Infant":
-        st.sidebar.header("Infant Weekly Diet Recommendations")
-        diet_recommendation = infant_diet_recommendation()
-        st.sidebar.write(diet_recommendation)
-        return
-
-    if category == "Adult":
-        # User input for BMI insights (only for Adults)
-        st.sidebar.header("Enter your details for BMI insights:")
-        weight = st.sidebar.number_input("Enter your weight (kg)", min_value=30, max_value=200, value=70)
-        height_cm = st.sidebar.number_input("Enter your height (cm)", min_value=90, max_value=250, value=175)
-
-        if st.sidebar.button("Get BMI Insights"):
-            bmi_message = bmi_insights(weight, height_cm)
-            st.sidebar.success(bmi_message)
-
-        # Dropdown for selecting medical condition(s)
-        medical_condition = st.sidebar.selectbox(
-            "Select any medical condition", [
-                "None", "Diabetes", "Celiac Disease", "Hypertension", "Heart Disease", "Obesity", 
-                "Cholesterol Disorders", "IBS", "GERD", "Peptic Ulcers", "Gout", "Lactose Intolerance", 
-                "Kidney Disease", "Hypothyroidism", "Anemia", "Food Allergies", "Acid Reflux", 
-                "Crohn's Disease", "Osteoporosis", "Autoimmune Diseases", "Migraines", "PCOS", 
-                "CFS", "Mental Health"
-            ]
+
+# Function to download the file from Google Drive
+def download_file_from_gdrive(url, output):
+    import gdown
+    gdown.download(url, output, quiet=False)
+
+# Load the menu data from the downloaded file
+def load_data(file_path):
+    return pd.read_excel(file_path)
+
+# Get user inputs for menu generation
+def get_user_inputs():
+    # Get user input via Streamlit widgets
+    user_type = st.selectbox("Is this menu for Adult or Infant?", ["Adult", "Infant"])
+
+    if user_type == "Adult":
+        season = st.selectbox("Select the season", ["Summer", "Winter"])
+        outings = st.slider("Enter the number of outings for the week:", 0, 7, 0)
+        budget = st.number_input("Enter your weekly budget (in Rs):", min_value=0, step=100)
+
+        sub_category_prefs = {}
+        categories = ["Sabzi", "Rice", "Daal", "Meat", "Chicken", "Fish"]
+        for category in categories:
+            count = st.number_input(f"Enter the number of {category} dishes:", min_value=0)
+            sub_category_prefs[category] = count
+
+        return user_type, season, outings, budget, sub_category_prefs
+
+    elif user_type == "Infant":
+        return user_type, None, None, None, None
+
+# Fetch recipe from Groq API for a given dish name
+def get_recipe(dish_name):
+    client = Groq(api_key="your-api-key-here")  # Replace with your actual API key
+
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[{
+                "role": "user",
+                "content": f"Provide a detailed recipe for {dish_name}.",
+            }],
+            model="llama-3.3-70b-versatile",
         )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"Error fetching recipe: {e}"
+
+# Generate menu based on user inputs
+def generate_menu(data, user_type, season=None, outings=None, budget=None, sub_category_prefs=None):
+    if user_type == "Adult":
+        # Filter data by season
+        filtered_data = data[data['Season'].isin([season, 'All Seasons'])]
+        main_courses = filtered_data[filtered_data['Course'] == 'Main Course']
+        side_dishes = filtered_data[filtered_data['Course'] == 'Side Dish']
+
+        # Initialize menu
+        homemade_dishes_count = 14 - outings
+        weekly_menu = []
+
+        # Add dishes from each sub-category based on user preferences
+        for category, count in sub_category_prefs.items():
+            category_dishes = main_courses[main_courses['Sub-Category'] == category]
+            available_count = len(category_dishes)
+
+            while count > available_count:
+                st.warning(f"You requested {count} {category} dishes, but only {available_count} are available.")
+                count = st.number_input(f"Please revise your selection for {category} dishes (Maximum: {available_count}):", min_value=0, max_value=available_count)
+                sub_category_prefs[category] = count
+
+            selected_dishes = category_dishes.sample(n=count).to_dict('records')
+            weekly_menu.extend(selected_dishes)
+
+        # Fill remaining dishes if fewer than 14 are selected
+        remaining_count = homemade_dishes_count - len(weekly_menu)
+        if remaining_count > 0:
+            additional_dishes = main_courses.sample(n=min(remaining_count, len(main_courses))).to_dict('records')
+            weekly_menu.extend(additional_dishes)
+
+        # Randomly select outings
+        available_outings = len(main_courses)
+        if outings > available_outings:
+            st.warning(f"You requested {outings} outings, but only {available_outings} are available.")
+            outings = st.number_input(f"Please revise your selection for outings (Maximum: {available_outings}):", min_value=0, max_value=available_outings)
+        outing_dishes = main_courses.sample(n=min(outings, len(main_courses))).to_dict('records')
+
+        # Combine homemade and outing dishes
+        weekly_menu = random.sample(weekly_menu, k=homemade_dishes_count) + outing_dishes
+
+        # Add 2 side dishes
+        selected_side_dishes = side_dishes.sample(n=min(len(side_dishes), 2)).to_dict('records')
+
+        # Ensure budget constraint
+        total_expenditure = sum(dish['Cost per 4 persons'] for dish in weekly_menu + selected_side_dishes)
+        if total_expenditure > budget:
+            st.warning(f"Your budget of Rs. {budget} is too low for the selected preferences.")
+            return None, None, None, None
+
+        # Split menu into lunch and dinner
+        lunch_menu = weekly_menu[:7]
+        dinner_menu = weekly_menu[7:14]
+
+        # Assign days of the week
+        days_of_week = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']
+        lunch_menu = {days_of_week[i]: dish['Name'] for i, dish in enumerate(lunch_menu)}
+        dinner_menu = {days_of_week[i]: dish['Name'] for i, dish in enumerate(dinner_menu)}
+
+        return lunch_menu, dinner_menu, selected_side_dishes, total_expenditure
+
+    elif user_type == "Infant":
+        baby_dishes_data = data[data['Course'] == 'Baby Dish']
+        baby_dishes = baby_dishes_data['Name'].tolist()
+        if len(baby_dishes) >= 14:
+            infant_menu = random.sample(baby_dishes, k=14)
+        else:
+            infant_menu = random.choices(baby_dishes, k=14)
+
+        # Calculate total expenditure for the infant menu
+        baby_dishes_cost = {row['Name']: row['Cost per 4 persons'] for _, row in baby_dishes_data.iterrows()}
+        total_expenditure = sum(baby_dishes_cost[dish] for dish in infant_menu)
+
+        return {"Weekly Menu": infant_menu}, total_expenditure
+
+# Save the generated menu to an Excel file
+def save_menu_to_excel(lunch_menu=None, dinner_menu=None, side_dishes=None, infant_menu=None):
+    with pd.ExcelWriter("Weekly_Menu.xlsx") as writer:
+        if infant_menu:
+            infant_df = pd.DataFrame(infant_menu["Weekly Menu"], columns=['Dish'])
+            infant_df.to_excel(writer, sheet_name='Infant Menu', index=False)
+        else:
+            lunch_df = pd.DataFrame(list(lunch_menu.items()), columns=['Day', 'Lunch'])
+            lunch_df.to_excel(writer, sheet_name='Lunch Menu', index=False)
+
+            dinner_df = pd.DataFrame(list(dinner_menu.items()), columns=['Day', 'Dinner'])
+            dinner_df.to_excel(writer, sheet_name='Dinner Menu', index=False)
+
+            side_dishes_df = pd.DataFrame(side_dishes)
+            side_dishes_df = side_dishes_df[['Name']]
+            side_dishes_df.to_excel(writer, sheet_name='Side Dishes', index=False)
+
+# Main function to run the app
+def main():
+    # File URL and output path
+    file_url = "https://drive.google.com/uc?id=1BJFao3C6p8k3_KjLfmDo5KbNMrTs7MRo"
+    output_file = "menu_data.xlsx"
+
+    # Download the file
+    download_file_from_gdrive(file_url, output_file)
+
+    # Load data
+    data = load_data(output_file)
 
-        if medical_condition != "None":
-            condition_recommendation = medical_condition_recommendation(medical_condition)
-            st.sidebar.write(condition_recommendation)
-
-    # Google Drive file URL
-    file_url = 'https://drive.google.com/uc?id=1BJFao3C6p8k3_KjLfmDo5KbNMrTs7MRo'  # Replace with actual file link
-    data = load_data(file_url)
-
-    # Main menu generation section (for Adults)
-    if category == "Adult":
-        season = st.selectbox("Select the current season", ["Summer", "Winter"])
-        outings = st.number_input("Enter the number of outings this week", min_value=0, max_value=14, value=0)
-        budget = st.number_input("Enter your budget for the week (PKR)", min_value=0, value=1000)
-
-        st.header("How many dishes from each sub-category would you like?")
-        sub_category_pref = {
-            'Sabzi': st.number_input("Number of Sabzi dishes", min_value=0, value=0),
-            'Rice': st.number_input("Number of Rice dishes", min_value=0, value=0),
-            'Daal': st.number_input("Number of Daal dishes", min_value=0, value=0),
-            'Meat': st.number_input("Number of Meat dishes", min_value=0, value=0),
-            'Chicken': st.number_input("Number of Chicken dishes", min_value=0, value=0),
-            'Fish': st.number_input("Number of Fish dishes", min_value=0, value=0)
-        }
-
-        total_main_courses = sum(sub_category_pref.values()) + outings
-        if total_main_courses > 14:
-            st.warning("The total number of main courses exceeds 14! Please adjust your preferences.")
-            return
-
-        if st.button("Generate Menu"):
-            lunch_menu, dinner_menu, total_cost = generate_menu(data, season, sub_category_pref, outings, budget)
-
-            if lunch_menu is None or dinner_menu is None:
+    st.title("Weekly Menu Generator")
+
+    # Option selection
+    choice = st.selectbox("What would you like to do?", ["Generate a Weekly Menu", "Get a Recipe for a Dish", "Exit"])
+
+    if choice == "Generate a Weekly Menu":
+        # Get user inputs
+        user_type, season, outings, budget, sub_category_prefs = get_user_inputs()
+
+        # Generate menu
+        if user_type == "Adult":
+            lunch_menu, dinner_menu, side_dishes, total_expenditure = generate_menu(
+                data, user_type, season, outings, budget, sub_category_prefs
+            )
+            if lunch_menu is None:
+                st.warning("Unable to generate a menu within the given budget and preferences. Please try again.")
                 return
 
-            st.header("Weekly Menu")
-            st.subheader("Lunch Menu (Day 1 - Day 7)")
+            # Display results
+            st.subheader("Lunch Menu")
+            for day, dish in lunch_menu.items():
+                st.write(f"{day}: {dish}")
 
-            # Assign days of the week to the menu
-            days_of_week = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
-            lunch_menu['Day'] = days_of_week[:len(lunch_menu)]
+            st.subheader("Dinner Menu")
+            for day, dish in dinner_menu.items():
+                st.write(f"{day}: {dish}")
 
-            st.write(lunch_menu)
+            st.subheader("Side Dishes")
+            for dish in side_dishes:
+                st.write(f"- {dish['Name']}")
 
-            st.subheader("Dinner Menu (Day 1 - Day 7)")
+            st.write(f"Total Weekly Expenditure: Rs. {total_expenditure}")
 
-            dinner_menu['Day'] = days_of_week[:len(dinner_menu)]
-            st.write(dinner_menu)
+            # Save menu to Excel
+            save_menu_to_excel(lunch_menu, dinner_menu, side_dishes)
 
-            st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")
+        elif user_type == "Infant":
+            infant_menu, total_expenditure = generate_menu(data, user_type)
+
+            # Display results
+            st.subheader("Infant Weekly Menu")
+            for dish in infant_menu["Weekly Menu"]:
+                st.write(f"- {dish}")
+
+            st.write(f"Total Weekly Expenditure: Rs. {total_expenditure}")
 
             # Save menu to Excel
-            with pd.ExcelWriter("weekly_menu_split.xlsx") as writer:
-                lunch_menu.to_excel(writer, sheet_name="Lunch", index=False)
-                dinner_menu.to_excel(writer, sheet_name="Dinner", index=False)
+            save_menu_to_excel(infant_menu=infant_menu)
+
+    elif choice == "Get a Recipe for a Dish":
+        dish_name = st.text_input("Enter the name of a dish you want the recipe for:")
+        if dish_name:
+            recipe = get_recipe(dish_name)
+            st.subheader(f"Recipe for {dish_name}")
+            st.write(recipe)
 
-            st.success("Weekly menu saved as 'weekly_menu_split.xlsx'")
+    elif choice == "Exit":
+        st.write("Thank you for using the Weekly Menu Generator!")
 
 if __name__ == "__main__":
-    main()
+    main()