Frontend / app.py
vallabbharath's picture
Force rebuild
6926a5e verified
Raw
History Blame Contribute Delete
2.53 kB
import streamlit as st
import pandas as pd
import requests
# Hugging Face backend API endpoint
API_URL = "https://vallabbharath-Backend.hf.space/predict"
st.set_page_config(page_title="Sales Prediction App", page_icon="πŸ›οΈ", layout="wide")
st.title("πŸ›’ Sales Prediction Dashboard")
st.markdown("""
Upload or enter product and store details below to predict expected sales.
""")
# --- Input Section ---
st.subheader("Enter Input Data")
# You can also adjust these based on your backend model features
col1, col2 = st.columns(2)
with col1:
product_weight = st.number_input("Product Weight", min_value=0.0, step=0.1)
product_sugar = st.selectbox("Product Sugar Content", ['Low Sugar', 'No Sugar', 'Regular', 'reg'])
product_type = st.selectbox("Product Type", ['Baking Goods','Breads','Breakfast','Canned','Dairy',
'Frozen Foods','Fruits and Vegetables','Hard Drinks','Health and Hygiene',
'Household','Meat','Others','Seafood','Snack Foods','Soft Drinks','Starchy Foods'])
product_mrp = st.number_input("Product MRP", min_value=0.0, step=0.1)
product_allocated_area = st.number_input("Product Allocated Area", min_value=0.001, step=0.001)
with col2:
store_id = st.text_input("Store ID")
store_est_year = st.number_input("Store Establishment Year", min_value=1900, max_value=2025, value=2010)
store_size = st.selectbox("Store Size", ['High', 'Medium', 'Small'])
store_loc = st.selectbox("Store Location City Type", ['Tier 1', 'Tier 2', 'Tier 3'] )
store_type = st.selectbox("Store Type", ['Departmental Store', 'Food Mart', 'Supermarket Type1', 'Supermarket Type2'])
if st.button("Predict"):
# Prepare data for backend
data = {
"Product_Weight": product_weight,
"Product_Sugar_Content": product_sugar,
"Product_Allocated_Area": product_allocated_area, # if needed, or remove if not used
"Product_Type": product_type,
"Product_MRP": product_mrp,
"Store_Id": store_id,
"Store_Establishment_Year": int(store_est_year),
"Store_Size": store_size,
"Store_Location_City_Type": store_loc,
"Store_Type": store_type
}
try:
response = requests.post(API_URL, json=data)
if response.status_code == 200:
result = response.json()
st.success(f"🟒 Predicted Sales: **{result['prediction']:.2f}**")
else:
st.error(f"Backend Error: {response.text}")
except Exception as e:
st.error(f"Error connecting to backend: {e}")