Update app.py

app.py

@@ -1,290 +1,115 @@
-from flask import Flask,request,render_template
-from flask_cors import cross_origin
+import streamlit as st
 import pickle
 import pandas as pd
 
-model = pickle.load(open('flight_rf.pkl','rb'))
+model = pickle.load(open('flight_rf.pkl', 'rb'))
 
-app = Flask(__name__)
+def main():
+    st.title("Flight Price Prediction App")
 
-@app.route('/')
-@cross_origin()
-def home():
-	return render_template('home.html')
+    dep_time = st.text_input("Departure Time", "2023-01-01T12:00")
+    arrival_time = st.text_input("Arrival Time", "2023-01-01T14:00")
+    
+    Total_stops = st.selectbox("Number of Stops", [0, 1, 2, 3, 4])
 
-@app.route('/predict',methods=['GET','POST'])
-@cross_origin()
-def predict():
-    if request.method=='POST':
-        dep_time = request.form['Dep_Time']
+    airline = st.selectbox("Select Airline", ["Jet Airways", "IndiGo", "Air India", "Multiple carriers", "SpiceJet", "Vistara", "GoAir", "Multiple carriers Premium economy", "Jet Airways Business", "Vistara Premium economy", "Trujet"])
 
-        Journey_day = pd.to_datetime(dep_time,format="%Y-%m-%dT%H:%M").day
-        Journey_month = pd.to_datetime(dep_time,format="%Y-%m-%dT%H:%M").month
+    Source = st.selectbox("Select Source", ["Delhi", "Kolkata", "Mumbai", "Chennai"])
 
-        Departure_hour = pd.to_datetime(dep_time,format="%Y-%m-%dT%H:%M").hour
-        Departure_min = pd.to_datetime(dep_time,format="%Y-%m-%dT%H:%M").minute
+    Destination = st.selectbox("Select Destination", ["Cochin", "Delhi", "Hyderabad", "Kolkata", "Banglore"])
 
-        arrival_time = request.form['Arrival_Time']
-        Arrival_hour =  pd.to_datetime(arrival_time,format="%Y-%m-%dT%H:%M").hour
-        Arrival_min =  pd.to_datetime(arrival_time,format="%Y-%m-%dT%H:%M").minute
+    if st.button("Predict"):
+        Journey_day = pd.to_datetime(dep_time, format="%Y-%m-%dT%H:%M").day
+        Journey_month = pd.to_datetime(dep_time, format="%Y-%m-%dT%H:%M").month
 
-        Total_stops = int(request.form['stops'])
+        Departure_hour = pd.to_datetime(dep_time, format="%Y-%m-%dT%H:%M").hour
+        Departure_min = pd.to_datetime(dep_time, format="%Y-%m-%dT%H:%M").minute
 
-        dur_hour = abs(Arrival_hour-Departure_hour)
-        dur_min = abs(Arrival_min-Departure_min)
+        Arrival_hour = pd.to_datetime(arrival_time, format="%Y-%m-%dT%H:%M").hour
+        Arrival_min = pd.to_datetime(arrival_time, format="%Y-%m-%dT%H:%M").minute
 
-        airline=request.form['airline']
-        if(airline=='Jet Airways'):
-            Jet_Airways = 1
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0 
+        Jet_Airways, IndiGo, Air_India, Multiple_carriers, SpiceJet, Vistara, GoAir, Multiple_carriers_Premium_economy, Jet_Airways_Business, Vistara_Premium_economy, Trujet = [0] * 11
 
-        elif (airline=='IndiGo'):
-            Jet_Airways = 0
+        if airline == 'Jet Airways':
+            Jet_Airways = 1
+        elif airline == 'IndiGo':
             IndiGo = 1
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0 
-
-        elif (airline=='Air India'):
-            Jet_Airways = 0
-            IndiGo = 0
+        elif airline == 'Air India':
             Air_India = 1
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0 
-            
-        elif (airline=='Multiple carriers'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
+        elif airline == 'Multiple carriers':
             Multiple_carriers = 1
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0 
-            
-        elif (airline=='SpiceJet'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
+        elif airline == 'SpiceJet':
             SpiceJet = 1
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0 
-            
-        elif (airline=='Vistara'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
+        elif airline == 'Vistara':
             Vistara = 1
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0
-
-        elif (airline=='GoAir'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
+        elif airline == 'GoAir':
             GoAir = 1
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0
-
-        elif (airline=='Multiple carriers Premium economy'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
+        elif airline == 'Multiple carriers Premium economy':
             Multiple_carriers_Premium_economy = 1
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0
-
-        elif (airline=='Jet Airways Business'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
+        elif airline == 'Jet Airways Business':
             Jet_Airways_Business = 1
-            Vistara_Premium_economy = 0
-            Trujet = 0
-
-        elif (airline=='Vistara Premium economy'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
+        elif airline == 'Vistara Premium economy':
             Vistara_Premium_economy = 1
-            Trujet = 0
-            
-        elif (airline=='Trujet'):
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
+        elif airline == 'Trujet':
             Trujet = 1
 
-        else:
-            Jet_Airways = 0
-            IndiGo = 0
-            Air_India = 0
-            Multiple_carriers = 0
-            SpiceJet = 0
-            Vistara = 0
-            GoAir = 0
-            Multiple_carriers_Premium_economy = 0
-            Jet_Airways_Business = 0
-            Vistara_Premium_economy = 0
-            Trujet = 0
-        
-        Source = request.form["Source"]
-        if (Source == 'Delhi'):
-            s_Delhi = 1
-            s_Kolkata = 0
-            s_Mumbai = 0
-            s_Chennai = 0
+        s_Delhi, s_Kolkata, s_Mumbai, s_Chennai = 0, 0, 0, 0
 
-        elif (Source == 'Kolkata'):
-            s_Delhi = 0
+        if Source == 'Delhi':
+            s_Delhi = 1
+        elif Source == 'Kolkata':
             s_Kolkata = 1
-            s_Mumbai = 0
-            s_Chennai = 0
-
-        elif (Source == 'Mumbai'):
-            s_Delhi = 0
-            s_Kolkata = 0
+        elif Source == 'Mumbai':
             s_Mumbai = 1
-            s_Chennai = 0
-
-        elif (Source == 'Chennai'):
-            s_Delhi = 0
-            s_Kolkata = 0
-            s_Mumbai = 0
+        elif Source == 'Chennai':
             s_Chennai = 1
 
-        else:
-            s_Delhi = 0
-            s_Kolkata = 0
-            s_Mumbai = 0
-            s_Chennai = 0
-
+        d_Cochin, d_Delhi, d_Hyderabad, d_Kolkata = 0, 0, 0, 0
 
-        Destination = request.form["Destination"]
-        if (Destination == 'Cochin'):
+        if Destination == 'Cochin':
             d_Cochin = 1
-            d_Delhi = 0
-            d_Hyderabad = 0
-            d_Kolkata = 0
-        
-        elif (Destination == 'Delhi'):
-            d_Cochin = 0
+        elif Destination == 'Delhi':
             d_Delhi = 1
-            d_Hyderabad = 0
-            d_Kolkata = 0
-
-        elif (Destination == 'Hyderabad'):
-            d_Cochin = 0
-            d_Delhi = 0
+        elif Destination == 'Hyderabad':
             d_Hyderabad = 1
-            d_Kolkata = 0
-
-        elif (Destination == 'Kolkata'):
-            d_Cochin = 0
-            d_Delhi = 0
-            d_Hyderabad = 0
+        elif Destination == 'Kolkata':
             d_Kolkata = 1
 
-        else:#Banglore
-            d_Cochin = 0
-            d_Delhi = 0
-            d_Hyderabad = 0
-            d_Kolkata = 0
-
-        output = model.predict([[Total_stops,
-            Journey_day,
-            Journey_month,
-            Departure_hour,
-            Departure_min,
-            Arrival_hour,
-            Arrival_min,
-            dur_hour,
-            dur_min,
-            Air_India,
-            GoAir,
-            IndiGo,
-            Jet_Airways,
-            Jet_Airways_Business,
-            Multiple_carriers,
-            Multiple_carriers_Premium_economy,
-            SpiceJet,
-            Trujet,
-            Vistara,
-            Vistara_Premium_economy,
-            s_Chennai,
-            s_Delhi,
-            s_Kolkata,
-            s_Mumbai,
-            d_Cochin,
-            d_Delhi,
-            d_Hyderabad,
-            d_Kolkata]])
-
-        output = round(output[0],2)
-        return render_template('home.html',predictions='You will have to Pay approx Rs. {}'.format(output))
+        dur_hour = abs(Arrival_hour - Departure_hour)
+        dur_min = abs(Arrival_min - Departure_min)
+
+        prediction = model.predict([[Total_stops,
+                                     Journey_day,
+                                     Journey_month,
+                                     Departure_hour,
+                                     Departure_min,
+                                     Arrival_hour,
+                                     Arrival_min,
+                                     dur_hour,
+                                     dur_min,
+                                     Air_India,
+                                     GoAir,
+                                     IndiGo,
+                                     Jet_Airways,
+                                     Jet_Airways_Business,
+                                     Multiple_carriers,
+                                     Multiple_carriers_Premium_economy,
+                                     SpiceJet,
+                                     Trujet,
+                                     Vistara,
+                                     Vistara_Premium_economy,
+                                     s_Chennai,
+                                     s_Delhi,
+                                     s_Kolkata,
+                                     s_Mumbai,
+                                     d_Cochin,
+                                     d_Delhi,
+                                     d_Hyderabad,
+                                     d_Kolkata]])
+
+        prediction = round(prediction[0], 2)
+        st.success('Predicted Price: Rs. {}'.format(prediction))
 
 
 if __name__ == '__main__':
-	app.run(debug=True)
+    main()