code update

app.py

@@ -1,108 +1,108 @@
-import streamlit as st
-import pandas as pd
-import numpy as np
-import joblib
-import pickle
-import datetime as dt
-
-## loading in trained model
-model = joblib.load('GradientBoost_Flight_Fair_Model')
-# model = pickle.load(open("FlightPrice.pkl", "rb"))
-
-@st.cache()
-def make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline):
-    # preprocessing data before pre
-    # dictions
-    pred_input = []
-
-    stops = int(stops)
-    pred_input.append(stops)
-
-    # departure Date
-    journey_day = int(pd.to_datetime(journey_date, format="%Y-%m-%dT%H:%M").day)
-    pred_input.append(journey_day)
-
-    journey_month = int(pd.to_datetime(journey_date, format ="%Y-%m-%dT%H:%M").month)
-    pred_input.append(journey_month)
-
-    dep_min = int(journey_time.minute)
-    pred_input.append(dep_min)
-
-    dep_hour = int(journey_time.hour)
-    pred_input.append(dep_hour)
-
-    arrival_min = int(arrival_time.minute)
-    pred_input.append(arrival_min)
-
-    arrival_hour = int(arrival_time.hour)
-    pred_input.append(arrival_hour)
-
-    arrival_day = int(pd.to_datetime(arrival_date, format="%Y-%m-%dT%H:%M").day)
-    pred_input.append(arrival_day)
-
-    duration_min = abs(arrival_min - dep_min)
-    pred_input.append(duration_min)
-
-    duration_hour = abs(arrival_hour - dep_hour)
-    pred_input.append(duration_hour)
-
-    air_list = ['IndiGo', 'Air India', 'Jet Airways', 'SpiceJet', 'Multiple carriers', 'GoAir', 'Vistara', 'Air Asia', 'Vistara Premium economy', 'Jet Airways Business', 'Multiple carriers Premium economy', 'Trujet']
-    for a in air_list:
-        if a == airline:
-            pred_input.append(1)
-        else:
-            pred_input.append(0)
-
-    src_list = ['Banglore', 'Kolkata', 'Delhi', 'Chennai', 'Mumbai']
-    for i in src_list:
-        if i == source:
-            pred_input.append(1)
-        else:
-            pred_input.append(0)
-
-    dst_list = ['New Delhi', 'Banglore', 'Cochin', 'Kolkata', 'Delhi', 'Hyderabad']
-    for d in dst_list:
-        if d == destination:
-            pred_input.append(1)
-        else:
-            pred_input.append(0)
-
-    prediction = model.predict(np.array([pred_input]))
-
-    return int(prediction)
-
-
-def main():
-    
-    st.title('Flight Fair Proce Predictor')
-    st.subheader('Fill the following details to get the idea about flight fair price')
-
-    col1, col2 = st.columns([2, 1])
-    journey_date = col1.date_input('Journey Date')
-    journey_time = col2.time_input('Departure time')
-
-    col3, col4 = st.columns([2, 1])
-    arrival_date = col3.date_input('Arroval Date')
-    arrival_time = col4.time_input('Arrival time')
-
-    col5, col6 = st.columns(2)
-    source = col5.selectbox('Departure city',['Banglore', 'Kolkata', 'Delhi', 'Chennai', 'Mumbai'])
-    destination = col6.selectbox('Destination city', ['New Delhi', 'Banglore', 'Cochin', 'Kolkata', 'Delhi', 'Hyderabad'])
-
-    stops = st.selectbox('Total Stops', ['non-stop', 1, 2, 3, 4])
-
-    airline = st.selectbox('Choose Airline', ['IndiGo', 'Air India', 'Jet Airways', 'SpiceJet', 'Multiple carriers', 'GoAir', 'Vistara', 'Air Asia', 'Vistara Premium economy', 'Jet Airways Business', 'Multiple carriers Premium economy', 'Trujet'])
-
-    predict = st.button('Make Prediction',)
-
-    if stops == 'non-stop':
-        stops = 0
-    
-    # make prediction button logic
-    if predict:
-        with st.spinner('Wait for prediction....'):
-            t = make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline)
-        st.success(f'Fair Price will be around Rs.{t - 1000}')
-
-if __name__=='__main__': 
+import streamlit as st
+import pandas as pd
+import numpy as np
+import joblib
+import pickle
+import datetime as dt
+
+## loading in trained model
+model = joblib.load('GradientBoost_Flight_Fair_Model')
+# model = pickle.load(open("FlightPrice.pkl", "rb"))
+
+@st.cache()
+def make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline):
+    # preprocessing data before pre
+    # dictions
+    pred_input = []
+
+    stops = int(stops)
+    pred_input.append(stops)
+
+    # departure Date
+    journey_day = int(pd.to_datetime(journey_date, format="%Y-%m-%dT%H:%M").day)
+    pred_input.append(journey_day)
+
+    journey_month = int(pd.to_datetime(journey_date, format ="%Y-%m-%dT%H:%M").month)
+    pred_input.append(journey_month)
+
+    dep_min = int(journey_time.minute)
+    pred_input.append(dep_min)
+
+    dep_hour = int(journey_time.hour)
+    pred_input.append(dep_hour)
+
+    arrival_min = int(arrival_time.minute)
+    pred_input.append(arrival_min)
+
+    arrival_hour = int(arrival_time.hour)
+    pred_input.append(arrival_hour)
+
+    arrival_day = int(pd.to_datetime(arrival_date, format="%Y-%m-%dT%H:%M").day)
+    pred_input.append(arrival_day)
+
+    duration_min = abs(arrival_min - dep_min)
+    pred_input.append(duration_min)
+
+    duration_hour = abs(arrival_hour - dep_hour)
+    pred_input.append(duration_hour)
+
+    air_list = ['IndiGo', 'Air India', 'Jet Airways', 'SpiceJet', 'Multiple carriers', 'GoAir', 'Vistara', 'Air Asia', 'Vistara Premium economy', 'Jet Airways Business', 'Multiple carriers Premium economy', 'Trujet']
+    for a in air_list:
+        if a == airline:
+            pred_input.append(1)
+        else:
+            pred_input.append(0)
+
+    src_list = ['Banglore', 'Kolkata', 'Delhi', 'Chennai', 'Mumbai']
+    for i in src_list:
+        if i == source:
+            pred_input.append(1)
+        else:
+            pred_input.append(0)
+
+    dst_list = ['New Delhi', 'Banglore', 'Cochin', 'Kolkata', 'Delhi', 'Hyderabad']
+    for d in dst_list:
+        if d == destination:
+            pred_input.append(1)
+        else:
+            pred_input.append(0)
+
+    prediction = model.predict(np.array([pred_input]))
+
+    return int(prediction)
+
+
+def main():
+    
+    st.title('Flight Fair Proce Predictor')
+    st.subheader('Fill the following details to get the idea about flight fair price')
+
+    col1, col2 = st.columns([2, 1])
+    journey_date = col1.date_input('Journey Date')
+    journey_time = col2.time_input('Departure time')
+
+    col3, col4 = st.columns([2, 1])
+    arrival_date = col3.date_input('Arroval Date')
+    arrival_time = col4.time_input('Arrival time')
+
+    col5, col6 = st.columns(2)
+    source = col5.selectbox('Departure city',['Banglore', 'Kolkata', 'Delhi', 'Chennai', 'Mumbai'])
+    destination = col6.selectbox('Destination city', ['New Delhi', 'Banglore', 'Cochin', 'Kolkata', 'Delhi', 'Hyderabad'])
+
+    stops = st.selectbox('Total Stops', ['non-stop', 1, 2, 3, 4])
+
+    airline = st.selectbox('Choose Airline', ['IndiGo', 'Air India', 'Jet Airways', 'SpiceJet', 'Multiple carriers', 'GoAir', 'Vistara', 'Air Asia', 'Vistara Premium economy', 'Jet Airways Business', 'Multiple carriers Premium economy', 'Trujet'])
+
+    predict = st.button('Make Prediction',)
+
+    if stops == 'non-stop':
+        stops = 0
+    
+    # make prediction button logic
+    if predict:
+        with st.spinner('Wait for prediction....'):
+            t = make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline)
+        st.success(f'Fair Price will be around Rs.{t}')
+
+if __name__=='__main__': 
     main()