add arima

app.py

@@ -59,7 +59,7 @@ class TS:
         fig = go.Figure()
         fig.add_trace(go.Scatter(x= data.ds, 
                                 y=data.y,
-                                name = f"{self.symbol}_true"
+                                name = f"{self.symbol}"
                                 ))
         fig.add_trace(go.Scatter(x= future.ds, 
                                      y=future.yhat,
@@ -72,13 +72,12 @@ class TS:
         model = ARIMA(df.y, order = (1, 0, 7))
         model = model.fit()
         print(len(df))
-        future = pd.DataFrame()
+        forecast = pd.DataFrame()
         predictions = model.forecast(period)
-        future["y"] = predictions
-        print(df.index.max())
-        future.index = pd.date_range(start = df.index.max() + timedelta(days = 1), end = df.index.max() + timedelta(days = period))
+        forecast["y"] = predictions
+        forecast["ds"] = pd.date_range(start = df.index.max() + timedelta(days = 1), end = df.index.max() + timedelta(days = period))
         
-        return self.viz(df, future)
+        return self.viz(df, forecast)