Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+ import streamlit as st
 from prompt_builder import build_prompt 
 from model_api import query_model 
 from diet_builder import build_diet_prompt
@@ -39,9 +39,28 @@ if 'diet_plan' not in st.session_state: st.session_state.diet_plan = None
 # --- SIDEBAR ---
 with st.sidebar:
     st.markdown("# 👤 FitPlan AI")
-    if st.button("📊 Dashboard", use_container_width=True): st.session_state.page = 'dashboard'; st.rerun()
-    if st.button("🏋️ Workout Plan", use_container_width=True): st.session_state.page = 'input'; st.rerun()
-    if st.button("🥗 Dietary Plan", use_container_width=True): st.session_state.page = 'diet'; st.rerun()
+    if st.button("📊 Dashboard", use_container_width=True): 
+        st.session_state.page = 'dashboard'
+        st.rerun()
+    if st.button("🏋️ Workout Plan", use_container_width=True): 
+        # If a plan exists, show the result, otherwise show input
+        if st.session_state.workout_plan:
+            st.session_state.page = 'result'
+        else:
+            st.session_state.page = 'input'
+        st.rerun()
+    if st.button("🥗 Dietary Plan", use_container_width=True): 
+        st.session_state.page = 'diet'
+        st.rerun()
+    
+    if st.session_state.workout_plan:
+        st.write("---")
+        if st.button("🔄 Create New Profile"):
+            st.session_state.workout_plan = None
+            st.session_state.diet_plan = None
+            st.session_state.user_data = None
+            st.session_state.page = 'input'
+            st.rerun()
 
 # --- 1. DASHBOARD PAGE ---
 if st.session_state.page == 'dashboard':
@@ -52,14 +71,12 @@ if st.session_state.page == 'dashboard':
         with col1: st.markdown(f'<div class="stat-card"><div style="font-size:24px; font-weight:bold;">{d["name"]}</div><div style="color:#636e72;">USER</div></div>', unsafe_allow_html=True)
         with col2: st.markdown(f'<div class="stat-card"><div style="font-size:24px; font-weight:bold;">{d["bmi"]:.1f}</div><div style="color:#636e72;">BMI</div></div>', unsafe_allow_html=True)
         with col3: st.markdown(f'<div class="stat-card"><div style="font-size:24px; font-weight:bold; color:{d["color"]}">{d["status"]}</div><div style="color:#636e72;">HEALTH STATUS</div></div>', unsafe_allow_html=True)
-        
         st.write("---")
-        st.subheader("Current Focus")
-        st.info(f"Targeting: **{d['goal']}** at **{d['level']}** intensity.")
+        st.success(f"Current Goal: **{d['goal']}**")
     else:
-        st.warning("No data found. Please go to 'Workout Plan' and generate your profile.")
+        st.warning("Go to 'Workout Plan' to generate your profile.")
 
-# --- 2. INPUT PAGE (Your Original Logic) ---
+# --- 2. INPUT PAGE ---
 elif st.session_state.page == 'input':
     st.title("🏋️ Fitness Profile")
     c1, c2, c3 = st.columns([2, 1, 1])
@@ -78,11 +95,11 @@ elif st.session_state.page == 'input':
         with st.spinner("AI is crafting your routine..."):
             st.session_state.workout_plan = query_model(prompt)
             st.session_state.user_data = {"name": name, "bmi": bmi, "status": status, "color": color, "age": age, "gender": gender, "goal": goal, "height": height, "weight": weight, "level": level}
-            st.session_state.diet_plan = None # Reset diet so it generates fresh for this user
+            st.session_state.diet_plan = None 
             st.session_state.page = 'result'
             st.rerun()
 
-# --- 3. WORKOUT RESULT (Preserving your exact Row/Pill layout) ---
+# --- 3. WORKOUT RESULT (Stored Memory) ---
 elif st.session_state.page == 'result':
     d = st.session_state.user_data
     st.markdown(f'<div class="assessment-card"><h2>Hello {d["name"]}</h2><p>BMI: <strong>{d["bmi"]:.2f}</strong> | <span style="color:{d["color"]}">{d["status"]}</span></p></div>', unsafe_allow_html=True)
@@ -104,11 +121,12 @@ elif st.session_state.page == 'result':
             st.markdown('</div>', unsafe_allow_html=True)
     st.download_button("📥 Download Plan", data=raw, file_name=f"{d['name']}_Plan.txt")
 
-# --- 4. DIETARY PAGE (Now Functional) ---
+# --- 4. DIETARY PAGE (Stored Memory) ---
 elif st.session_state.page == 'diet':
     st.title("🥗 Dietary Plan")
     if st.session_state.user_data:
         d = st.session_state.user_data
+        # Only query the model if the diet doesn't exist yet
         if not st.session_state.diet_plan:
             d_prompt = build_diet_prompt(d['name'], d['gender'], d['age'], d['height'], d['weight'], d['goal'])
             with st.spinner("Generating unique meal plan..."):
@@ -123,6 +141,5 @@ elif st.session_state.page == 'diet':
                     m_n, m_t, m_i, m_c = parts
                     st.markdown(f'<div class="item-row"><div class="ex-name">🍴 {m_n}<br><small>{m_i}</small></div><div class="stat-group"><div class="stat-pill"><span class="stat-val">{m_t}</span><span class="stat-label">Time</span></div><div class="stat-pill" style="background:#e8f5e9;"><span class="stat-val" style="color:#2e7d32;">{m_c}</span><span class="stat-label">Energy</span></div></div></div>', unsafe_allow_html=True)
         st.markdown('</div>', unsafe_allow_html=True)
-        if st.button("Refresh Meals"): st.session_state.diet_plan = None; st.rerun()
     else:
         st.warning("Generate a profile in 'Workout Plan' first.")