from flask_wtf import FlaskForm
from wtforms import IntegerField, SelectField, StringField
from wtforms.validators import DataRequired, NumberRange

class UserDataForm(FlaskForm):
    age = IntegerField('Age', validators=[
        DataRequired(),
        NumberRange(min=18, max=100)
    ])
    
    activity_level = SelectField('Activity Level',
        choices=[
            ('sedentary', 'Sedentary'),
            ('light', 'Light Activity'),
            ('moderate', 'Moderately Active'),
            ('very', 'Very Active')
        ]
    )
    
    pregnancy_status = SelectField('Pregnancy Status',
        choices=[
            ('not_pregnant', 'Not Pregnant'),
            ('first_trimester', 'First Trimester'),
            ('second_trimester', 'Second Trimester'),
            ('third_trimester', 'Third Trimester'),
            ('postpartum', 'Postpartum')
        ]
    )
    
    health_conditions = SelectField('Health Conditions',
        choices=[
            ('none', 'None'),
            ('pelvic_pain', 'Pelvic Pain'),
            ('incontinence', 'Incontinence'),
            ('diastasis', 'Diastasis Recti'),
            ('other', 'Other')
        ]
    )
    
    fitness_goal = SelectField('Fitness Goals',
        choices=[
            ('strength', 'Build Strength'),
            ('recovery', 'Postpartum Recovery'),
            ('maintenance', 'General Maintenance'),
            ('prevention', 'Preventive Care')
        ]
    )