Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,19 +2,6 @@ import streamlit as st
 
 # 1. Basic Configuration
 st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
-frpm transformers import pipeline
-def load_model():
-
-    return pipeline(
-
-        "text-generation",
-
-        model="google/flan-t5-base"
-
-    )
- 
-generator = load_model()
- 
 
 # 2. Enhanced CSS
 st.markdown("""
@@ -120,48 +107,37 @@ with st.form("fitness_form", clear_on_submit=False):
     
     level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"], horizontal=True)
 
-   if st.button("Submit Profile"):
- 
-    if not name:
-        st.error("Please enter your name.")
- 
-    elif not equipment:
-        st.error("Please select at least one equipment option.")
- 
-    elif bmi is None:
-        st.error("Please enter valid height and weight.")
- 
+    # 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"""
-        Generate a 5-day structured workout plan.
- 
-        User Details:
-        Name: {name}
-        Gender: {gender}
-        BMI: {bmi:.2f} ({bmi_status})
-        Goal: {goal}
-        Fitness Level: {fitness_level}
-        Available Equipment: {equipment_list}
- 
-        Requirements:
-        - Include warmup
-        - Include exercises with sets and reps
-        - Include rest time
-        - Adjust intensity based on BMI and fitness level
-        - Keep it structured day-wise
-        """
- 
-        with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
- 
-        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}")