Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,16 +5,17 @@ import torch
 # 1. Page Configuration
 st.set_page_config(page_title="Fitness Profile Generator", layout="wide")
 
-# 2. Load AI Model
+# 2. Load AI Model (Large Version)
 @st.cache_resource
 def load_model():
+    # Note: flan-t5-large is ~3GB.
     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 (Forced Black Text & Alignment)
+# 3. Custom Styling (Preserved Gradient & Black Sidebar)
 st.markdown("""
     <style>
     .stApp {
@@ -31,32 +32,33 @@ st.markdown("""
         color: #2D3436 !important;
         font-weight: 600;
     }
-    /* Fixed Day Card Visibility and Text Color */
-    .day-card {
-        background-color: #ffffff !important;
-        padding: 20px;
-        border-radius: 12px;
-        margin-bottom: 20px;
-        border-left: 8px solid #004d57;
-        color: #000000 !important; /* Force Pure Black */
-        white-space: pre-wrap !important;
-        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+    div[data-baseweb="input"], div[data-baseweb="select"] {
+        background-color: rgba(255, 255, 255, 0.9) !important;
+        border-radius: 8px !important;
     }
-    .day-header {
-        color: #004d57 !important;
-        font-size: 20px;
-        font-weight: bold;
-        margin-bottom: 10px;
-        display: block;
+    .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: #ffffff !important;
+        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;
+    }
+    .day-card {
+        background-color: rgba(255, 255, 255, 0.8);
+        padding: 15px;
+        border-radius: 10px;
+        margin-bottom: 10px;
+        border-left: 5px solid #fda085;
     }
     </style>
     """, unsafe_allow_html=True)
@@ -65,85 +67,131 @@ st.markdown("""
 with st.sidebar:
     st.markdown("# 👤 Personalised Fitness Plan")
     st.write("---")
+    st.write("Your digital fitness profile manager.")
 
 # --- MAIN CONTENT ---
 st.title("🏋️ Create Your Fitness Profile")
 
-# INPUTS
+# SECTION 1: PERSONAL INFORMATION
+st.markdown("### 1️⃣ Personal Information")
 col_name, col_gender, col_age = st.columns([2, 1, 1])
 with col_name:
-    name = st.text_input("Full Name *")
+    name = st.text_input("Full Name *", placeholder="e.g. John Doe")
 with col_gender:
     gender = st.selectbox("Gender *", ["Male", "Female", "Other"])
 with col_age:
-    age = st.number_input("Age *", min_value=1, value=25)
+    age = st.number_input("Age *", min_value=1, max_value=120, value=25)
 
 col_h, col_w = st.columns(2)
 with col_h:
-    height_cm = st.number_input("Height (cm) *", min_value=0.0)
+    height_cm = st.number_input("Height in centimeters *", min_value=0.0, step=0.1)
 with col_w:
-    weight_kg = st.number_input("Weight (kg) *", min_value=0.0)
+    weight_kg = st.number_input("Weight in kilograms *", min_value=0.0, step=0.1)
 
-goal = st.selectbox("Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
-fitness_level = st.selectbox("Level", ["Beginner", "Intermediate", "Advanced"])
-equipment = st.multiselect("Equipment", ["Dumbbells", "Resistance Band", "Yoga Mat", "Kettlebells", "Pull-up Bar", "No Equipment"], default=["No Equipment"])
+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"]
+)
+
+st.write("---")
+
+# --- SUBMIT BUTTON & LOGIC ---
 if st.button("Submit Profile"):
-    if not name or height_cm <= 0 or weight_kg <= 0:
-        st.error("Please fill all fields.")
+    if not name or height_cm <= 0 or weight_kg <= 0 or not equipment:
+        st.error("Please fill in all required fields (*) correctly.")
     else:
-        bmi = weight_kg / ((height_cm/100)**2)
-        bmi_status = "Normal" if 18.5 <= bmi < 25 else "Underweight" if bmi < 18.5 else "Overweight"
-        
-        st.markdown(f'<div class="result-card"><h2>Assessment for {name}</h2><p>BMI: {bmi:.2f} ({bmi_status})</p></div>', unsafe_allow_html=True)
-
-        # 5. REFINED PROMPT
-        eq_list = ", ".join(equipment)
-        prompt = f"""Task: Create a unique 5-day workout routine for a {gender} ({fitness_level}) aiming for {goal}.
-Available Equipment: {eq_list}.
-BMI: {bmi:.2f}.
-
-Provide specific exercise names, sets, and reps for each day. 
-Day 3 must be a recovery/rest day.
-Finish completely at Day 5.
-
-Plan:
+        # BMI Logic
+        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"
+
+        st.success("Profile Submitted Successfully!")
+
+        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>
+            </div>
+        """, unsafe_allow_html=True)
+
+        # 5. AI Prompt (Optimized to Stop at Day 5)
+        equipment_list = ", ".join(equipment)
+        prompt = f"""Task: Write a 5-day workout plan for {name}.
+Details: {gender}, Goal: {goal}, Level: {fitness_level}, Equipment: {equipment_list}.
+Requirement: Stop exactly after Day 5. Do not include Day 6.
+
+Structure:
+Day 1: [Exercises]
+Day 2: [Exercises]
+Day 3: [Rest/Recovery]
+Day 4: [Exercises]
+Day 5: [Exercises]
+
+Workout Plan:
 Day 1:"""
 
-        with st.spinner("Generating exercises..."):
+        # 6. AI Generation
+        with st.spinner("Generating your aligned workout plan..."):
             try:
                 inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
-                # ADJUSTED PARAMETERS FOR BETTER DIVERSITY
                 outputs = model.generate(
                     **inputs,
-                    max_new_tokens=700,
-                    min_new_tokens=350,
+                    max_new_tokens=450, 
+                    min_new_tokens=150,
                     do_sample=True,
-                    temperature=0.85, # Higher temperature for more variety
-                    top_k=50,         # Limits weird word choices
-                    repetition_penalty=3.0 # Forces the model to move to next days
+                    temperature=0.7,
+                    repetition_penalty=2.5
                 )
                 
-                result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
-                full_text = "Day 1: " + result.split("Day 6")[0] # Safety cut
-
+                raw_result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+                
+                # Cut off anything from "Day 6" onwards
+                clean_result = raw_result.split("Day 6")[0]
+                
+                # Alignment Logic: Splitting the text by "Day" keywords to create visual blocks
                 st.markdown("### 📋 Your Personalized Routine")
                 
+                # Re-assembling the full text to ensure we don't miss Day 1
+                full_text = "Day 1: " + clean_result
+                
                 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:
+                        # Find where the next day starts to cut the text
                         end = full_text.find(days[i+1]) if i+1 < len(days) else len(full_text)
-                        content = full_text[start:end].replace(days[i], "").strip()
+                        day_content = full_text[start:end].strip()
                         
-                        # Only display if there's actual content
-                        if len(content) > 5:
-                            st.markdown(f"""
-                                <div class="day-card">
-                                    <span class="day-header">{days[i]}</span>
-                                    {content}
-                                </div>
-                            """, unsafe_allow_html=True)
+                        # Display in a styled card
+                        st.markdown(f"""
+                            <div class="day-card">
+                                <strong>{day_content[:6]}</strong><br>
+                                {day_content[6:].strip()}
+                            </div>
+                        """, unsafe_allow_html=True)
+
             except Exception as e:
-                st.error(f"Error: {e}")
+                st.error(f"AI Generation failed: {e}")