Update app.py

app.py

@@ -4,83 +4,116 @@ from model_api import query_model
 
 st.set_page_config(page_title="FitPlan AI", layout="wide")
 
-# CSS to Hide Header and Center Login
+# Updated CSS for Single Box Per Day
 st.markdown("""
     <style>
-    /* Hide the top Streamlit header and padding */
-    header {visibility: hidden;}
-    .main .block-container { padding-top: 0rem; }
+    .stApp { background: linear-gradient(135deg, #f6d365 0%, #fda085 100%); background-attachment: fixed; }
+    section[data-testid="stSidebar"] { background-color: #000000 !important; }
+    section[data-testid="stSidebar"] h1, section[data-testid="stSidebar"] p { color: #FFFFFF !important; }
     
-    .stApp { 
-        background: linear-gradient(135deg, #f6d365 0%, #fda085 100%); 
-        background-attachment: fixed;
-    }
-    
-    /* Perfect Centering for Login */
-    .login-wrapper {
-        display: flex;
-        flex-direction: column;
-        justify-content: center;
-        align-items: center;
-        height: 80vh; /* Centers vertically within view */
-    }
-    
-    .login-box { 
-        background: white; 
-        padding: 50px; 
-        border-radius: 25px; 
-        box-shadow: 0 15px 35px rgba(0,0,0,0.15); 
-        width: 100%;
-        max-width: 500px;
-        text-align: center;
-    }
-
-    /* Unified Day Card Style */
-    .day-container { 
+    /* Main Card for the Day */
+    .day-card { 
         background-color: white !important; 
         padding: 30px; 
-        border-radius: 25px; 
-        margin-bottom: 35px; 
-        box-shadow: 0 10px 30px rgba(0,0,0,0.08);
+        border-radius: 20px; 
+        margin-bottom: 40px; 
+        box-shadow: 0 8px 24px rgba(0,0,0,0.1);
         border-left: 12px solid #004d57;
     }
-    .day-header { color: #004d57; font-size: 26px; font-weight: bold; border-bottom: 2px solid #f0f2f6; padding-bottom: 15px; margin-bottom: 20px; }
-    .exercise-row { display: flex; justify-content: space-between; align-items: center; padding: 15px 0; border-bottom: 1px solid #f1f1f1; }
-    .stat-pill { background: #f8f9fa; padding: 6px 12px; border-radius: 12px; text-align: center; min-width: 80px; }
+    
+    .day-title { color: #004d57; font-size: 28px; font-weight: bold; margin-bottom: 25px; border-bottom: 2px solid #f0f0f0; padding-bottom: 10px; }
+
+    /* Exercise Row - No border/shadow to keep it inside the single box */
+    .exercise-row { 
+        display: flex; 
+        align-items: center; 
+        justify-content: space-between; 
+        padding: 15px 0;
+        border-bottom: 1px solid #eee;
+    }
+    .exercise-row:last-child { border-bottom: none; }
+
+    .ex-name { flex: 2; font-weight: 600; color: #2d3436; font-size: 18px; }
+    .stat-container { display: flex; gap: 20px; flex: 1.5; justify-content: flex-end; }
+    .stat-box { text-align: center; min-width: 70px; }
+    .stat-val { background: #e0f2f1; color: #004d57; padding: 4px 12px; border-radius: 12px; font-weight: bold; display: block; }
+    .stat-label { font-size: 10px; color: #636e72; text-transform: uppercase; margin-top: 4px; display: block; }
     </style>
     """, unsafe_allow_html=True)
 
-if 'logged_in' not in st.session_state: st.session_state.logged_in = False
 if 'page' not in st.session_state: st.session_state.page = 'input'
+if 'workout_plan' not in st.session_state: st.session_state.workout_plan = None
 
-# --- LOGIN PAGE ---
-if not st.session_state.logged_in:
-    # Use a div wrapper to apply the flexbox centering
-    st.markdown('<div class="login-wrapper">', unsafe_allow_html=True)
-    st.markdown('<div class="login-box">', unsafe_allow_html=True)
-    st.title("🏋️ FitPlan AI Login")
-    email_input = st.text_input("Email Address", placeholder="example@email.com")
-    
-    if st.button("Login", use_container_width=True):
-        if "@" in email_input and "." in email_input:
-            st.session_state.logged_in = True
+# --- SIDEBAR ---
+with st.sidebar:
+    st.markdown("# 👤 Personalised Fitness Plan")
+    if st.session_state.page == 'result':
+        if st.button("⬅️ New Profile"):
+            st.session_state.page = 'input'
             st.rerun()
-        else:
-            st.error("Please enter a valid email.")
-    st.markdown('</div></div>', unsafe_allow_html=True)
 
-# --- PROTECTED GENERATOR PAGE ---
-else:
-    if st.session_state.page == 'input':
-        st.title("🚀 Fitness Plan Generator")
-        # Your Profile Input Form here...
-        # [Profile inputs like Name, Age, Height, etc.]
-        if st.button("Generate Plan"):
-            # build_prompt and query_model logic here
+# --- PAGE 1: INPUT ---
+if st.session_state.page == 'input':
+    st.title("🏋️ Create Your Fitness Profile")
+    col1, col2, col3 = st.columns([2, 1, 1])
+    with col1: name = st.text_input("Name")
+    with col2: gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+    with col3: age = st.number_input("Age", min_value=1, value=19)
+    
+    h, w = st.columns(2)
+    with h: height = st.number_input("Height (cm)", min_value=1.0)
+    with w: weight = st.number_input("Weight (kg)", min_value=1.0)
+    
+    goal = st.selectbox("Goal", ["Build Muscle", "Weight Loss", "Strength", "Flexibility"])
+    level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
+    equip = st.multiselect("Equipment", ["Dumbbells", "Kettlebells", "Pull-up Bar", "Resistance Bands", "Yoga Mat", "No Equipment"])
+
+    if st.button("Submit Profile"):
+        prompt, bmi, bmi_status, status_color = build_prompt(name, gender, age, height, weight, goal, level, equip)
+        with st.spinner("Building your single-box workout plan..."):
+            raw_workout = query_model(prompt)
+            st.session_state.workout_plan = raw_workout
+            st.session_state.assessment_data = {"name": name, "bmi": bmi, "status": bmi_status, "color": status_color, "age": age, "gender": gender}
             st.session_state.page = 'result'
             st.rerun()
 
-    elif st.session_state.page == 'result':
-        # Result Page with Unified Day Cards
-        st.title("📋 Your Personalized Routine")
-        # Loop through workout_plan and display using .day-container
+# --- PAGE 2: RESULTS ---
+elif st.session_state.page == 'result':
+    d = st.session_state.assessment_data
+    st.markdown(f'<div class="result-card"><h2>Assessment for {d["name"]}</h2><p>{d["gender"]} | {d["age"]} Years Old</p><p>BMI: <strong>{d["bmi"]:.2f}</strong> | <span style="color:{d["color"]}">{d["status"]}</span></p></div>', unsafe_allow_html=True)
+
+    raw_content = st.session_state.workout_plan
+    days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
+    
+    for i in range(len(days)):
+        start = raw_content.find(days[i])
+        if start != -1:
+            end = raw_content.find(days[i+1]) if i+1 < len(days) else len(raw_content)
+            day_block = raw_content[start:end].strip()
+            
+            # Start of Single White Box for the Day
+            st.markdown(f'<div class="day-card"><div class="day-title">💠 {days[i][:-1]}</div>', unsafe_allow_html=True)
+            
+            lines = day_block.split("\n")
+            for line in lines:
+                if "|" in line:
+                    parts = [p.strip() for p in line.replace("- ", "").split("|")]
+                    if len(parts) == 4:
+                        ex_name, sets, reps, rest = parts
+                        st.markdown(f"""
+                            <div class="exercise-row">
+                                <div class="ex-name">🔥 {ex_name}</div>
+                                <div class="stat-container">
+                                    <div class="stat-box"><span class="stat-val">{sets}</span><span class="stat-label">Sets</span></div>
+                                    <div class="stat-box"><span class="stat-val">{reps}</span><span class="stat-label">Reps</span></div>
+                                    <div class="stat-box"><span class="stat-val" style="background:#fff3e0;">{rest}</span><span class="stat-label">Rest</span></div>
+                                </div>
+                            </div>
+                        """, unsafe_allow_html=True)
+            
+            if "Rest Day" in day_block or "Recovery" in day_block:
+                st.info("🧘 Day 3: Active recovery and stretching.")
+            
+            st.markdown('</div>', unsafe_allow_html=True) # End of Day Card
+
+    st.download_button("📥 Download Full Plan", data=raw_content, file_name=f"{d['name']}_Plan.txt")