Update app.py

app.py

@@ -197,7 +197,7 @@ def transfer_learning_forecasting():
             st.session_state.df = df
         
         # Column selection
-        columns = df.columns if not df.empty else []
+        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, index=columns.index('y') if 'y' in columns else 1)
         unique_id_col = st.text_input("Unique ID column (default: '1')", value="1")
@@ -252,7 +252,7 @@ def dynamic_forecasting():
             st.session_state.df = df
         
         # Column selection
-        columns = df.columns if not df.empty else []
+        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, index=columns.index('y') if 'y' in columns else 1)
         unique_id_col = st.text_input("Unique ID column (default: '1')", value="1")
@@ -284,3 +284,4 @@ try:
     pg.run()
 except Exception as e:
     st.sidebar.error(f"Something went wrong: {e}", icon=":material/error:")
+