Update streamlit_app.py

streamlit_app.py

@@ -11,9 +11,13 @@ BACKEND_URL = "https://dhani10-SuperKart-Backend.hf.space/predict"
 @st.cache_data
 def load_dropdown_data():
     df = pd.read_csv("SuperKart.csv")
-    product_ids = sorted(df['Product_Id'].dropna().unique())
-    store_ids = sorted(df['Store_Id'].dropna().unique())
-    years = sorted(df['Store_Establishment_Year'].dropna().unique())
+    # product_ids = sorted(df['Product_Id'].dropna().unique())
+    # store_ids = sorted(df['Store_Id'].dropna().unique())
+    # years = sorted(df['Store_Establishment_Year'].dropna().unique())
+    product_ids = sorted(df['Product_Id'].dropna().astype(str).unique())
+    store_ids = sorted(df['Store_Id'].dropna().astype(str).unique())
+    years = sorted(df['Store_Establishment_Year'].dropna().astype(int).unique())
+
     return product_ids, store_ids, years, df
 
 # Page Title
@@ -52,20 +56,34 @@ with st.form("prediction_form"):
 
 # Prediction logic
 if submit:
+    # data = {
+    #     "Product_Id": product_id,
+    #     "Product_Weight": product_weight,
+    #     "Product_Sugar_Content": sugar_content,
+    #     "Product_Allocated_Area": allocated_area,
+    #     "Product_Type": product_type,
+    #     "Product_MRP": mrp,
+    #     "Store_Id": store_id,
+    #     "Store_Establishment_Year": establishment_year,
+    #     "Store_Size": store_size,
+    #     "Store_Location_City_Type": city_type,
+    #     "Store_Type": store_type
+    # }
     data = {
-        "Product_Id": product_id,
-        "Product_Weight": product_weight,
-        "Product_Sugar_Content": sugar_content,
-        "Product_Allocated_Area": allocated_area,
-        "Product_Type": product_type,
-        "Product_MRP": mrp,
-        "Store_Id": store_id,
-        "Store_Establishment_Year": establishment_year,
-        "Store_Size": store_size,
-        "Store_Location_City_Type": city_type,
-        "Store_Type": store_type
+    "Product_Id": str(product_id),
+    "Product_Weight": float(product_weight),
+    "Product_Sugar_Content": str(sugar_content),
+    "Product_Allocated_Area": float(allocated_area),
+    "Product_Type": str(product_type),
+    "Product_MRP": float(mrp),
+    "Store_Id": str(store_id),
+    "Store_Establishment_Year": int(establishment_year),  # FIXED LINE
+    "Store_Size": str(store_size),
+    "Store_Location_City_Type": str(city_type),
+    "Store_Type": str(store_type)
     }
 
+
     try:
         response = requests.post(BACKEND_URL, json=data)
         if response.status_code == 200: