main.py

'''
App entrypoint
'''
import streamlit as st

# pylint: disable=wildcard-import unused-wildcard-import
from utils.datetime import *
from utils.yfinance import load_data, load_indices
from utils.util import scale_indices
from utils.fbprophet import forecast_raw
from utils.plotly import live_plot, add_indices
from fbprophet.plot import plot_plotly
from plotly.subplots import make_subplots

st.set_page_config(layout='wide')
st.title('MUNE: Stock Analytics')
ticker = st.text_input('Enter a ticker', 'SHOP')
end_time = st.slider('Forecast end date:', TWO_DAYS_FROM_TODAY,
                    TWO_YEARS_FROM_TODAY, value=SIXTY_DAYS_FROM_TODAY)

load_state = st.text('Loading data...')

data_start = data_start_date(end_time)
data = load_data(ticker, data_start)
indices_data = scale_indices(load_indices(data_start), data)

st.subheader('Live Data')
fig = make_subplots(specs=[[{'secondary_y': True}]])
fig = add_indices(fig, indices_data)
live_fig = live_plot(fig, data)
st.plotly_chart(live_fig, use_container_width=True)

load_state.text('Loading data... Done!')


load_state.text('Training data...')
m, forecast = forecast_raw(data, total_duration(end_time))
load_state.text('Training data... Done!')

st.subheader('Price Forecast')

fig = plot_plotly(m, forecast)
fig.update_layout(width=1900, height=600,
    margin={ 'l': 40, 'r': 40, 't': 10, 'b': 10 })
st.plotly_chart(fig, use_container_width=True)