Update pages/comps_data.py

pages/comps_data.py

@@ -12,135 +12,138 @@ import matplotlib.pyplot as plt
 import seaborn as sns
 from datetime import datetime, timedelta
 
-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)
-    for index in checkbox_col:
-        df.at[index, "Checkbox"] = True
-    return df
-
-    
-data = {
-    'Address': ['Location_A', 'Location_B', 'Location_C', 'Location_D', 'Location_E',
-            'Location_F', 'Location_G', 'Location_H'
-            ],
-    'latitude': np.random.uniform(40.7, 40.8, size=8),  # Assuming latitude range between 40.7 and 40.8
-    'longitude': np.random.uniform(-74.0, -73.9, size=8),  # Assuming longitude range between -74.0 and -73.9
-    'Match score': [90, 89, 88, 87, 86, 85, 84, 83],
-    'Market': ["M1", "M1", "M1", "M1", "M1", "M1", "M1", "M1"],
-    'Sub-market': ["S1", "S1", "S1", "S1", "S1", "S1", "S1", "S1"],
-    'Lease Date': ["2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1"],
-    'LSF': [20000, 30000, 20000, 30000, 20000, 30000, 50000, 35000],
-    'RSF': [20000, 30000, 20000, 30000, 20000, 30000, 50000, 35000],
-    'Rent (NNN)': [11, 11, 11, 12, 12, 12, 12, 15],
-    'Year Built': [2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019],
-    'Office %': [20, 20, 20, 20, 20, 20, 20, 20],
-    'Clear Height':[19, 18, 19, 18, 17, 19, 19, 18],
-    'Doors (drive in / Dock)': [2, 2, 2, 2, 2, 2, 2, 2],
-    'Lease Term ': [60, 60, 60, 60, 60, 60, 60, 60],
-    'Rent (Gross)': [11, 11, 11, 12, 12, 12, 12, 15],
-    'TIs ': [1, 1, 1, 1, 1, 1,  1, 1]
-}
-
-# Create DataFrame
-df_data =  pd.DataFrame(data)
-
-
-df_data_checkbox = df_data.copy()
-df_data_checkbox.insert(loc=0, column='Select rows', value=[False]*len(df_data))
-
-
-filtered_data2 = df_data[['Address', 'Market', 'Sub-market', 'Lease Date', 'LSF', 'RSF', 'Rent (NNN)', 'Year Built', 'Office %', 'Clear Height', 'Doors (drive in / Dock)', 'Lease Term ', 'Rent (Gross)']]
-filtered_data = pd.concat([filtered_data2, filtered_data2, filtered_data2, filtered_data2])
-
-# Display the filtered data
-col_1_1, col_1_2 = st.columns([2, 1])
-with col_1_1:
-    st.write('Comps list:')
-    st.write(filtered_data)
-with col_1_2:
-    # Create a map object
-    m = folium.Map(width=500, height=440, location=(df_data['latitude'].mean(), df_data['longitude'].mean()), zoom_start=10)
-    
-    # Add markers to the map
-    all_markers = folium.FeatureGroup(name='All Markers')
-    active_markers = folium.FeatureGroup(name='Active Markers', show=False)
-    inactive_markers = folium.FeatureGroup(name='Inactive Markers', show=False)
+def main():
+    st.set_page_config(initial_sidebar_state="collapsed", layout="wide")
     
     
-    for index, row in df_data.iterrows():
-        status_color = 'green' if index%2==0 else 'red'
-        html_content = f"""
-        <div style="
-            display: inline-block;
-            background-color: white;
-            border: 2px solid black;
-            border-radius: 50%;
-            width: 20px;
-            height: 20px;
-            text-align: center;
-            line-height: 20px;
-            font-size: 8pt;
-            color: {status_color};
-        ">{index}</div>
-        """
+    def add_checkbox_column(df):
+        checkbox_col = st.multiselect("Select rows", df.index.tolist(), [])
+        df.insert(0, "Checkbox", False)
+        for index in checkbox_col:
+            df.at[index, "Checkbox"] = True
+        return df
     
-        # Create a DivIcon with custom HTML content
-        icon = folium.DivIcon(html=html_content)
-        marker = folium.Marker([row['latitude'], row['longitude']], popup=row['Address'], icon=icon).add_to(m)
         
+    data = {
+        'Address': ['Location_A', 'Location_B', 'Location_C', 'Location_D', 'Location_E',
+                'Location_F', 'Location_G', 'Location_H'
+                ],
+        'latitude': np.random.uniform(40.7, 40.8, size=8),  # Assuming latitude range between 40.7 and 40.8
+        'longitude': np.random.uniform(-74.0, -73.9, size=8),  # Assuming longitude range between -74.0 and -73.9
+        'Match score': [90, 89, 88, 87, 86, 85, 84, 83],
+        'Market': ["M1", "M1", "M1", "M1", "M1", "M1", "M1", "M1"],
+        'Sub-market': ["S1", "S1", "S1", "S1", "S1", "S1", "S1", "S1"],
+        'Lease Date': ["2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1", "2024/1/1"],
+        'LSF': [20000, 30000, 20000, 30000, 20000, 30000, 50000, 35000],
+        'RSF': [20000, 30000, 20000, 30000, 20000, 30000, 50000, 35000],
+        'Rent (NNN)': [11, 11, 11, 12, 12, 12, 12, 15],
+        'Year Built': [2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019],
+        'Office %': [20, 20, 20, 20, 20, 20, 20, 20],
+        'Clear Height':[19, 18, 19, 18, 17, 19, 19, 18],
+        'Doors (drive in / Dock)': [2, 2, 2, 2, 2, 2, 2, 2],
+        'Lease Term ': [60, 60, 60, 60, 60, 60, 60, 60],
+        'Rent (Gross)': [11, 11, 11, 12, 12, 12, 12, 15],
+        'TIs ': [1, 1, 1, 1, 1, 1,  1, 1]
+    }
     
-    # Add layer control to toggle marker visibility
-    folium.LayerControl().add_to(m)
+    # Create DataFrame
+    df_data =  pd.DataFrame(data)
     
-    polygon_coords = [ [np.random.uniform(40.7, 40.8, size=6)[i], np.random.uniform(-74.0, -73.9, size=6)[i]] for i in range(6)]
     
-    # Create a polygon object
-    polygon = folium.Polygon(locations=polygon_coords, color='blue', fill=True, fill_color='blue', fill_opacity=0.3)
+    df_data_checkbox = df_data.copy()
+    df_data_checkbox.insert(loc=0, column='Select rows', value=[False]*len(df_data))
     
-    # Add the tag to the polygon 
-    popup_text = "This is my polygon"
-    popup = folium.Popup(popup_text, parse_html=True)
-    polygon.add_child(popup)
     
-    # Add the polygon to the map
-    m.add_child(polygon)
+    filtered_data2 = df_data[['Address', 'Market', 'Sub-market', 'Lease Date', 'LSF', 'RSF', 'Rent (NNN)', 'Year Built', 'Office %', 'Clear Height', 'Doors (drive in / Dock)', 'Lease Term ', 'Rent (Gross)']]
+    filtered_data = pd.concat([filtered_data2, filtered_data2, filtered_data2, filtered_data2])
     
+    # Display the filtered data
+    col_1_1, col_1_2 = st.columns([2, 1])
+    with col_1_1:
+        st.write('Comps list:')
+        st.write(filtered_data)
+    with col_1_2:
+        # Create a map object
+        m = folium.Map(width=500, height=440, location=(df_data['latitude'].mean(), df_data['longitude'].mean()), zoom_start=10)
         
-    # Render the map
-    folium_static(m)
-col_2_1, col_2_2, col_2_3  = st.columns([1, 1, 1])
-with col_2_1:
-    option = st.radio("Add comps:", (":rainbow[On]", ":rainbow[Off]"), horizontal=True,  index=1)
-         
-with col_2_2:
-    if st.button("Comps"):
-        st.switch_page("pages/comps_data.py")
-
-# Add the second button on the right side
-with col_2_3:
-    st.button("Right Button")
-# if option == ":rainbow[On]":
-
-col_3_1, col_3_2  = st.columns(2) 
-with col_3_1: 
-    # c1 = st.container(height=120)
-    # df_data_checkbox = add_checkbox_column(df_data)
-
-    st.data_editor(
-        df_data_checkbox,
-        column_config={
-            "Select rows": st.column_config.CheckboxColumn(required=True),
-        },
-        disabled=["widgets"],
-        hide_index=True,
-    )
-
-with col_3_2: 
-    c2 = st.container(height=120)
-    c2.write('Comps list:')
-    c2.write(filtered_data) 
+        # Add markers to the map
+        all_markers = folium.FeatureGroup(name='All Markers')
+        active_markers = folium.FeatureGroup(name='Active Markers', show=False)
+        inactive_markers = folium.FeatureGroup(name='Inactive Markers', show=False)
+        
+        
+        for index, row in df_data.iterrows():
+            status_color = 'green' if index%2==0 else 'red'
+            html_content = f"""
+            <div style="
+                display: inline-block;
+                background-color: white;
+                border: 2px solid black;
+                border-radius: 50%;
+                width: 20px;
+                height: 20px;
+                text-align: center;
+                line-height: 20px;
+                font-size: 8pt;
+                color: {status_color};
+            ">{index}</div>
+            """
+        
+            # Create a DivIcon with custom HTML content
+            icon = folium.DivIcon(html=html_content)
+            marker = folium.Marker([row['latitude'], row['longitude']], popup=row['Address'], icon=icon).add_to(m)
+            
+        
+        # Add layer control to toggle marker visibility
+        folium.LayerControl().add_to(m)
+        
+        polygon_coords = [ [np.random.uniform(40.7, 40.8, size=6)[i], np.random.uniform(-74.0, -73.9, size=6)[i]] for i in range(6)]
+        
+        # Create a polygon object
+        polygon = folium.Polygon(locations=polygon_coords, color='blue', fill=True, fill_color='blue', fill_opacity=0.3)
+        
+        # Add the tag to the polygon 
+        popup_text = "This is my polygon"
+        popup = folium.Popup(popup_text, parse_html=True)
+        polygon.add_child(popup)
+        
+        # Add the polygon to the map
+        m.add_child(polygon)
+        
+            
+        # Render the map
+        folium_static(m)
+    col_2_1, col_2_2, col_2_3  = st.columns([1, 1, 1])
+    with col_2_1:
+        option = st.radio("Add comps:", (":rainbow[On]", ":rainbow[Off]"), horizontal=True,  index=1)
+             
+    with col_2_2:
+        if st.button("Comps"):
+            st.switch_page("pages/comps_data.py")
+    
+    # Add the second button on the right side
+    with col_2_3:
+        st.button("Right Button")
+    # if option == ":rainbow[On]":
+    
+    col_3_1, col_3_2  = st.columns(2) 
+    with col_3_1: 
+        # c1 = st.container(height=120)
+        # df_data_checkbox = add_checkbox_column(df_data)
+    
+        st.data_editor(
+            df_data_checkbox,
+            column_config={
+                "Select rows": st.column_config.CheckboxColumn(required=True),
+            },
+            disabled=["widgets"],
+            hide_index=True,
+        )
+    
+    with col_3_2: 
+        c2 = st.container(height=120)
+        c2.write('Comps list:')
+        c2.write(filtered_data) 
 
+if __name__ == "__main__":
+    main()