Update app.py

app.py

@@ -4,23 +4,40 @@ from model_api import query_model
 
 st.set_page_config(page_title="FitPlan AI", layout="wide")
 
-# Enhanced CSS for the Icon-Table Grid UI
+# Updated CSS for Single Box Per Day
 st.markdown("""
     <style>
     .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; }
     
-    .result-card { background-color: white !important; padding: 25px; border-radius: 15px; border-left: 10px solid #004d57; color: #000000 !important; margin-bottom: 20px; }
+    /* Main Card for the Day */
+    .day-card { 
+        background-color: white !important; 
+        padding: 30px; 
+        border-radius: 20px; 
+        margin-bottom: 40px; 
+        box-shadow: 0 8px 24px rgba(0,0,0,0.1);
+        border-left: 12px solid #004d57;
+    }
     
-    .day-container { background-color: #f8f9fa; border-radius: 20px; padding: 25px; margin-bottom: 30px; border: 1px solid #e0e0e0; }
-    .day-header { color: #004d57; font-size: 24px; font-weight: bold; margin-bottom: 20px; display: flex; align-items: center; }
-    
-    .exercise-row { background: white; border-radius: 12px; padding: 15px; margin-bottom: 10px; display: flex; align-items: center; justify-content: space-between; box-shadow: 0 2px 4px rgba(0,0,0,0.05); }
-    .ex-name { flex: 2; font-weight: 600; color: #2d3436; font-size: 16px; }
-    .stat-box { flex: 1; text-align: center; display: flex; flex-direction: column; align-items: center; }
-    .stat-val { background: #e0f2f1; color: #004d57; padding: 5px 15px; border-radius: 20px; font-weight: bold; font-size: 14px; min-width: 60px; }
-    .stat-label { font-size: 11px; color: #636e72; text-transform: uppercase; margin-top: 4px; }
+    .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)
 
@@ -35,7 +52,7 @@ with st.sidebar:
             st.session_state.page = 'input'
             st.rerun()
 
-# --- INPUT PAGE ---
+# --- PAGE 1: INPUT ---
 if st.session_state.page == 'input':
     st.title("🏋️ Create Your Fitness Profile")
     col1, col2, col3 = st.columns([2, 1, 1])
@@ -43,9 +60,9 @@ if st.session_state.page == 'input':
     with col2: gender = st.selectbox("Gender", ["Male", "Female", "Other"])
     with col3: age = st.number_input("Age", min_value=1, value=19)
     
-    col_h, col_w = st.columns(2)
-    with col_h: height = st.number_input("Height (cm)", min_value=1.0)
-    with col_w: weight = st.number_input("Weight (kg)", min_value=1.0)
+    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"])
@@ -53,17 +70,17 @@ if st.session_state.page == 'input':
 
     if st.button("Submit Profile"):
         prompt, bmi, bmi_status, status_color = build_prompt(name, gender, age, height, weight, goal, level, equip)
-        with st.spinner("Generating Professional UI..."):
+        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()
 
-# --- RESULT PAGE ---
+# --- PAGE 2: RESULTS ---
 elif st.session_state.page == 'result':
     d = st.session_state.assessment_data
-    st.markdown(f'<div class="result-card"><h2>Hello {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)
+    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:"]
@@ -74,25 +91,30 @@ elif st.session_state.page == 'result':
             end = raw_content.find(days[i+1]) if i+1 < len(days) else len(raw_content)
             day_block = raw_content[start:end].strip()
             
-            st.markdown(f'<div class="day-container"><div class="day-header">🔹 {days[i][:-1]}</div>', unsafe_allow_html=True)
+            # 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)
             
-            # Parse individual exercises
             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:
-                        name, sets, reps, rest = parts
+                        ex_name, sets, reps, rest = parts
                         st.markdown(f"""
                             <div class="exercise-row">
-                                <div class="ex-name">🏋️ {name}</div>
-                                <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; color:#e65100;">{rest}</span><span class="stat-label">Rest</span></div>
+                                <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)
-                elif "Rest" in line:
-                    st.info("🧘 Enjoy your Rest Day! Recovery is part of progress.")
-            st.markdown('</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 Plan", data=raw_content, file_name="Workout.txt")
+    st.download_button("📥 Download Full Plan", data=raw_content, file_name=f"{d['name']}_Plan.txt")
+