Update app.py

app.py

@@ -1,71 +1,50 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
+# IMPORT your new prompt builder functions
+from prompt_builder import build_prompt 
 
 # 1. Page Configuration
 st.set_page_config(page_title="Fitness Profile Generator", layout="wide")
 
-# 2. Load AI Model (Large Version)
+# 2. Load AI Model
 @st.cache_resource
 def load_model():
-    # Note: flan-t5-large is ~3GB. Ensure your Hugging Face Space has enough RAM.
     tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
     model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
     return tokenizer, model
 
 tokenizer, model = load_model()
 
-# 3. Custom Styling (Gradient and Black Sidebar)
+# 3. Custom Styling (Same as your original)
 st.markdown("""
     <style>
-    .stApp {
-        background: linear-gradient(135deg, #f6d365 0%, #fda085 100%);
-        background-attachment: fixed;
+    .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; }
+    h1, h2, h3, p, label { color: #2D3436 !important; font-weight: 600; }
+    .result-card { 
+        background-color: white; padding: 25px; border-radius: 15px; 
+        box-shadow: 0 10px 25px rgba(0,0,0,0.1); margin-top: 25px; border-left: 10px solid #004d57; 
+        color: #000000 !important;
     }
-    section[data-testid="stSidebar"] {
-        background-color: #000000 !important;
-    }
-    section[data-testid="stSidebar"] h1, section[data-testid="stSidebar"] p {
-        color: #FFFFFF !important;
-    }
-    h1, h2, h3, p, label {
-        color: #2D3436 !important;
-        font-weight: 600;
-    }
-    div[data-baseweb="input"], div[data-baseweb="select"] {
-        background-color: rgba(255, 255, 255, 0.9) !important;
-        border-radius: 8px !important;
-    }
-    .stButton > button {
-        background-color: #004d57 !important;
-        color: white !important;
-        border-radius: 8px !important;
-        padding: 10px 24px !important;
-        font-weight: bold !important;
-        width: 100% !important;
-        border: none !important;
-    }
-    .result-card {
-        background-color: white;
-        padding: 25px;
-        border-radius: 15px;
-        box-shadow: 0 10px 25px rgba(0,0,0,0.1);
-        margin-top: 25px;
-        border-left: 10px solid #004d57;
+    .day-card {
+        background-color: #ffffff !important; padding: 20px; border-radius: 12px;
+        margin-bottom: 15px; border-left: 5px solid #004d57; color: #000000 !important;
+        white-space: pre-wrap !important;
     }
     </style>
     """, unsafe_allow_html=True)
 
 # --- SIDEBAR ---
 with st.sidebar:
-    st.markdown("# 👤 Personalised Fitness Plan Generator")
+    st.markdown("# 👤 Personalised Fitness Plan")
     st.write("---")
-    st.write("Your digital coach for a healthier lifestyle.")
+    st.write("Your digital coach.")
 
 # --- MAIN CONTENT ---
 st.title("🏋️ Create Your Fitness Profile")
 
-# SECTION 1: PERSONAL INFORMATION
 st.markdown("### 1️⃣ Personal Information")
 col_name, col_gender, col_age = st.columns([2, 1, 1])
 with col_name:
@@ -82,51 +61,53 @@ with col_w:
     weight_kg = st.number_input("Weight in kilograms *", min_value=0.0, step=0.1)
 
 st.write("---")
-
-# SECTION 2: FITNESS DETAILS
 st.markdown("### 2️⃣ Fitness Details")
-col_goal, col_level = st.columns(2)
-with col_goal:
-    goal = st.selectbox("Fitness Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
-with col_level:
-    fitness_level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
-
-equipment = st.multiselect(
-    "Available Equipment *",
-    ["Dumbbells", "Resistance Band", "Yoga Mat", "Kettlebells", "Pull-up Bar", "No Equipment"],
-    default=["No Equipment"]
-)
+goal = st.selectbox("Fitness Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
+fitness_level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
+equipment = st.multiselect("Available Equipment *", ["Dumbbells", "Resistance Band", "Yoga Mat", "Kettlebells", "Pull-up Bar", "No Equipment"], default=["No Equipment"])
 
-st.write("---")
-
-# --- SUBMIT BUTTON & LOGIC ---
+# --- SUBMIT BUTTON ---
 if st.button("Submit Profile"):
-    if not name:
-        st.error("Please enter your name.")
-    elif height_cm <= 0 or weight_kg <= 0:
-        st.error("Please enter valid height and weight.")
-    elif not equipment:
-        st.error("Please select at least one equipment option.")
+    if not name or height_cm <= 0 or weight_kg <= 0 or not equipment:
+        st.error("Please fill in all required fields.")
     else:
-        height_m = height_cm / 100
-        bmi = weight_kg / (height_m ** 2)
-        bmi_rounded = round(bmi, 2)
-
-        if bmi_rounded < 18.5:
-            bmi_status, status_color = "Underweight", "#3498db"
-        elif 18.5 <= bmi_rounded < 25:
-            bmi_status, status_color = "Normal", "#2ecc71"
-        elif 25 <= bmi_rounded < 30:
-            bmi_status, status_color = "Overweight", "#f1c40f"
-        else:
-            bmi_status, status_color = "Obese", "#e74c3c"
+        # CALL the function from prompt_builder.py
+        prompt, bmi, bmi_status, status_color = build_prompt(
+            name, gender, height_cm, weight_kg, goal, fitness_level, equipment
+        )
 
         st.success("Profile Submitted Successfully!")
 
+        # Display Assessment Card
         st.markdown(f"""
             <div class="result-card">
                 <h2 style="color: #004d57; margin-top:0;">Assessment for {name}</h2>
                 <p>{gender} | {age} Years Old</p>
-                <p>Calculated BMI: <strong>{bmi_rounded}</strong> | Category: <span style="color: {status_color}; font-weight: bold;">{bmi_status}</span></p>
+                <p>Calculated BMI: <strong>{bmi:.2f}</strong> | Category: <span style="color: {status_color}; font-weight: bold;">{bmi_status}</span></p>
             </div>
-        """, unsafe_allow_html=True)
+        """, unsafe_allow_html=True)
+
+        # AI Generation
+        with st.spinner("Generating your aligned workout plan..."):
+            try:
+                inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+                outputs = model.generate(
+                    **inputs, max_new_tokens=600, min_new_tokens=250,
+                    do_sample=True, temperature=0.7, repetition_penalty=2.5
+                )
+                result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+                
+                # Cleanup and Display
+                clean_result = result.split("Day 6")[0]
+                full_text = "Day 1: " + clean_result
+                
+                st.markdown("### 📋 Your Personalized Routine")
+                days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
+                for i in range(len(days)):
+                    start = full_text.find(days[i])
+                    if start != -1:
+                        end = full_text.find(days[i+1]) if i+1 < len(days) else len(full_text)
+                        day_content = full_text[start:end].strip()
+                        st.markdown(f'<div class="day-card">{day_content}</div>', unsafe_allow_html=True)
+            except Exception as e:
+                st.error(f"Generation failed: {e}")