penscola's picture
Update app.py
6fd2099
raw
history blame
4.71 kB
# import pandas as pd
# import numpy as np
import streamlit as st
# import os
# import time
# import pickle
# import seaborn as sns
# import matplotlib.pyplot as plt
# import pip
try:
#insert headers
st.header(" Welcome to Sales Prediction Using Prophet ")
st.subheader("To help you know your future sales📈...")
st.image("future.png", width=500, caption="Sales Prediction")
Disp_results = pd.DataFrame() # Initialize for download
# Take input
with st.form("This form", clear_on_submit=True):
st.subheader("Enter the number of day(s)/Week(s) you want to predict, And the frequency as D for Daily or W for weekly ")
frequency = str(st.text_input("Frequency 'D' for Daily 'W' for weekly ")).upper() # convert to string and change to upper
Number_of_days = int(st.number_input("Number of day(s)/Week(s)")) # convert to int
submit = st.form_submit_button("Predict your sales")
# process the input
if submit:
# check if we have the right data type
if frequency == "D" or frequency == 'W':
st.success("Inputs received successfully ✅")
# import model
with open('prophet_model.pkl', 'rb') as f:
model = pickle.load(f)
# pass inputs to the model(To make predictions, prophet requires number of days and frequency)
future = model.make_future_dataframe(periods=Number_of_days, freq=str(frequency), include_history=False)
# Make prediction
forecast = model.predict(future)
# show results
print(f'[INFO]: The whole results {forecast}')
# pick the relevant columns from the forecast
sales_forecast = forecast[['ds', 'yhat_lower', 'yhat_upper', 'yhat']]
# rename the columns
Disp_results = sales_forecast.rename(columns={'ds': 'Date', 'yhat_lower': 'lowest Expected sales', 'yhat_upper': 'Highest Expected Sales', 'yhat': 'Expected Sales'})
# print result dataframe to terminal
print(f'[INFO]: results dataframe {Disp_results}')
# show progress
with st.spinner("Prediction in progress..."):
time.sleep(2)
st.balloons()
st.success("Great✅")
# Display results
if frequency == "W":
output_frequency = 'Week(s)'
else:
output_frequency = 'Day(s)'
# Check frequency
st.write(f"These are your predicted sales in the next {Number_of_days} {output_frequency}")
st.dataframe(Disp_results)
# Display the graph of sales
st.title(f"Line Graph Of Predicted Sales Over {Number_of_days} {output_frequency} ")
# Line Graph
st.line_chart(data=Disp_results, x='Date', y='Expected Sales')
print('[INFO]: Line Chart displayed')
else:
st.error("Input the right frequency or Days ⚠")
# Print input to the terminal
print(f'[INFO]: These are the inputs to the model {Number_of_days},{frequency}')
print(f"[INFO]: Inputs received")
# Create a function to convert df to csv
def convert_to_csv(df):
return df.to_csv()
# Create an expander
expand = st.expander('Download Results as CSV')
with expand:
st.download_button(
'Download results',
convert_to_csv(Disp_results),
'prediction_results.csv',
'text/csv',
'download'
)
# Create Sidebar for Description
sidebar = st.sidebar.title('Sales Prediction')
# first option
option1 = st.sidebar.button('About', key="About")
# second option
option2 = st.sidebar.button('About the sales prediction', key="sales prediction")
# Display text for a selected option
if option1:
st.sidebar.write('This is a Sales prediction app Using Prophet(Developed by meta), this project was done under the Azubi Africa Data Analysis Training program ')
elif option2:
st.sidebar.write('This is a time series analysis & forecasting problem. In this project, we shalll predict store sales on data from Corporation Favorita, a large Ecuadorian-based grocery retailer. Specifically, this app predicts the sales for up to weeks in advance for Corporation Favorita ')
except:
st.error('''something went wrong: Make sure you entered the correct number of days
otherwise contact admin!
'''
)