Update app.py

app.py

@@ -1,11 +1,10 @@
 import streamlit as st
 from prompt_builder import build_prompt 
-from model_api import query_model # Import your API function
+from model_api import query_model 
 
-# 1. Page Configuration
 st.set_page_config(page_title="FitPlan AI", layout="wide")
 
-# 2. Custom Styling (Forced Black Text)
+# Styling for visibility
 st.markdown("""
     <style>
     .stApp { background: linear-gradient(135deg, #f6d365 0%, #fda085 100%); background-attachment: fixed; }
@@ -24,31 +23,27 @@ st.markdown("""
 
 st.title("🏋️ Personalised Fitness Plan")
 
-# --- INPUT SECTION ---
-st.markdown("### 1️⃣ User Information")
+# Input Section
 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=25)
+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=10.0)
+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)
 
-st.markdown("### 2️⃣ Training Goals")
 goal = st.selectbox("Goal", ["Build Muscle", "Weight Loss", "Strength", "Flexibility"])
 level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
-equip = st.multiselect("Equipment", ["Dumbbells", "Barbell", "Resistance Bands", "Yoga Mat", "No Equipment"])
+equip = st.multiselect("Equipment", ["Dumbbells", "Kettlebells", "Pull-up Bar", "Resistance Bands", "Yoga Mat", "No Equipment"])
 
-# --- GENERATION ---
-if st.button("Generate My Professional Plan"):
+if st.button("Submit Profile"):
     if not name or height <= 0 or weight <= 0:
         st.error("Please fill in all details.")
     else:
-        # Build the prompt using your external file
+        # Line 75 Fix: Matches the 4 values returned by prompt_builder
         prompt, bmi, bmi_status, status_color = build_prompt(name, gender, height, weight, goal, level, equip)
 
-        # Show Assessment
         st.markdown(f"""
             <div class="result-card">
                 <h2>Assessment for {name}</h2>
@@ -56,26 +51,18 @@ if st.button("Generate My Professional Plan"):
             </div>
         """, unsafe_allow_html=True)
 
-        with st.spinner("AI Trainer is calculating your routine..."):
-            # Call your model_api.py function
+        with st.spinner("AI Trainer is thinking..."):
             raw_workout = query_model(prompt)
 
             if "Error" in raw_workout:
                 st.error(raw_workout)
             else:
                 st.markdown("### 📋 Your 5-Day Routine")
-                
-                # Visual Parsing logic
                 days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
                 
                 for i in range(len(days)):
                     start = raw_workout.find(days[i])
                     if start != -1:
-                        # Logic to find the end of the day's text
                         end = raw_workout.find(days[i+1]) if i+1 < len(days) else len(raw_workout)
                         day_content = raw_workout[start:end].strip()
-                        
-                        # Render in Black Text Card
-                        st.markdown(f'<div class="day-card">{day_content}</div>', unsafe_allow_html=True)
-
-st.info("Note: Consult a doctor before starting any new exercise program.")
+                        st.markdown(f'<div class="day-card">{day_content}</div>', unsafe_allow_html=True)