Update app.py

app.py

@@ -206,11 +206,10 @@ def transfer_learning_forecasting():
             df = load_default()  
             st.session_state.df = df
         
-        # Column selection
         columns = df.columns.tolist()  # Convert Index to list
+        options = columns.pop('ds').pop('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=columns, index=columns.index('ds') if 'ds' in columns else 0)
-        # unique_id_col = st.text_input("Unique ID column (default: '1')", value="1")
+        y_col = st.selectbox("Select Target column", options=options, index=0)
 
     st.session_state.ds_col = ds_col
     st.session_state.y_col = y_col
@@ -288,11 +287,10 @@ def dynamic_forecasting():
             df = load_default()
             st.session_state.df = df
         
-        # Column selection
         columns = df.columns.tolist()  # Convert Index to list
+        options = columns.pop('ds').pop('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=columns, index=columns.index('ds') if 'ds' in columns else 0)
-        # unique_id_col = st.text_input("Unique ID column (default: '1')", value="1")
+        y_col = st.selectbox("Select Target column", options=options, index=0)
 
     df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
     
@@ -315,7 +313,7 @@ def dynamic_forecasting():
 def timegpt_fcst():
     nixtla_token = os.environ.get("NIXTLA_API_KEY")
     nixtla_client = NixtlaClient(
-    api_key = api_key
+    api_key = nixtla_token
     )
 
     
@@ -331,8 +329,9 @@ def timegpt_fcst():
         
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
+        options = columns.pop('ds').pop('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=columns, index=columns.index('ds') if 'ds' in columns else 0)
+        y_col = st.selectbox("Select Target column", options=options, index=0)
 
         df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
 
@@ -356,7 +355,7 @@ def timegpt_fcst():
 def timegpt_anom():
     nixtla_token = os.environ.get("NIXTLA_API_KEY")
     nixtla_client = NixtlaClient(
-    api_key = api_key
+    api_key = nixtla_token
     )
 
     
@@ -372,8 +371,9 @@ def timegpt_anom():
         
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
+        options = columns.pop('ds').pop('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=columns, index=columns.index('ds') if 'ds' in columns else 0)
+        y_col = st.selectbox("Select Target column", options=options, index=0)
 
         df = df.rename(columns={ds_col: 'ds', y_col: 'y'})