bug fix

app.py

@@ -57,9 +57,11 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker, mapping):
 def predict(_model, _dataloader, datepicker):
     out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
     preds = raw_preds_to_df(out, quantiles = None)
+
+    def add_dates(group, datepicker):
+        return group["date_imputed"] = [datepicker + datetime.timedelta(days=x) for x in range(30)]
     
-    date_list = [datepicker + datetime.timedelta(days=x) for x in range(30)]
-    preds["date_imputed"] = date_list
+    preds["date_imputed"] = preds.groupby("Group").transform(add_dates)
     
     return preds[["date_imputed", "Group", "pred"]]