Update app.py

app.py

@@ -207,7 +207,9 @@ def transfer_learning_forecasting():
             st.session_state.df = df
         
         columns = df.columns.tolist()  # Convert Index to list
-        options = columns.pop('ds').pop('unique_id')
+        if 'ds' in columns and 'unique_id' in columns:
+            options = columns.remove('ds')
+            options.remove('unique_id')
         ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
         y_col = st.selectbox("Select Target column", options=options, index=0)
 
@@ -288,7 +290,9 @@ def dynamic_forecasting():
             st.session_state.df = df
         
         columns = df.columns.tolist()  # Convert Index to list
-        options = columns.pop('ds').pop('unique_id')
+        if 'ds' in columns and 'unique_id' in columns:
+            options = columns.remove('ds')
+            options.remove('unique_id')
         ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
         y_col = st.selectbox("Select Target column", options=options, index=0)
 
@@ -329,7 +333,9 @@ def timegpt_fcst():
         
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
-        options = columns.pop('ds').pop('unique_id')
+        if 'ds' in columns and 'unique_id' in columns:
+            options = columns.remove('ds')
+            options.remove('unique_id')
         ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
         y_col = st.selectbox("Select Target column", options=options, index=0)
 
@@ -371,7 +377,9 @@ def timegpt_anom():
         
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
-        options = columns.pop('ds').pop('unique_id')
+        if 'ds' in columns and 'unique_id' in columns:
+            options = columns.remove('ds')
+            options.remove('unique_id')
         ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
         y_col = st.selectbox("Select Target column", options=options, index=0)
 
@@ -381,17 +389,15 @@ def timegpt_anom():
         df['unique_id']=id_col
         freq = determine_frequency(df)
         
-        forecast_df = nixtla_client.forecast(
+        anom_df = nixtla_client.detect_anomalies(
             df=df,
-            h=7,
             freq=freq,
-            level=[90]
+            level=90
         )
 
         nixtla_client.plot(
             forecast_df,
-            level=[90],
-            max_insample_length=365
+            anom_df
         )