Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,14 +8,14 @@ st.set_page_config(page_title="Fitness Profile Generator", layout="wide")
 # 2. Load AI Model (Large Version)
 @st.cache_resource
 def load_model():
-    # Note: flan-t5-large is ~3GB.
+    # 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 (Preserved Gradient & Black Sidebar)
+# 3. Custom Styling (Gradient and Black Sidebar)
 st.markdown("""
     <style>
     .stApp {
@@ -53,21 +53,14 @@ st.markdown("""
         margin-top: 25px;
         border-left: 10px solid #004d57;
     }
-    .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)
 
 # --- SIDEBAR ---
 with st.sidebar:
-    st.markdown("# 👤 Personalised Fitness Plan")
+    st.markdown("# 👤 Personalised Fitness Plan Generator")
     st.write("---")
-    st.write("Your digital fitness profile manager.")
+    st.write("Your digital coach for a healthier lifestyle.")
 
 # --- MAIN CONTENT ---
 st.title("🏋️ Create Your Fitness Profile")
@@ -108,10 +101,13 @@ st.write("---")
 
 # --- SUBMIT BUTTON & LOGIC ---
 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 (*) correctly.")
+    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.")
     else:
-        # BMI Logic
         height_m = height_cm / 100
         bmi = weight_kg / (height_m ** 2)
         bmi_rounded = round(bmi, 2)
@@ -135,63 +131,40 @@ if st.button("Submit Profile"):
             </div>
         """, unsafe_allow_html=True)
 
-        # 5. AI Prompt (Optimized to Stop at Day 5)
+        # 5. AI Prompt Preparation (Slightly adjusted for better logic)
         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.
+        prompt = f"""Task: Create a detailed 5-day workout plan for a {gender} with a BMI of {bmi_rounded} ({bmi_status}).
+Goal: {goal}
+Level: {fitness_level}
+Available Equipment: {equipment_list}
 
-Structure:
-Day 1: [Exercises]
-Day 2: [Exercises]
-Day 3: [Rest/Recovery]
-Day 4: [Exercises]
-Day 5: [Exercises]
+Instructions: Provide a structured routine for Day 1, Day 2, Day 3, Day 4, and Day 5. Include exercises and rest days.
 
 Workout Plan:
 Day 1:"""
 
-        # 6. AI Generation
-        with st.spinner("Generating your aligned workout plan..."):
+        # 6. AI Generation (Using Optimized Parameters)
+        with st.spinner("Generating your AI workout plan..."):
             try:
                 inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+                
                 outputs = model.generate(
                     **inputs,
-                    max_new_tokens=450, 
-                    min_new_tokens=150,
+                    max_new_tokens=512,      # 512 is usually enough for T5-Large
+                    min_new_tokens=150,      # Forces the model to provide a longer response
                     do_sample=True,
-                    temperature=0.7,
-                    repetition_penalty=2.5
+                    temperature=0.8,         # Slight increase for variety
+                    top_p=0.9,
+                    repetition_penalty=2.0   # Prevents it from getting stuck on one phrase
                 )
                 
-                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]
+                result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
                 
-                # Alignment Logic: Splitting the text by "Day" keywords to create visual blocks
-                st.markdown("### 📋 Your Personalized Routine")
+                # Prepending "Day 1:" because the prompt cuts off there
+                final_plan = "Day 1: " + result
                 
-                # 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)
-                        day_content = full_text[start:end].strip()
-                        
-                        # 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)
-
+                st.subheader(" Your Personalized Workout Plan")
+                st.write(final_plan)
             except Exception as e:
                 st.error(f"AI Generation failed: {e}")