File size: 2,492 Bytes
b753fd3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
# Streamlit frontend that collects product/store details and calls the Flask backend API
import streamlit as st
import requests
# Backend URL: within the shared Docker network, containers reach each other
# by container name. Defaults to the container name used when running the
# backend container (see deployment instructions).
BACKEND_URL = os.environ.get("BACKEND_URL", "http://superkart-backend:7860")
st.set_page_config(page_title="SuperKart Sales Forecasting", layout="centered")
st.title("SuperKart Sales Forecasting")
st.write("Enter the product and store details below to forecast sales revenue.")
# ----- Input widgets -----
product_weight = st.number_input("Product Weight", min_value=0.0, value=12.66, step=0.1)
product_sugar_content = st.selectbox("Product Sugar Content", ["Low Sugar", "Regular", "No Sugar"])
product_allocated_area = st.number_input(
"Product Allocated Area", min_value=0.0, max_value=1.0, value=0.027, step=0.001, format="%.3f"
)
product_mrp = st.number_input("Product MRP", min_value=0.0, value=117.08, step=0.1)
store_size = st.selectbox("Store Size", ["High", "Medium", "Small"])
store_location_city_type = st.selectbox("Store Location City Type", ["Tier 1", "Tier 2", "Tier 3"])
store_type = st.selectbox(
"Store Type",
["Departmental Store", "Supermarket Type1", "Supermarket Type2", "Supermarket Type3", "Food Mart"],
)
product_id_char = st.selectbox("Product Id Prefix", ["FD", "DR", "NC"])
store_age_years = st.number_input("Store Age (Years)", min_value=0, value=16, step=1)
product_type_category = st.selectbox("Product Type Category", ["Perishables", "Non Perishables"])
if st.button("Predict Sales"):
payload = {
"Product_Weight": product_weight,
"Product_Sugar_Content": product_sugar_content,
"Product_Allocated_Area": product_allocated_area,
"Product_MRP": product_mrp,
"Store_Size": store_size,
"Store_Location_City_Type": store_location_city_type,
"Store_Type": store_type,
"Product_Id_char": product_id_char,
"Store_Age_Years": store_age_years,
"Product_Type_Category": product_type_category,
}
try:
response = requests.post(f"{BACKEND_URL}/v1/predict", json=payload)
response.raise_for_status()
prediction = response.json()["predicted_Product_Store_Sales_Total"]
st.success(f"Predicted Sales: {prediction}")
except Exception as e:
st.error(f"Error while calling the backend API: {e}")
|