from flask_wtf import FlaskForm
from wtforms.fields.choices import RadioField
from wtforms import StringField, PasswordField, SubmitField, SelectField,RadioField
from wtforms.validators import DataRequired, Email, EqualTo, Length

class RegisterForm(FlaskForm):
    user_type = RadioField('Login As', choices=[('user', 'User'), ('seller', 'Seller')], default='user')
    first_name = StringField('First Name', validators=[DataRequired()])
    last_name = StringField('Last Name', validators=[DataRequired()])
    email = StringField('Email or Username', validators=[DataRequired()])
    username = StringField('Username', validators=[DataRequired()])

    password = PasswordField('Password', validators=[DataRequired(), Length(min=6)])
    confirm_password = PasswordField('Confirm Password', validators=[EqualTo('password')])
    submit = SubmitField('Register')

class LoginForm(FlaskForm):
    user_type = SelectField('Login As', choices=[('user', 'User'), ('seller', 'Seller')])
    email = StringField('Email or Username', validators=[DataRequired()])
    password = PasswordField('Password', validators=[DataRequired()])
    submit = SubmitField('Login')