Update app.py

app.py

@@ -1,18 +1,8 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torch
 
 # 1. Basic Configuration
 st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
 
-def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
-    return tokenizer, model
- 
-tokenizer, model = load_model()
-
-generator = load_model()
 # 2. Enhanced CSS
 st.markdown("""
 <style>
@@ -24,10 +14,12 @@ st.markdown("""
         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;
@@ -36,6 +28,7 @@ st.markdown("""
         border: none !important;
         box-shadow: 0 15px 35px rgba(0,0,0,0.5);
     }
+
     /* --- FONT SIZE UPDATES --- */
     /* Main Title (FitPlan AI) */
     .main-title {
@@ -47,6 +40,7 @@ st.markdown("""
         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;
@@ -57,6 +51,7 @@ st.markdown("""
         font-style: italic;
         text-shadow: 1px 1px 3px rgba(0,0,0,0.8);
     }
+
     /* Form Section Headers (Orange) */
     .form-header {
         color: #f97316 !important;
@@ -65,12 +60,14 @@ st.markdown("""
         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%;
@@ -117,53 +114,37 @@ with st.form("fitness_form", clear_on_submit=False):
     
     level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"], horizontal=True)
 
-    # SUBMIT BUTTON
-if st.button(" Submit Profile"):
- 
-    if not name:
-        st.error("Please enter your name.")
- 
-    elif height <= 0 or weight <= 0:
-        st.error("Please enter valid height and weight.")
- 
-    elif not equipment:
-        st.error("Please select at least one equipment option.")
- 
+    # 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:
-        st.success(" Profile Submitted Successfully!")
- 
-        bmi_status = bmi_category(bmi)
-        equipment_list = ", ".join(equipment)
- 
-        prompt = f"""
-You are a certified professional fitness trainer.
- 
-Create a detailed 5-day workout plan.
- 
-User Information:
-- Gender: {gender}
-- BMI: {bmi:.2f} ({bmi_status})
-- Goal: {goal}
-- Fitness Level: {fitness_level}
-- Equipment Available: {equipment_list}
- 
-Start directly with:
- 
-Day 1:
-"""
- 
-        with st.spinner("Generating your AI workout plan..."):
- 
-            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
- 
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=900,
-                temperature=0.7,
-                do_sample=True
-            )
- 
-            result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
- 
-        st.subheader(" Your Personalized Workout Plan")
-        st.write(result)
+        # 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}")