Update pages/comps_data.py

pages/comps_data.py

@@ -8,6 +8,9 @@ import seaborn as sns
 from datetime import datetime, timedelta
 
 def main():
+    st.set_page_config(initial_sidebar_state="collapsed", layout="wide")
+
+    
     data = {
         "favorite": [True, False, False, True, True, False, False, True],
         'Address': ['B', 'B', 'B', 'B', 'B',
@@ -33,11 +36,8 @@ def main():
     
     # Create DataFrame
     df5 =  pd.DataFrame(data)
-    df5.insert(loc=0, column='Select rows', value=[False]*len(df5))
+    # df5.insert(loc=0, column='Select rows', value=[False]*len(df5))
 
-    st.set_page_config(initial_sidebar_state="collapsed", layout="wide")
-    
-    
     def add_checkbox_column(df):
         checkbox_col = st.multiselect("Select rows", df.index.tolist(), [])
         df.insert(0, "Checkbox", False)
@@ -136,11 +136,13 @@ def main():
             
         # Render the map
         folium_static(m)
-    col_2_1, col_2_2  = st.columns([8, 1])
+    col_2_1, col_2_2, col_2_3  = st.columns([4, 4, 1])
     with col_2_1:
         option = st.radio("Add comps:", (":rainbow[On]", ":rainbow[Off]"), horizontal=True,  index=1)
              
     with col_2_2:
+        st.button("Predictions")
+    with col_2_3:
         if st.button("Rexy"):
             st.switch_page("pages/comps_data.py")