Basic-Flask-Project/forms.py at master · Samriddhi2711/Basic-Flask-Project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import pandas as pd
from flask_wtf import FlaskForm
from wtforms import (
    SelectField,
    DateField,
    TimeField,
    IntegerField,
    SubmitField

)
from wtforms.validators import DataRequired

train = pd.read_csv("data/train.csv")
val = pd.read_csv("data/val.csv")
X_data= pd.concat([train,val],axis=0).drop(columns="price")

class InputForm(FlaskForm):
    airline = SelectField(
        label="AirLine",
        choices=X_data.airline.unique().tolist(),
        validators=[DataRequired()]
    )

    date_of_journey= DateField(
        label="Date of Journey",
        validators=[DataRequired()]
    )

    source = SelectField(
        label="Sources",
        choices=X_data.source.unique().tolist(),
        validators=[DataRequired()]
    )

    destination = SelectField(
        label="Destination",
        choices=X_data.destination.unique().tolist(),
        validators=[DataRequired()])

    dep_time= TimeField(
        label="Departure Time",
        validators=[DataRequired()]
    )

    arrival_time= TimeField(
        label="Arrival Time",
        validators=[DataRequired()]
    )

    duration = IntegerField(
        label="Duration",
        validators=[DataRequired()]
    )

    total_stops = IntegerField(
        label="Total Stops",
        validators=[DataRequired()]
    )

    additional_info = SelectField(
        label="Additional Info",
        choices=X_data.additional_info.unique().tolist(),
        validators=[DataRequired()])

    submit = SubmitField("Predict")