Update app.py

app.py

@@ -346,14 +346,24 @@ def timegpt_fcst():
 
         df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
 
+        st.write(df)
         id_col = 'ts_test'
         df['unique_id']=id_col
+        df = df[['unique_id','ds','y']]
+        st.session_state.df = df
+        
+        st.session_state.ds_col = ds_col
+        st.session_state.y_col = y_col
+        
+
         freq = determine_frequency(df)
 
         if st.sidebar.button("Submit"):
             forecast_df = nixtla_client.forecast(
                 df=df,
                 h=7,
+                time_col='ds',
+                target_col='y',
                 freq=freq,
                 level=[90]
             )
@@ -394,11 +404,17 @@ def timegpt_anom():
 
         id_col = 'ts_test'
         df['unique_id']=id_col
-        freq = determine_frequency(df)
+        df = df[['unique_id','ds','y']]
+        st.session_state.df = df
+        
+        st.session_state.ds_col = ds_col
+        st.session_state.y_col = y_col
 
         if st.sidebar.button("Submit"):
             anom_df = nixtla_client.detect_anomalies(
                 df=df,
+                time_col='ds',
+                target_col='y',
                 freq=freq,
                 level=90
             )