Add title and custom symbol

app.py

@@ -8,16 +8,24 @@ import plotly.graph_objects as go
 import statsmodels.api as sm
 from statsmodels.tsa.arima.model import ARIMA
 from prophet import Prophet
+import pytz
 
 
+st.title("Vietnam Trading by Prophet")
+sb = st.text_input('Symbol', 'FPT')
+periods = st.slider('period', 1, 365, 28)
+
+start_date = str((datetime.now(pytz.timezone('Asia/Ho_Chi_Minh')) - timedelta(days=365)).strftime("%Y-%m-%d"))
+end_date = str((datetime.now(pytz.timezone('Asia/Ho_Chi_Minh')) - timedelta(days=0)).strftime("%Y-%m-%d"))
+loader = dt.DataLoader(symbol, start_date, end_date)
+data = loader.download()
+data.columns = [col[0] for col in data.columns]
+m = Prophet()
+pdf = pd.DataFrame()
+pdf['ds'] = data.index 
+pdf['y'] = data.close.values
+
 def prophet_ts(symbol, periods = 10):
-    loader = dt.DataLoader(symbol, '2020-01-01','2022-06-01')
-    data = loader.download()
-    data.columns = [col[0] for col in data.columns]
-    m = Prophet()
-    pdf = pd.DataFrame()
-    pdf['ds'] = data.index 
-    pdf['y'] = data.close.values
     m.fit(pdf)
     future = m.make_future_dataframe(periods=periods)
     forecast = m.predict(future)
@@ -31,8 +39,5 @@ def prophet_ts(symbol, periods = 10):
                                  name = f"{symbol}_pred"
                                 ))
     return fig
-st.title("Vietnam Trading by Prophet")
-sb = st.text_input('Symbol', 'FPT')
-periods = st.slider('period', 1, 365, 28)
 fig = prophet_ts(symbol=sb, periods = periods)
 st.plotly_chart(fig, use_container_width=True)