Upload folder using huggingface_hub

app.py

@@ -1,29 +1,42 @@
 
+import joblib
+import numpy as np
+import pandas as pd
 from flask import Flask, request, jsonify
 
 superkart_api = Flask(__name__)
 
-@superkart_api.route("/", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"])
-@superkart_api.route("/predict", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"])
-@superkart_api.route("/test", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"])
-def catch_all():
-    method = request.method
-    path = request.path
-    try:
-        json_data = request.get_json(force=True)
-    except Exception:
-        json_data = None
+print("🔄 Loading model...")
+model = joblib.load("superkart_model.joblib")
+print("✅ Model loaded.")
 
-    print(f"📡 {method} request to {path}")
-    print(f"📥 Payload: {json_data}")
-
-    return jsonify({
-        "status": "received",
-        "method": method,
-        "path": path,
-        "payload": json_data
-    })
+@superkart_api.route("/", methods=["GET", "POST"])
+def index():
+    print(f"📡 Incoming {request.method} request to /")
+    if request.method == "POST":
+        try:
+            data = request.get_json(force=True)
+            print(f"📥 Received JSON: {data}")
+            input_dict = {
+                "Product_Weight": data.get("Product_Weight", 0.0),
+                "Product_Sugar_Content": data.get("Product_Sugar_Content", "Regular"),
+                "Product_Allocated_Area": data.get("Product_Allocated_Area", 0.0),
+                "Product_Type": data.get("Product_Type", "Other"),
+                "Product_MRP": data.get("Product_MRP", 0.0),
+                "Store_Establishment_Year": data.get("Store_Establishment_Year", 2000),
+                "Store_Size": data.get("Store_Size", "Medium"),
+                "Store_Location_City_Type": data.get("Store_Location_City_Type", "Tier 2"),
+                "Store_Type": data.get("Store_Type", "Supermarket Type1")
+            }
+            input_df = pd.DataFrame([input_dict])
+            prediction = model.predict(input_df)
+            return jsonify({"prediction": float(prediction[0])})
+        except Exception as e:
+            print(f"❌ Error: {str(e)}")
+            return jsonify({"error": str(e)}), 400
+    else:
+        return "✅ SuperKart API root: ready."
 
 if __name__ == "__main__":
-    print("🚀 Starting universal logger Flask server")
+    print("🚀 Starting Flask server...")
     superkart_api.run(host="0.0.0.0", port=7860)