Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import requests
-import json
 
+# --- Streamlit UI config ---
 st.set_page_config(page_title="SuperKart Sales Prediction", layout="centered")
 
 st.title("🛒 SuperKart Sales Prediction")
@@ -23,14 +23,17 @@ product_type = st.selectbox("Product Category", [
     "Snack Foods", "Soft Drinks", "Starchy Foods"
 ])
 
-# Create one-hot encoded product types
-product_type_features = {f"Product_Type_{pt}": int(pt == product_type) for pt in [
-    "Breads", "Breakfast", "Canned", "Dairy", "Frozen Foods", "Fruits and Vegetables",
-    "Hard Drinks", "Health and Hygiene", "Household", "Meat", "Others", "Seafood",
-    "Snack Foods", "Soft Drinks", "Starchy Foods"
-]}
+# --- One-hot encode the product type ---
+product_type_features = {
+    f"Product_Type_{pt}": int(pt == product_type)
+    for pt in [
+        "Breads", "Breakfast", "Canned", "Dairy", "Frozen Foods", "Fruits and Vegetables",
+        "Hard Drinks", "Health and Hygiene", "Household", "Meat", "Others", "Seafood",
+        "Snack Foods", "Soft Drinks", "Starchy Foods"
+    ]
+}
 
-# --- PAYLOAD CREATION ---
+# --- Create input JSON ---
 input_data = {
     "Product_Weight": product_weight,
     "Product_Sugar_Content": product_sugar,
@@ -48,13 +51,18 @@ if st.button("Predict Sales"):
     with st.spinner("Fetching prediction from backend..."):
         try:
             response = requests.post(
-                "https://lokiiparihar-superkartbackendmodaldeploy-xgboost.hf.space/predict",
+                "https://lokiiparihar-SuperkartBackendModalDeploy-XGBoost.hf.space/predict",
                 json=input_data
             )
             if response.status_code == 200:
-                prediction = response.json().get("prediction", "N/A")
-                st.success(f"✅ Predicted Sales: {prediction .json()['prediction']:.2f} units")
+                result = response.json()
+                prediction = result.get("prediction", "N/A")
+                try:
+                    prediction = float(prediction)
+                    st.success(f"✅ Predicted Sales: **{prediction:.2f} units**")
+                except ValueError:
+                    st.error(f"❌ Prediction format error: {prediction}")
             else:
-                st.error(f"❌ API Error: {response.status_code}")
+                st.error(f"❌ API Error: Status code {response.status_code}")
         except Exception as e:
-            st.error(f"Request failed: {e}")
+            st.error(f"❌ Request failed: {e}")