Update app.py

app.py

@@ -40,22 +40,24 @@ elif selection == "EDA":
 elif selection == "Make a Prediction":
     st.title("Make a Sales Prediction")
     with st.form("input_form"):
+
         # Capture all inputs as per the original dataset
         order_date = st.date_input('Order Date', datetime.now())
-        ship_date = st.date_input('Ship Date', datetime.now())
-        ship_mode = st.selectbox('Ship Mode', ['First Class', 'Second Class', 'Standard Class', 'Same Day'])
-        segment = st.selectbox('Segment', ['Consumer', 'Corporate', 'Home Office'])
+        ship_date = st.date_input('Ship Date', datetime.now() + timedelta(days=1))  # Assume shipping the next day
+        ship_mode = st.selectbox('Ship Mode', ['First Class', 'Second Class', 'Standard Class', 'Same Day'], index=2)  # Default to Standard Class
+        segment = st.selectbox('Segment', ['Consumer', 'Corporate', 'Home Office'], index=0)  # Default to Consumer
         country = st.text_input('Country', value='United States')
-        city = st.text_input('City')
-        state = st.text_input('State')
-        postal_code = st.text_input('Postal Code')
-        region = st.selectbox('Region', ['South', 'West', 'Central', 'East'])
-        category = st.selectbox('Category', ['Furniture', 'Office Supplies', 'Technology'])
-        sub_category = st.selectbox('Sub-Category', ['Bookcases', 'Chairs', 'Labels', 'Tables', 'Storage', 'Furnishings', 'Art', 'Phones', 'Binders', 'Appliances', 'Paper', 'Accessories', 'Envelopes', 'Fasteners', 'Supplies', 'Machines', 'Copiers'])
-        product_name = st.text_input('Product Name')
-        sales = st.number_input('Sales', value=0.0, format="%.2f")
-        quantity = st.number_input('Quantity', value=1, format="%d")
-        discount = st.number_input('Discount', value=0.0, format="%.2f")
+        city = st.text_input('City', value='Los Angeles')  # Example city
+        state = st.text_input('State', value='California')  # Example state
+        postal_code = st.text_input('Postal Code', value='90001')  # Example postal code
+        region = st.selectbox('Region', ['South', 'West', 'Central', 'East'], index=1)  # Default to West
+        category = st.selectbox('Category', ['Furniture', 'Office Supplies', 'Technology'], index=1)  # Default to Office Supplies
+        sub_category = st.selectbox('Sub-Category', ['Bookcases', 'Chairs', 'Labels', 'Tables', 'Storage', 'Furnishings', 'Art', 'Phones', 'Binders', 'Appliances', 'Paper', 'Accessories', 'Envelopes', 'Fasteners', 'Supplies', 'Machines', 'Copiers'], index=10)  # Default to Paper
+        product_name = st.text_input('Product Name', value='Staple papers')  # Example product
+        sales = st.number_input('Sales', value=100.0, format="%.2f")  # Example sales amount
+        quantity = st.number_input('Quantity', value=2, format="%d")  # Example quantity
+        discount = st.number_input('Discount', value=0.0, format="%.2f")  # Example discount
+
 
         submit_button = st.form_submit_button("Predict")
 
@@ -63,7 +65,7 @@ elif selection == "Make a Prediction":
         # Construct the input DataFrame
         input_features = pd.DataFrame([[sub_category, sales, quantity, discount
         ]], columns=[
-            'Sub-Category', 'Sales', 'Quantity', 'Discount'
+            'Sub-Category', 'Product Name', 'Sales', 'Quantity', 'Discount'
         ])
         
         # Preprocess and predict (You'll need to adjust this part based on how your model expects input)