Upload 2 files

app.py

@@ -0,0 +1,543 @@
+import streamlit as st
+from nutrition_agent import nutrition_agent
+from PIL import Image
+import requests
+from io import BytesIO
+import google.generativeai as genai
+import os
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+genai.configure(api_key=GOOGLE_API_KEY)
+
+# Enhanced page configuration
+st.set_page_config(
+    page_title="🥗 Smartest AI Nutrition Assistant", 
+    layout="wide",
+    initial_sidebar_state="collapsed",
+    page_icon="🥗"
+)
+
+# Custom CSS for enhanced UI
+st.markdown("""
+<style>
+    /* Main background and theme */
+    .main {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        padding: 2rem;
+    }
+    
+    /* Hide Streamlit branding */
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
+    header {visibility: hidden;}
+    
+    /* Custom title styling */
+    .custom-title {
+        text-align: center;
+        color: white;
+        font-size: 3.5rem;
+        font-weight: 800;
+        margin-bottom: 1rem;
+        text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
+        animation: fadeInDown 1s ease-out;
+    }
+    
+    .custom-subtitle {
+        text-align: center;
+        color: rgba(255,255,255,0.9);
+        font-size: 1.2rem;
+        margin-bottom: 2rem;
+        animation: fadeInUp 1s ease-out 0.3s both;
+    }
+    
+    /* Enhanced tabs */
+    .stTabs [data-baseweb="tab-list"] {
+        gap: 20px;
+        background: rgba(255,255,255,0.1);
+        border-radius: 25px;
+        padding: 10px;
+        backdrop-filter: blur(10px);
+    }
+    
+    .stTabs [data-baseweb="tab"] {
+        height: 60px;
+        padding: 0 30px;
+        background: rgba(255,255,255,0.2);
+        border-radius: 20px;
+        color: white;
+        font-weight: 600;
+        border: 1px solid rgba(255,255,255,0.3);
+        transition: all 0.3s ease;
+    }
+    
+    .stTabs [aria-selected="true"] {
+        background: white;
+        color: #667eea;
+        box-shadow: 0 5px 20px rgba(0,0,0,0.2);
+    }
+    
+    /* Form styling */
+    .stForm {
+        background: white;
+        padding: 2rem;
+        border-radius: 20px;
+        box-shadow: 0 20px 40px rgba(0,0,0,0.1);
+        margin: 20px 0;
+    }
+    
+    /* Fix label visibility */
+    .stForm label,
+    .stTextInput label,
+    .stSelectbox label,
+    .stTextArea label,
+    .stRadio label,
+    .stCheckbox label {
+        color: #333 !important;
+        font-weight: 600 !important;
+        font-size: 1rem !important;
+        margin-bottom: 8px !important;
+    }
+    
+    /* Fix section headings */
+    .stForm h4 {
+        color: #667eea !important;
+        font-weight: 700 !important;
+        margin-top: 1.5rem !important;
+        margin-bottom: 1rem !important;
+        border-bottom: 2px solid #e0e0e0;
+        padding-bottom: 0.5rem;
+    }
+    
+    /* Fix markdown text in forms */
+    .stForm .stMarkdown {
+        color: #333 !important;
+    }
+    
+    /* Fix radio button labels */
+    .stRadio > div > div > div > label {
+        color: #555 !important;
+        font-weight: 500 !important;
+    }
+    
+    /* Input field enhancements */
+    .stTextInput > div > div > input,
+    .stSelectbox > div > div > select,
+    .stTextArea > div > div > textarea {
+        border: 2px solid #e0e0e0;
+        border-radius: 10px;
+        padding: 12px;
+        font-size: 1rem;
+        transition: all 0.3s ease;
+        background: #f8f9fa;
+    }
+    
+    .stTextInput > div > div > input:focus,
+    .stSelectbox > div > div > select:focus,
+    .stTextArea > div > div > textarea:focus {
+        border-color: #667eea;
+        background: white;
+        box-shadow: 0 0 0 3px rgba(102, 126, 234, 0.1);
+    }
+    
+    /* Button enhancements */
+    .stButton > button {
+        width: 100%;
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        color: white;
+        border: none;
+        border-radius: 25px;
+        padding: 15px 30px;
+        font-size: 1.1rem;
+        font-weight: 600;
+        transition: all 0.3s ease;
+        box-shadow: 0 5px 15px rgba(0,0,0,0.2);
+    }
+    
+    .stButton > button:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 10px 25px rgba(0,0,0,0.3);
+    }
+    
+    /* Card styling for content */
+    .content-card {
+        background: white;
+        padding: 2rem;
+        border-radius: 20px;
+        box-shadow: 0 10px 30px rgba(0,0,0,0.1);
+        margin: 20px 0;
+        border-left: 5px solid #667eea;
+    }
+    
+    /* Fix text color in content cards */
+    .content-card h3,
+    .content-card h4,
+    .content-card p,
+    .content-card div {
+        color: #333 !important;
+    }
+    
+    /* Fix markdown headings */
+    .stMarkdown h3,
+    .stMarkdown h4,
+    .stMarkdown h5 {
+        color: #667eea !important;
+        font-weight: 700 !important;
+    }
+    
+    /* Fix all text content */
+    .stMarkdown p,
+    .stMarkdown div {
+        color: #555 !important;
+    }
+    
+    /* Image upload area styling */
+    .stFileUploader > div {
+        border: 3px dashed #667eea;
+        border-radius: 15px;
+        padding: 2rem;
+        text-align: center;
+        background: rgba(102, 126, 234, 0.05);
+        transition: all 0.3s ease;
+    }
+    
+    .stFileUploader > div:hover {
+        border-color: #764ba2;
+        background: rgba(102, 126, 234, 0.1);
+    }
+    
+    /* Progress bar styling */
+    .stProgress > div > div > div > div {
+        background: linear-gradient(90deg, #667eea, #764ba2);
+    }
+    
+    /* Alert styling */
+    .stAlert {
+        border-radius: 15px;
+        border: none;
+        box-shadow: 0 5px 15px rgba(0,0,0,0.1);
+    }
+    
+    /* Animation keyframes */
+    @keyframes fadeInDown {
+        from { opacity: 0; transform: translateY(-30px); }
+        to { opacity: 1; transform: translateY(0); }
+    }
+    
+    @keyframes fadeInUp {
+        from { opacity: 0; transform: translateY(30px); }
+        to { opacity: 1; transform: translateY(0); }
+    }
+    
+    /* Feature icons */
+    .feature-icon {
+        font-size: 2rem;
+        margin-bottom: 0.5rem;
+        display: block;
+    }
+    
+    /* Grid layout for forms */
+    .form-grid {
+        display: grid;
+        grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
+        gap: 20px;
+    }
+    
+    /* Additional text visibility fixes */
+    .stApp > div > div > div > div {
+        color: inherit;
+    }
+    
+    /* Force white background for form containers */
+    .element-container:has(.stForm) {
+        background: white;
+        border-radius: 20px;
+        padding: 1rem;
+        margin: 1rem 0;
+    }
+    
+    /* Ensure all text in white containers is dark */
+    div[data-testid="stForm"] {
+        background: white !important;
+        color: #333 !important;
+    }
+    
+    div[data-testid="stForm"] * {
+        color: #333 !important;
+    }
+    
+    /* Override any inherited white text */
+    .stTextInput > label,
+    .stSelectbox > label,
+    .stTextArea > label {
+        color: #333 !important;
+        font-weight: 600 !important;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# Enhanced Header
+st.markdown('<h1 class="custom-title">🥗 Smartest AI Nutrition Assistant</h1>', unsafe_allow_html=True)
+st.markdown('<p class="custom-subtitle">Your personal AI-powered nutritionist for healthier living</p>', unsafe_allow_html=True)
+
+# Feature highlights
+col1, col2, col3 = st.columns(3)
+with col1:
+    st.markdown("""
+    <div style="text-align: center; color: white; padding: 20px;">
+        <span class="feature-icon">🎯</span>
+        <h3>Personalized Plans</h3>
+        <p>Customized diet plans based on your unique profile</p>
+    </div>
+    """, unsafe_allow_html=True)
+
+with col2:
+    st.markdown("""
+    <div style="text-align: center; color: white; padding: 20px;">
+        <span class="feature-icon">📱</span>
+        <h3>Smart Analysis</h3>
+        <p>AI-powered food image recognition and analysis</p>
+    </div>
+    """, unsafe_allow_html=True)
+
+with col3:
+    st.markdown("""
+    <div style="text-align: center; color: white; padding: 20px;">
+        <span class="feature-icon">⚡</span>
+        <h3>Instant Results</h3>
+        <p>Get nutrition insights in seconds</p>
+    </div>
+    """, unsafe_allow_html=True)
+
+st.markdown("<br>", unsafe_allow_html=True)
+
+# Enhanced Tabs
+tab1, tab2 = st.tabs(["📋 Get Personalized Diet Plan", "📷 Image-Based Nutrition Analysis"])
+
+# ---------------- Tab 1: Enhanced Personalized Diet Plan ----------------
+with tab1:
+    st.markdown('<div class="content-card">', unsafe_allow_html=True)
+    st.markdown("### 📋 Create Your Personal Nutrition Profile")
+    st.markdown("Fill out the form below to receive a customized diet plan tailored to your needs and goals.")
+    
+    with st.form("diet_form", clear_on_submit=False):
+        # Personal Information Section
+        st.markdown("#### 👤 Personal Information")
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            name = st.text_input("Full Name", placeholder="Enter your full name")
+            age = st.text_input("Age", placeholder="e.g., 25")
+            gender = st.selectbox("Gender", ["Select...", "Male", "Female", "Other"])
+        
+        with col2:
+            location = st.text_input("Location", placeholder="e.g., Mumbai, India")
+            diet_type = st.selectbox("Diet Preference", ["Select...", "Vegetarian", "Non-Vegetarian", "Vegan"])
+            calories = st.text_input("Daily Calorie Target", placeholder="e.g., 2000 (optional)")
+        
+        st.markdown("#### 🎯 Health & Goals")
+        col3, col4 = st.columns(2)
+        
+        with col3:
+            health_issues = st.text_area("Health Conditions", 
+                                       placeholder="Any existing health issues, allergies, or medical conditions",
+                                       height=100)
+            health_goals = st.text_area("Fitness Goals", 
+                                      placeholder="e.g., Weight loss, muscle gain, better energy",
+                                      height=100)
+        
+        with col4:
+            likes = st.text_area("Foods You Love", 
+                                placeholder="List your favorite foods and cuisines",
+                                height=100)
+            dislikes = st.text_area("Foods to Avoid", 
+                                  placeholder="Foods you dislike or want to avoid",
+                                  height=100)
+        
+        # Enhanced submit button
+        submitted = st.form_submit_button("🚀 Generate My Personal Diet Plan", use_container_width=True)
+    
+    st.markdown('</div>', unsafe_allow_html=True)
+
+    if submitted:
+        # Validation
+        if not name or not age or gender == "Select..." or diet_type == "Select...":
+            st.error("⚠️ Please fill in all required fields (Name, Age, Gender, Diet Preference)")
+        else:
+            user_profile = (
+                f"My name is {name}. I am {age} years old, {gender}, {diet_type}, living in {location}. "
+                f"I have the following health issues: {health_issues}. "
+                f"My goal is to {health_goals}. I like eating {likes} but dislike {dislikes}. "
+                f"My daily calorie need is {calories}. "
+                "Please generate a personalized meal plan broken into breakfast, lunch, snacks, dinner. "
+                "Also include a clear explanation of nutritional content and a conclusion for health advice."
+            )
+
+            with st.spinner("🔄 Creating your personalized nutrition plan..."):
+                progress_bar = st.progress(0)
+                for i in range(100):
+                    progress_bar.progress(i + 1)
+                
+                result = nutrition_agent.run(user_profile)
+                progress_bar.empty()
+
+            # Enhanced result display
+            if result:
+                st.markdown('<div class="content-card">', unsafe_allow_html=True)
+                st.markdown("### 🎉 Your Personalized Diet Plan")
+                st.success("✅ Your custom nutrition plan has been generated!")
+                
+                if isinstance(result, str):
+                    st.markdown(result)
+                elif hasattr(result, 'content'):
+                    st.markdown(result.content)
+                else:
+                    st.write(result)
+                
+                # Add download button
+                st.download_button(
+                    label="📥 Download Diet Plan",
+                    data=str(result),
+                    file_name=f"{name}_diet_plan.txt",
+                    mime="text/plain"
+                )
+                st.markdown('</div>', unsafe_allow_html=True)
+
+# ---------------- Tab 2: Enhanced Image-Based Nutrition Analysis ----------------
+with tab2:
+    st.markdown('<div class="content-card">', unsafe_allow_html=True)
+    st.markdown("### 📷 Smart Food Analysis")
+    st.markdown("Upload a food image or paste an image URL to get instant nutritional analysis powered by AI.")
+    
+    image = None
+    image_input_type = st.radio(
+        "Choose your preferred input method:",
+        ["📁 Upload Image", "🔗 Paste Image URL"],
+        horizontal=True
+    )
+
+    if image_input_type == "📁 Upload Image":
+        uploaded_file = st.file_uploader(
+            "Drop your food image here or click to browse",
+            type=["png", "jpg", "jpeg"],
+            help="Supported formats: PNG, JPG, JPEG"
+        )
+        if uploaded_file:
+            try:
+                image = Image.open(uploaded_file)
+                st.success("✅ Image uploaded successfully!")
+            except Exception as e:
+                st.error(f"❌ Error opening image: {e}")
+    else:
+        image_url = st.text_input(
+            "Enter image URL",
+            placeholder="https://example.com/food-image.jpg",
+            help="Paste a direct link to your food image"
+        )
+        if image_url:
+            try:
+                with st.spinner("🔄 Loading image from URL..."):
+                    response = requests.get(image_url)
+                    response.raise_for_status()
+                    image = Image.open(BytesIO(response.content))
+                st.success("✅ Image loaded successfully!")
+            except Exception as e:
+                st.error(f"❌ Error loading image: {e}")
+
+    if image is not None:
+        # Enhanced image display
+        col1, col2 = st.columns([2, 1])
+        
+        with col1:
+            st.image(image, caption="🍽️ Selected Food Image", use_column_width=True)
+        
+        with col2:
+            st.markdown("#### 🔍 Analysis Options")
+            analysis_depth = st.selectbox(
+                "Analysis Detail Level",
+                ["Basic Analysis", "Detailed Analysis", "Health Recommendations"]
+            )
+            
+            include_calories = st.checkbox("Include Calorie Estimation", value=True)
+            include_health = st.checkbox("Include Health Rating", value=True)
+            include_ingredients = st.checkbox("Identify Ingredients", value=False)
+
+        if st.button("🔬 Analyze Food Nutrition", use_container_width=True):
+            # Enhanced prompt based on user selections
+            prompt_parts = ["Analyze the nutritional content of the food shown in this image."]
+            
+            if analysis_depth == "Detailed Analysis":
+                prompt_parts.append("Provide detailed nutritional breakdown including macronutrients and micronutrients.")
+            elif analysis_depth == "Health Recommendations":
+                prompt_parts.append("Include health recommendations and dietary suggestions.")
+            
+            if include_calories:
+                prompt_parts.append("Estimate the caloric content.")
+            
+            if include_health:
+                prompt_parts.append("Rate the healthiness of this food on a scale of 1-10.")
+            
+            if include_ingredients:
+                prompt_parts.append("Identify and list the main ingredients visible.")
+            
+            prompt = " ".join(prompt_parts)
+            
+            try:
+                model = genai.GenerativeModel('gemini-2.5-flash-preview-04-17')
+                with st.spinner("🤖 AI is analyzing your food image..."):
+                    progress_bar = st.progress(0)
+                    for i in range(100):
+                        progress_bar.progress(i + 1)
+                    
+                    result = model.generate_content([prompt, image])
+                    progress_bar.empty()
+                
+                # Enhanced result display
+                st.markdown("---")
+                st.markdown("### 🧠 AI Nutrition Analysis Results")
+                st.success("✅ Analysis complete!")
+                
+                # Create expandable sections for better organization
+                with st.expander("📊 Detailed Analysis", expanded=True):
+                    st.markdown(result.text)
+                
+                # Add action buttons
+                col1, col2, col3 = st.columns(3)
+                with col1:
+                    st.download_button(
+                        label="📥 Save Analysis",
+                        data=result.text,
+                        file_name="nutrition_analysis.txt",
+                        mime="text/plain"
+                    )
+                with col2:
+                    if st.button("🔄 Analyze Again"):
+                        st.experimental_rerun()
+                with col3:
+                    if st.button("📤 Share Results"):
+                        st.info("📋 Copy the analysis text to share with others!")
+                
+            except Exception as e:
+                st.error(f"❌ Analysis failed: {e}")
+                st.info("💡 Try uploading a clearer image or check your internet connection.")
+    else:
+        # Enhanced placeholder
+        st.markdown("""
+        <div style="text-align: center; padding: 3rem; background: rgba(102, 126, 234, 0.05); border-radius: 15px; border: 2px dashed #667eea;">
+            <h3>🍽️ Ready to Analyze Your Food?</h3>
+            <p>Upload an image or paste a URL to get started with AI-powered nutrition analysis!</p>
+        </div>
+        """, unsafe_allow_html=True)
+    
+    st.markdown('</div>', unsafe_allow_html=True)
+
+# Enhanced Footer
+st.markdown("---")
+st.markdown("""
+<div style="text-align: center; color: white; padding: 20px;">
+    <p>💡 <strong>Pro Tip:</strong> For best results, use clear, well-lit images of your food!</p>
+    <p>🤖 Powered by AI • 🥗 Built for Health • ❤️ Made with Streamlit</p>
+</div>
+""", unsafe_allow_html=True)

nutrition_agent.py

@@ -0,0 +1,34 @@
+from phi.agent import Agent
+from phi.model.google import Gemini
+import os
+from phi.tools.duckduckgo import DuckDuckGo
+from dotenv import load_dotenv
+import google.generativeai as genai
+
+# Load  API key
+load_dotenv()
+
+
+
+
+# Define the nutrition assistant agent
+nutrition_agent = Agent(
+    name="nutrition_agent",
+    model=Gemini(id="gemini-2.5-flash-preview-04-17"),
+    #gemini-2.5-pro-exp-03-25
+    #gemini-2.5-pro-preview-05-06
+    tools=[DuckDuckGo()],
+
+    instructions=[
+        "You are the smartest AI Nutrition Assistant. ",
+        "Ask the user for their name, age, gender, dietary preference (veg/non-veg), location, "
+        "health complaints, fitness goals, food likes/dislikes, and calorie needs. "
+        "Then provide a detailed personalized diet plan (breakfast/lunch/dinner) "
+        "along with clear nutritional explanations tailored to their inputs.",
+    "Always respond in well-structured markdown format using clear headings like ### Breakfast, ### Lunch, etc.",
+        "Do not generate one big paragraph. Instead, use bullet points, headings, and whitespace for clarity."
+
+    ],
+    show_tool_calls=True,
+    markdown=True
+)