Update app.py

app.py

@@ -1,13 +1,30 @@
+from flask import Flask, request, jsonify
+import pandas as pd
+import joblib
+
+#  Initialize Flask app
+app = Flask(__name__)
+
+#  Load trained model
+model = joblib.load("model.pkl")
+
+
+@app.route("/")
+def home():
+    return "Backend is running successfully"
+
+
 @app.route("/predict", methods=["POST"])
 def predict():
     try:
         data = request.get_json()
 
-        import pandas as pd
-
+        #  Convert incoming data to DataFrame
         df = pd.DataFrame([data])
 
-        #  Rename to training names
+        print("Incoming Data:", df)
+
+        #  Rename columns to match training dataset
         df.rename(columns={
             "Product_Weight": "Item_Weight",
             "Product_Sugar_Content": "Item_Fat_Content",
@@ -21,29 +38,35 @@ def predict():
             "Product_Id_char": "Item_Identifier"
         }, inplace=True)
 
-        # 🔥 FORCE correct categories (VERY IMPORTANT)
+        #  FIX STORE TYPE MISMATCH (CRITICAL)
+        df["Outlet_Type"] = df["Outlet_Type"].map({
+            "Supermarket": "Supermarket Type1",
+            "Grocery Store": "Food Mart"
+        })
+
+        #  OPTIONAL: Fix Outlet Size if needed
         df["Outlet_Size"] = df["Outlet_Size"].map({
             "Small": "Small",
             "Medium": "Medium",
             "High": "High"
         })
 
-        df["Outlet_Type"] = df["Outlet_Type"].map({
-            "Supermarket": "Supermarket Type1",
-            "Grocery Store": "Grocery Store"
-        })
-
-        #  One-hot encoding
+        #  Convert categorical → numerical
         df = pd.get_dummies(df)
 
-        #  Align with model EXACT columns
+        #  Align with model features EXACTLY
         df = df.reindex(columns=model.feature_names_in_, fill_value=0)
 
-        print("Final input to model:", df)
+        print("Processed Data:", df)
 
+        # Predict
         prediction = model.predict(df)[0]
 
         return jsonify({"Sales": float(prediction)})
 
     except Exception as e:
-        return jsonify({"error": str(e)})
+        return jsonify({"error": str(e)})
+
+
+if __name__ == "__main__":
+    app.run(debug=True)