import requests
import streamlit as st
import pandas as pd

# Set the title of the Streamlit app
st.header("SuperKart Product Sales Prediction")

# Section for online prediction
st.subheader("Online Product Sales Prediction")

# Collect user input for product and store features
Product_Id=st.text_input("Product Id")
Product_Weight=st.number_input("Product Weight")
Product_Sugar_Content=st.selectbox("Product Sugar Content",['No Sugar', 'Low Sugar', 'Regular'])
Product_Allocated_Area=st.number_input("Product Allocated Area",min_value=0.0,max_value=1.0,value=0.001,step=0.001)# Allows increments of 0.001
Product_Type=st.selectbox("Product Type",['Frozen Foods', 'Dairy', 'Canned', 'Baking Goods',
       'Health and Hygiene', 'Snack Foods', 'Meat', 'Household',
       'Hard Drinks', 'Fruits and Vegetables', 'Breads', 'Soft Drinks',
       'Breakfast', 'Others', 'Starchy Foods', 'Seafood'])
Product_MRP=st.number_input("Product MRP")
Store_Id=st.selectbox("Store Id",['OUT001', 'OUT002', 'OUT003', 'OUT004'])
Store_Size=st.selectbox("Store Size",['Small', 'Medium', 'High'])
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',
       'Food Mart'])
Store_Establishment_Year=st.number_input("Store Establishment Year",min_value=1980, max_value=2025, value=1987)

# Convert user input into a DataFrame
input_data=pd.DataFrame([{
    'Product_Weight':Product_Weight,
    'Product_Sugar_Content':Product_Sugar_Content,
    'Product_Allocated_Area':Product_Allocated_Area,
    'Product_Type':Product_Type,
    'Product_MRP':Product_MRP,
    'Store_Id':Store_Id,
    'Store_Size':Store_Size,
    'Store_Location_City_Type':Store_Location_City_Type,
    'Store_Type':Store_Type,
    'Store_Establishment_Year':Store_Establishment_Year
}])
# Make prediction when the "Predict" button is clicked
if st.button("Predict"):
  response = requests.post("https://Parthi07-SuperKartProductPricePrediction.hf.space/v1/sales", json=input_data.to_dict(orient='records')[0])
  if response.status_code == 200:
    prediction = response.json()['Predicted Product Sales Price']
    st.success(f"Product Sales for Product ID {Product_Id} is {prediction:,.2f}")
  else:
    st.error(f"Error making prediction: {response.status_code}")


# Section for batch prediction
st.subheader("Batch Product Sales Prediction")

# Allow users to upload a CSV file for batch prediction
uploaded_file = st.file_uploader("Upload a CSV file for Batch Product Sales Prediction", type=["csv"])

# Make batch prediction when the "Predict Batch" button is clicked
if uploaded_file is not None:
  if st.button("Predict Batch"):
    response = requests.post("https://Parthi07-SuperKartProductPricePrediction.hf.space/v1/salesbatch", files={'file': uploaded_file})
    if response.status_code == 200:
      prediction = response.json()
      st.success("Batch Product Sales Prediction Successful!")
      st.write(prediction)
    else:
      st.error(f"Error making batch prediction: {response.status_code}")