Update app.py

app.py

@@ -9,6 +9,7 @@ from neuralforecast.utils import AirPassengersDF
 import time
 from st_aggrid import AgGrid
 from nixtla import NixtlaClient
+import os
 
 
 @st.cache_resource
@@ -208,7 +209,7 @@ def transfer_learning_forecasting():
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
         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)
+        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")
 
     st.session_state.ds_col = ds_col
@@ -290,7 +291,7 @@ def dynamic_forecasting():
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
         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)
+        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")
 
     df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
@@ -331,7 +332,7 @@ def timegpt_fcst():
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
         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)
+        y_col = st.selectbox("Select Target column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
 
         df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
 
@@ -372,7 +373,7 @@ def timegpt_anom():
         # Column selection
         columns = df.columns.tolist()  # Convert Index to list
         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)
+        y_col = st.selectbox("Select Target column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
 
         df = df.rename(columns={ds_col: 'ds', y_col: 'y'})