Delete src

src/streamlit_app.py

@@ -1,170 +0,0 @@
-import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torch
-
-# 1. Page Configuration
-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. 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 (Gradient and Black Sidebar)
-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;
-    }
-    div[data-baseweb="input"], div[data-baseweb="select"] {
-        background-color: rgba(255, 255, 255, 0.9) !important;
-        border-radius: 8px !important;
-    }
-    .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: 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;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-# --- SIDEBAR ---
-with st.sidebar:
-    st.markdown("# 👤 Personalised Fitness Plan Generator")
-    st.write("---")
-    st.write("Your digital coach for a healthier lifestyle.")
-
-# --- MAIN CONTENT ---
-st.title("🏋️ Create Your Fitness Profile")
-
-# 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 *", 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)
-
-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)
-
-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:
-        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:
-        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 Preparation (Slightly adjusted for better logic)
-        equipment_list = ", ".join(equipment)
-        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}
-
-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 (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=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.8,         # Slight increase for variety
-                    top_p=0.9,
-                    repetition_penalty=2.0   # Prevents it from getting stuck on one phrase
-                )
-                
-                result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
-                
-                # Prepending "Day 1:" because the prompt cuts off there
-                final_plan = "Day 1: " + result
-                
-                st.subheader(" Your Personalized Workout Plan")
-                st.write(final_plan)
-            except Exception as e:
-                st.error(f"AI Generation failed: {e}")
-