Update flightprice.py

flightprice.py

@@ -52,8 +52,7 @@ def get_day_quarter(hour):
     else:
         return 'Night'
 
-# --- 2. MAPPING FOR VALIDATION ONLY ---
-# We use this to help check if the Destination is in the Route string
+# --- 2. MAPPINGS (For UI Display Only) ---
 CITY_TO_CODE = {
     'Banglore': 'BLR', 'Bangalore': 'BLR',
     'Delhi': 'DEL', 'New Delhi': 'DEL',
@@ -68,20 +67,6 @@ CITY_TO_CODE = {
 def get_code(city):
     return CITY_TO_CODE.get(city.strip(), city[:3].upper())
 
-def is_route_valid(route_str, destination):
-    """
-    Checks if the Destination (Name or Code) appears in the Route string.
-    """
-    if pd.isna(route_str) or route_str == 'nan':
-        return False
-        
-    route_upper = route_str.upper()
-    dest_name_upper = destination.upper()
-    dest_code = get_code(destination)
-    
-    # Check if code (e.g., 'COK') or name (e.g., 'COCHIN') is in string
-    return (dest_code in route_upper) or (dest_name_upper in route_upper)
-
 # Load Data
 x_train, x_test, y_test = load_data()
 model = load_model()
@@ -108,15 +93,20 @@ if page == "💰 Price Prediction":
     st.title("✈️ Flight Price Prediction")
     st.markdown("### Enter Flight Details")
     
+    # ---------------------------------------------------------
+    # INPUT SECTION (Reactive - No Form)
+    # ---------------------------------------------------------
     c1, c2 = st.columns(2)
     with c1:
         st.subheader("Flight Info")
         
-        # Source (Updates instantly)
+        # 1. Source Selection
         source = st.selectbox("Source", sorted(x_train['Source'].unique()))
+        st.success(f"🛫 Code: **{get_code(source)}**")
         
-        # Destination (Updates instantly)
+        # 2. Destination Selection
         destination = st.selectbox("Destination", sorted(x_train['Destination'].unique()))
+        st.error(f"🛬 Code: **{get_code(destination)}**")
         
         airline = st.selectbox("Airline", sorted(x_train['Airline'].unique()))
         
@@ -137,37 +127,42 @@ if page == "💰 Price Prediction":
     stops_val = 0
     
     if 'Route' in x_train.columns:
-        # --- 1. SMART FILTERING ---
-        # Instead of guessing codes, we ask the DataFrame:
-        # "Give me all routes that exist for this specific Source"
-        valid_source_routes = x_train[x_train['Source'] == source]['Route'].unique()
-        valid_source_routes = sorted(valid_source_routes.astype(str))
+        # ---------------------------------------------------------
+        # DOUBLE FILTER LOGIC (Source AND Destination)
+        # ---------------------------------------------------------
+        # We query the dataframe for rows matching BOTH selected Source and Destination
+        filtered_routes = x_train[
+            (x_train['Source'] == source) & 
+            (x_train['Destination'] == destination)
+        ]['Route'].unique()
         
-        if len(valid_source_routes) > 0:
-            selected_route_raw = st.selectbox("Select Route", valid_source_routes)
+        filtered_routes = sorted(filtered_routes.astype(str))
+        
+        if len(filtered_routes) > 0:
+            # If routes exist, user selects one
+            selected_route = st.selectbox("Select Route", filtered_routes)
+            
+            # Lookup hidden features based on this valid route
+            stops_val = route_lookup.get(selected_route, 0)
+            
+            # Display Feedback
+            c_a, c_b = st.columns(2)
+            c_a.metric("Total Stops", stops_val)
+            c_b.success("✅ Valid Route Found")
             
-            # --- 2. VALIDATION ---
-            # Check if the route actually contains the destination
-            if is_route_valid(selected_route_raw, destination):
-                selected_route = selected_route_raw
-                stops_val = route_lookup.get(selected_route, 0)
-                st.metric("Total Stops", stops_val)
-                st.success("✅ Valid Route")
-            else:
-                # Warning if user picks a route that doesn't go to their destination
-                st.warning(f"⚠️ **{selected_route_raw}** does not seem to contain **{destination}** ({get_code(destination)}). Please confirm if this is correct.")
-                # We allow them to proceed but with a warning, or you can block it:
-                # selected_route = None 
-                selected_route = selected_route_raw 
-                stops_val = route_lookup.get(selected_route, 0)
-                
         else:
-            st.error(f"No routes found in database for Source: {source}")
+            # No routes connect these two cities in the training data
+            st.error(f"🚫 No flights found from **{source}** to **{destination}** in the database.")
+            st.info("Try changing the Destination or Source.")
+            selected_route = None
     else:
         st.error("Route column missing.")
 
     st.markdown("<br>", unsafe_allow_html=True)
     
+    # ---------------------------------------------------------
+    # PREDICTION BUTTON
+    # ---------------------------------------------------------
     if st.button("Predict Price", type="primary"):
         if selected_route:
             # Prepare Input
@@ -182,12 +177,12 @@ if page == "💰 Price Prediction":
                 'Total_Stops': [stops_val]
             })
             
-            # Align Cols
+            # Align Columns (Fill missing with 0)
             final_input = pd.DataFrame(columns=x_train.columns)
             for col in x_train.columns:
                 final_input.loc[0, col] = input_df.iloc[0].get(col, 0)
                 
-            # Types
+            # Ensure Types
             for col in final_input.columns:
                 if x_train[col].dtype == 'object':
                     final_input[col] = final_input[col].astype(str)
@@ -195,12 +190,21 @@ if page == "💰 Price Prediction":
                     final_input[col] = pd.to_numeric(final_input[col])
 
             try:
+                # Predict
                 pred = model.predict(final_input)[0]
-                st.success(f"### Estimated Price: ₹ {np.expm1(pred):,.2f}")
+                real_price = np.expm1(pred)
+                
+                # Show Result
+                st.markdown(f"""
+                <div style="background-color:#d4edda;padding:10px;border-radius:10px;border:2px solid #c3e6cb">
+                    <h2 style="color:#155724;text-align:center;">🎫 Estimated Price: ₹ {real_price:,.2f}</h2>
+                </div>
+                """, unsafe_allow_html=True)
+                
             except Exception as e:
                 st.error(f"Error: {e}")
         else:
-            st.error("Please select a valid route.")
+            st.error("Cannot predict without a valid route.")
 
 elif page == "📊 Model Evaluation":
     st.title("Model Evaluation")