Update app.py

app.py

@@ -1,113 +1,81 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torch
-# IMPORT your new prompt builder functions
 from prompt_builder import build_prompt 
+from model_api import query_model # Import your API function
 
 # 1. Page Configuration
-st.set_page_config(page_title="Fitness Profile Generator", layout="wide")
+st.set_page_config(page_title="FitPlan AI", layout="wide")
 
-# 2. Load AI Model
-@st.cache_resource
-def load_model():
-    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 (Same as your original)
+# 2. Custom Styling (Forced Black Text)
 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; }
     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;
+        background-color: white !important; padding: 25px; border-radius: 15px; 
+        border-left: 10px solid #004d57; color: #000000 !important; margin-bottom: 20px;
     }
     .day-card {
         background-color: #ffffff !important; padding: 20px; border-radius: 12px;
-        margin-bottom: 15px; border-left: 5px solid #004d57; color: #000000 !important;
+        margin-bottom: 15px; border-left: 8px solid #004d57; color: #000000 !important;
         white-space: pre-wrap !important;
     }
     </style>
     """, unsafe_allow_html=True)
 
-# --- SIDEBAR ---
-with st.sidebar:
-    st.markdown("# 👤 Personalised Fitness Plan")
-    st.write("---")
-    st.write("Your digital coach.")
+st.title("🏋️ Personalised Fitness Plan")
 
-# --- MAIN CONTENT ---
-st.title("🏋️ Create Your Fitness Profile")
-
-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 *", 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, max_value=120, value=25)
+# --- INPUT SECTION ---
+st.markdown("### 1️⃣ User Information")
+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)
 
 col_h, col_w = st.columns(2)
-with col_h:
-    height_cm = st.number_input("Height in centimeters *", min_value=0.0, step=0.1)
-with col_w:
-    weight_kg = st.number_input("Weight in kilograms *", min_value=0.0, step=0.1)
+with col_h: height = st.number_input("Height (cm)", min_value=10.0)
+with col_w: weight = st.number_input("Weight (kg)", min_value=1.0)
 
-st.write("---")
-st.markdown("### 2️⃣ Fitness Details")
-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.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"])
 
-# --- SUBMIT BUTTON ---
-if st.button("Submit Profile"):
-    if not name or height_cm <= 0 or weight_kg <= 0 or not equipment:
-        st.error("Please fill in all required fields.")
+# --- GENERATION ---
+if st.button("Generate My Professional Plan"):
+    if not name or height <= 0 or weight <= 0:
+        st.error("Please fill in all details.")
     else:
-        # 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!")
+        # Build the prompt using your external file
+        prompt, bmi, bmi_status, status_color = build_prompt(name, gender, height, weight, goal, level, equip)
 
-        # Display Assessment Card
+        # Show Assessment
         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:.2f}</strong> | Category: <span style="color: {status_color}; font-weight: bold;">{bmi_status}</span></p>
+                <h2>Assessment for {name}</h2>
+                <p>BMI: <strong>{bmi:.2f}</strong> | Category: <span style="color:{status_color}">{bmi_status}</span></p>
             </div>
         """, 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
+        with st.spinner("AI Trainer is calculating your routine..."):
+            # Call your model_api.py function
+            raw_workout = query_model(prompt)
+
+            if "Error" in raw_workout:
+                st.error(raw_workout)
+            else:
+                st.markdown("### 📋 Your 5-Day Routine")
                 
-                st.markdown("### 📋 Your Personalized Routine")
+                # Visual Parsing logic
                 days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
+                
                 for i in range(len(days)):
-                    start = full_text.find(days[i])
+                    start = raw_workout.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()
+                        # 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)
-            except Exception as e:
-                st.error(f"Generation failed: {e}")
+
+st.info("Note: Consult a doctor before starting any new exercise program.")