superkart-project / src /streamlit_app.py
labhara's picture
Update src/streamlit_app.py
b980a35 verified
import streamlit as st
import requests
st.title("🛒 SuperKart Sales Predictor")
# Input fields
inputs = {
'Product_Weight': st.number_input("Weight (kg)", value=12.0),
'Product_Allocated_Area': st.number_input("Allocated Area", value=0.05),
'Product_MRP': st.number_input("MRP (₹)", value=150.0),
'Store_Age': st.number_input("Store Age", value=15),
'Product_Sugar_Content': st.selectbox("Sugar Content", ['Low', 'Regular', 'No Sugar']),
'Product_Type': st.selectbox("Product Type", ['Dairy', 'Meat', 'Snack Foods', 'Others']),
'Store_Size': st.selectbox("Store Size", ['Small', 'Medium', 'High']),
'Store_Location_City_Type': st.selectbox("City Tier", ['Tier 1', 'Tier 2', 'Tier 3']),
'Store_Type': st.selectbox("Store Type", ['Departmental Store', 'Supermarket Type1', 'Food Mart'])
}
# Prediction button
if st.button("Predict Sales"):
try:
#response = requests.post("http://host.docker.internal:7860/predict", json=inputs)
response = requests.post("https://labhara-superkart-backend.hf.space/predict", json=inputs)
if response.ok:
result = response.json()["predicted_sales"]
st.success(f"💰 Predicted Sales: ₹{result}")
else:
st.error("Failed to get prediction.")
except:
st.error("Could not reach backend.")