add model

app.py

@@ -37,10 +37,45 @@ def prophet_ts(symbol, periods = 10):
                                  name = f"{symbol}_pred"
                                 ))
     return fig
+class TS: 
+    def __init__(self, symbol):
+        self.symbol = symbol
+    def get_data(self):
+        loader = dt.DataLoader(self.symbol, start_date, end_date)
+        data = loader.download()
+        data.columns = [col[0] for col in data.columns]
+        pdf = pd.DataFrame()
+        pdf['ds'] = data.index 
+        pdf['y'] = data.close.values
+        return pdf
+    def prophet(self, period = 28):
+        df = self.get_data()
+        model = Prophet()
+        model.fit(df)
+        future = model.make_future_dataframe(periods=period)
+        forecast = model.predict(future)
+        return self.viz(df, forecast)
+    def viz(self, data, future):
+        fig = go.Figure()
+        fig.add_trace(go.Scatter(x= data.ds, 
+                                y=data.y,
+                                name = f"{symbol}_true"
+                                ))
+        fig.add_trace(go.Scatter(x= future.ds, 
+                                     y=future.yhat,
+                                     name = f"{symbol}_pred"
+                                    ))
+        return fig
+
+
 
 st.title("Vietnam Trading by Time Series")
 model = st.selectbox("model", ["ARIMA", "Propphet"])
 sb = st.text_input('Symbol', 'FPT')
-periods = st.slider('Period', 1, 365, 28)
-fig = prophet_ts(symbol=sb, periods = periods)
+period = st.slider('Period', 1, 365, 28)
+# fig = prophet_ts(symbol=sb, periods = periods)
+ts = TS()
+fig = None
+if model == "Prophet":
+    fig = ts.prophet(period = period)
 st.plotly_chart(fig, use_container_width=True)