Spaces:
Sleeping
Sleeping
| 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! | |
| ''' | |
| ) | |