Update app.py

app.py

@@ -54,6 +54,7 @@ def insertintotable():
 
   
 
+
     def ARIMA_ALGO(df):
         # — 1) Ensure a DateTimeIndex —
         if 'Date' in df.columns:
@@ -79,12 +80,13 @@ def insertintotable():
         # — 4) Train/test split (65/35) —
         split = int(len(series) * 0.65)
         train, test = series.iloc[:split], series.iloc[split:]
-
+        
         # — 5) Rolling ARIMA + 7-day ahead —
         history, preds_all = list(train), []
-        for t in range(len(test) + 7):
-            m = ARIMA(history, order=(6,1,0)).fit()
-            yhat = float(m.forecast()[0])
+        m = ARIMA(history, order=(6,1,0)).fit()
+        for t in range(len(test) + 7): 
+            model = m.apply(history)
+            yhat = float(model.forecast()[0])
             preds_all.append(yhat)
             history.append(test.iloc[t] if t < len(test) else yhat)
 
@@ -104,7 +106,6 @@ def insertintotable():
 
         return preds_all, rmse, tomorrow, history_df, predict_df
 
-
     # Run ARIMA
     preds, rmse, tomorrow, hist_df, pred_df = ARIMA_ALGO(df)