Update pages/2_Transformations_and_Model_Build.py

pages/2_Transformations_and_Model_Build.py

@@ -73,7 +73,7 @@ st.title('1. Build Your Model')
 date_col = 'date'
 
 
-media_data=pd.read_excel(r'upf_data_converted.xlsx')
+media_data=pd.read_excel('upf_data_converted.xlsx')
 # with open("Pickle_files/main_df",'rb') as f:
 #    media_data= pickle.load(f)
    
@@ -94,13 +94,13 @@ media_data.drop([date_col],axis=1,inplace=True)
 media_data.reset_index(drop=True,inplace=True)
 
 
-if st.toggle('Apply Transformations on DMA/Panel Level'):
-  dma=st.selectbox('Select the Level of data ',[ col for col in media_data.columns if col.lower() in ['dma','panel', 'markets']])
-  panel_col= dma
-
-else:
-  #""" code to aggregate data on date """
+selection=[ col for col in media_data.columns if col.lower() in ['panel','Panel_2', 'markets']]
+selection.append('NA')
+dma=st.selectbox('Select the Level of data ',selection)
+panel_col= dma
 
+if dma=='NA':
+  panel_col=None
 
   dma=None