Update app.py

app.py

@@ -1,150 +1,58 @@
 import streamlit as st
+from model_api import query_model
+from prompt_builder import build_prompt
+
+st.set_page_config(page_title="FitPlan AI", layout="centered")
+
+st.title("🏋️ FitPlan AI – Personalized Workout Generator")
+
+# ---------------- INPUT ---------------- #
+
+name = st.text_input("Enter Your Name")
+
+gender = st.radio("Gender", ["Male", "Female"])
+
+height = st.number_input("Height (cm)", min_value=100.0, max_value=250.0)
+weight = st.number_input("Weight (kg)", min_value=30.0, max_value=200.0)
+
+goal = st.selectbox(
+    "Fitness Goal",
+    ["Build Muscle", "Lose Weight", "Improve Endurance", "General Fitness"]
+)
+
+fitness_level = st.radio(
+    "Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
+
+equipment = st.multiselect(
+    "Available Equipment",
+    [
+        "No Equipment", "Dumbbells", "Barbell",
+        "Pull-up Bar", "Resistance Bands",
+        "Treadmill", "Kettlebells", "Full Gym"
+    ]
+)
+
+# ---------------- GENERATE ---------------- #
+
+if st.button("Generate Workout Plan"):
+
+    if height > 0 and weight > 0:
+
+        prompt, bmi, bmi_status = build_prompt(
+            name, gender, height, weight,
+            goal, fitness_level, equipment
+        )
+
+        with st.spinner("Generating your personalized plan..."):
+            response = query_model(prompt)
+
+        st.subheader("📋 Your Personalized Workout Plan")
+        st.write(response)
+
+        st.info(f"Calculated BMI: {bmi:.2f} ({bmi_status})")
 
-# 1. Basic Configuration
-st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
-
-# 2. Enhanced CSS
-st.markdown("""
-<style>
-    /* Full Page Background */
-    [data-testid="stAppViewContainer"] {
-        background: linear-gradient(rgba(0, 0, 0, 0.75), rgba(0, 0, 0, 0.75)), 
-                    url("https://images.unsplash.com/photo-1517836357463-d25dfeac3438?q=80&w=2070&auto=format&fit=crop");
-        background-size: cover;
-        background-position: center;
-        background-attachment: fixed;
-    }
-
-    /* Hide the "Press Enter to submit form" hint globally */
-    [data-testid="InputInstructions"] {
-        display: none !important;
-    }
-
-    /* Form Container Polish */
-    div[data-testid="stForm"] {
-        background-color: rgba(255, 255, 255, 0.98) !important;
-        padding: 3rem !important;
-        border-radius: 20px !important;
-        border: none !important;
-        box-shadow: 0 15px 35px rgba(0,0,0,0.5);
-    }
-
-    /* --- FONT SIZE UPDATES --- */
-    /* Main Title (FitPlan AI) */
-    .main-title {
-        color: #FFFFFF !important;
-        font-size: 60px !important;
-        font-weight: 850 !important;
-        text-align: center;
-        margin-bottom: 0px;
-        text-shadow: 3px 3px 6px rgba(0,0,0,0.9);
-        font-family: 'Helvetica Neue', sans-serif;
-    }
-
-    /* Subtitle (Your personalized gym companion) */
-    .sub-title {
-        color: #E0E0E0 !important;
-        font-size: 24px !important;
-        text-align: center;
-        margin-top: -10px;
-        margin-bottom: 40px;
-        font-style: italic;
-        text-shadow: 1px 1px 3px rgba(0,0,0,0.8);
-    }
-
-    /* Form Section Headers (Orange) */
-    .form-header {
-        color: #f97316 !important;
-        font-size: 22px !important;
-        font-weight: bold !important;
-        text-transform: uppercase;
-        letter-spacing: 1px;
-    }
-
-    /* Labels for inputs */
-    label p {
-        font-size: 18px !important;
-        color: #1e293b !important;
-        font-weight: 600 !important;
-    }
-
-    /* Button Styling */
-    .stButton > button {
-        width: 100%;
-        background: linear-gradient(90deg, #f97316, #ea580c) !important;
-        color: white !important;
-        font-size: 20px !important;
-        font-weight: bold !important;
-        height: 3.5rem !important;
-        border-radius: 12px !important;
-        border: none !important;
-        margin-top: 20px;
-    }
-</style>
-""", unsafe_allow_html=True)
-
-# --------------------------------------------------
-# UI Header Section
-# --------------------------------------------------
-st.markdown('<p class="main-title">💪 FitPlan AI</p>', unsafe_allow_html=True)
-st.markdown('<p class="sub-title">Your personalized gym companion</p>', unsafe_allow_html=True)
-
-# --------------------------------------------------
-# Fitness Profile Form
-# --------------------------------------------------
-with st.form("fitness_form", clear_on_submit=False):
-    st.markdown('<p class="form-header">🧍‍♂️ 1. PERSONAL INFORMATION</p>', unsafe_allow_html=True)
-    
-    name = st.text_input("Full Name *", placeholder="Enter your full name")
-
-    col1, col2 = st.columns(2)
-    with col1:
-        height = st.number_input("Height (cm) *", min_value=0.0, step=1.0, value=0.0)
-    with col2:
-        weight = st.number_input("Weight (kg) *", min_value=0.0, step=0.1, value=0.0)
-
-    st.markdown("<br>", unsafe_allow_html=True)
-    st.markdown('<p class="form-header">🏋️ 2. FITNESS DETAILS</p>', unsafe_allow_html=True)
-    
-    goal = st.selectbox("Fitness Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
-    
-    equipment = st.multiselect("Available Equipment", 
-                               ["Dumbbells", "Resistance Band", "Yoga Mat", "Kettlebell", "Pull-up Bar", "No Equipment"],
-                               default=["No Equipment"])
-    
-    level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"], horizontal=True)
-
-    # Submission Button
-    submit = st.form_submit_button("GENERATE MY PLAN")
-
-# --------------------------------------------------
-# Logic & Calculation
-# --------------------------------------------------
-if submit:
-    if not name.strip() or height <= 50 or weight <= 10:
-        st.error("🚨 Please provide a valid name, height, and weight.")
     else:
-        # BMI Calculation
-        height_m = height / 100
-        bmi = round(weight / (height_m ** 2), 2)
-
-        if bmi < 18.5:
-            cat, color = "Underweight", "blue"
-        elif 18.5 <= bmi < 25:
-            cat, color = "Normal", "green"
-        elif 25 <= bmi < 30:
-            cat, color = "Overweight", "orange"
-        else:
-            cat, color = "Obese", "red"
-
-        # Results Display
-        st.balloons()
-        st.success(f"✅ Profile created for {name}")
-        
-        c1, c2 = st.columns(2)
-        with c1:
-            st.metric("Calculated BMI", bmi)
-        with c2:
-            st.markdown(f"### Status: :{color}[{cat}]")
-        
-        st.info(f"**Goal:** {goal} | **Experience:** {level}")
+        st.warning("Please enter valid height and weight.")
+