Update app.py

app.py

@@ -1,40 +1,193 @@
-# Hello 
+# app.py
+
 import os
 import streamlit as st
 from groq import Groq
 
-# Load your GROQ API Key (set this in Hugging Face Secrets)
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+# ----------------------------
+# CONFIG
+# ----------------------------
+st.set_page_config(
+    page_title="Diet Pal",
+    page_icon="🥗",
+    layout="centered"
+)
 
+# ----------------------------
+# API KEY
+# ----------------------------
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
-st.set_page_config(page_title="AI Recipe Generator", layout="centered")
-st.title("👨‍🍳 Smart Recipe Generator")
-st.caption("Tell me your ingredients and get 3 recipe options by calories!")
-
-# Step 1: User input
-ingredients = st.text_input("🥕 What ingredients do you have?", placeholder="e.g. chicken, tomato, rice")
-preferred_cuisine = st.text_input("🌍 What type of cuisine do you prefer?", placeholder="Optional: Chinese, Asian, Italian...")
-
-if st.button("🍽️ Generate Recipes") and ingredients.strip():
-    with st.spinner("Thinking of delicious recipes..."):
-        prompt = f"""
-You are a smart recipe generator AI. The user has these ingredients: {ingredients}.
-They prefer this cuisine: {preferred_cuisine if preferred_cuisine else 'any'}.
-You must decide what types of dishes can be made and suggest 3 recipes:
-1. One with low calories
-2. One with moderate calories
-3. One with high calories
-Give recipe names, steps, and estimated calories.
+# ----------------------------
+# TITLE
+# ----------------------------
+st.title("🥗 Diet Pal")
+st.caption("Your AI Nutrition Assistant | Recipes + Diet Plans")
+
+# ----------------------------
+# MAIN MENU
+# ----------------------------
+feature = st.radio(
+    "Choose an option:",
+    ["🍳 Recipe Generator", "📋 Diet Planner"]
+)
+
+# ==================================================
+# RECIPE GENERATOR
+# ==================================================
+if feature == "🍳 Recipe Generator":
+
+    st.markdown("## 🍽️ Smart Recipe Generator")
+
+    ingredients = st.text_input(
+        "🥕 What ingredients do you have?",
+        placeholder="e.g. chicken, tomato, rice"
+    )
+
+    cuisine = st.text_input(
+        "🌍 What type of cuisine do you prefer?",
+        placeholder="Optional: Chinese, Pakistani, Italian..."
+    )
+
+    if st.button("🍳 Generate Recipes"):
+
+        if ingredients.strip():
+
+            with st.spinner("Thinking of delicious recipes..."):
+
+                prompt = f"""
+You are Diet Pal AI.
+
+The user has these ingredients:
+{ingredients}
+
+Preferred cuisine:
+{cuisine if cuisine else "Any"}
+
+Generate 3 recipes:
+
+1. Low calorie recipe
+2. Moderate calorie recipe
+3. High calorie recipe
+
+For each recipe give:
+
+- Recipe Name
+- Estimated Calories
+- Ingredients
+- Steps
+- Protein / Carbs / Fats
+"""
+
+                chat_completion = client.chat.completions.create(
+                    messages=[
+                        {"role": "user", "content": prompt}
+                    ],
+                    model="llama-3.3-70b-versatile"
+                )
+
+                result = chat_completion.choices[0].message.content
+
+                st.success("🍳 Recipes Generated!")
+                st.markdown("### 📋 Your Recipes")
+                st.markdown(result)
+
+        else:
+            st.warning("Please enter ingredients first.")
+
+# ==================================================
+# DIET PLANNER
+# ==================================================
+elif feature == "📋 Diet Planner":
+
+    st.markdown("## 📋 Personal Diet Planner")
+
+    goal = st.radio(
+        "🎯 What is your goal?",
+        ["Lose Weight", "Gain Weight"]
+    )
+
+    age = st.number_input("🎂 Age", min_value=10, max_value=100, value=20)
+
+    weight = st.number_input(
+        "⚖️ Weight (kg)",
+        min_value=20,
+        max_value=250,
+        value=70
+    )
+
+    height = st.number_input(
+        "📏 Height (cm)",
+        min_value=100,
+        max_value=250,
+        value=170
+    )
+
+    gender = st.selectbox(
+        "🧑 Gender",
+        ["Male", "Female"]
+    )
+
+    activity = st.selectbox(
+        "🏃 Activity Level",
+        ["Low", "Moderate", "High"]
+    )
+
+    allergies = st.text_input(
+        "🚫 Allergies / Foods to Avoid",
+        placeholder="e.g. nuts, milk, gluten"
+    )
+
+    meals = st.selectbox(
+        "🍽️ Meals per Day",
+        [3, 4, 5, 6]
+    )
+
+    if st.button("🚀 Generate Diet Plan"):
+
+        with st.spinner("Creating your custom plan..."):
+
+            prompt = f"""
+You are Diet Pal AI nutrition expert.
+
+Create a personalized one-day diet plan.
+
+User Goal: {goal}
+Age: {age}
+Weight: {weight} kg
+Height: {height} cm
+Gender: {gender}
+Activity Level: {activity}
+Allergies: {allergies if allergies else "None"}
+Meals Per Day: {meals}
+
+Provide:
+
+1. Recommended Daily Calories
+2. Breakfast
+3. Lunch
+4. Dinner
+5. Snacks
+6. Water Intake
+7. Helpful Tips
 """
-        chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="llama-3.3-70b-versatile"
-        )
-
-        recipes = chat_completion.choices[0].message.content
-        st.success("🍳 Recipes Generated!")
-        st.markdown("### 📋 Your Recipes")
-        st.markdown(recipes)
-else:
-    st.info("Please enter the ingredients to generate recipes.")
+
+            chat_completion = client.chat.completions.create(
+                messages=[
+                    {"role": "user", "content": prompt}
+                ],
+                model="llama-3.3-70b-versatile"
+            )
+
+            result = chat_completion.choices[0].message.content
+
+            st.success("🥗 Diet Plan Ready!")
+            st.markdown("### 📋 Your Plan")
+            st.markdown(result)
+
+# ----------------------------
+# FOOTER
+# ----------------------------
+st.markdown("---")
+st.caption("Powered by Groq + Streamlit 🚀")