Update app.py

app.py

@@ -136,7 +136,7 @@ with st.sidebar:
     # Add a dropdown for filtering "Tipo"
     selected_tipo = st.selectbox('Filtrar por Tipo', data['Tipo'].unique())
 
-    data = data[data['Tipo'] == selected_tipo]
+    data_tipo = data[data['Tipo'] == selected_tipo]
     
     selected_coords = st.selectbox('Selecione o tipo de pesquisa', ['Ampla', 'Direcionada'])
 
@@ -164,8 +164,8 @@ with st.sidebar:
     if selected_coords == 'Direcionada':
         zoom_level = st.slider('Nível de zoom', min_value=1, max_value=15, value=zoom_level)
     else:
-        for index, row in data.iterrows():
-            distance_all = calculate_distance(row['latitude'], row['longitude'], data['latitude'].mean(), data['longitude'].mean())
+        for index, row in data_tipo.iterrows():
+            distance_all = calculate_distance(row['latitude'], row['longitude'], data_tipo['latitude'].mean(), data_tipo['longitude'].mean())
             if distance_all > max_distance_all:
                 max_distance_all = distance_all
 
@@ -180,31 +180,36 @@ with st.sidebar:
         radius_in_meters = st.slider('Selecione raio (em metros)', min_value=100, max_value=5000, value=1000)
 
     # Add sliders to filter data based
-    atotal_range = st.slider('Área Total', float(data['Atotal'].min()), float(data['Atotal'].max()), (float(data['Atotal'].min()), float(data['Atotal'].max())), step=.1)
-    apriv_range = st.slider('Área Privativa', float(data['Apriv'].min()), float(data['Apriv'].max()), (float(data['Apriv'].min()), float(data['Apriv'].max())), step=.1)
-    dorm_range = st.slider('Dormitórios', int(data['Dorm'].min()), int(data['Dorm'].max()), (int(data['Dorm'].min()), int(data['Dorm'].max())), step=1)
-    banho_range = st.slider('Banheiros', int(data['Banheiro'].min()), int(data['Banheiro'].max()), (int(data['Banheiro'].min()), int(data['Banheiro'].max())), step=1)
-    vaga_range = st.slider('Vaga de estacionamento', int(data['Vaga'].min()), int(data['Vaga'].max()), (int(data['Vaga'].min()), int(data['Vaga'].max())), step=1)
+    atotal_range = st.slider('Área Total', float(data_tipo['Atotal'].min()), float(data_tipo['Atotal'].max()), (float(data_tipo['Atotal'].min()), float(data_tipo['Atotal'].max())), step=.1)
+    apriv_range = st.slider('Área Privativa', float(data_tipo['Apriv'].min()), float(data_tipo['Apriv'].max()), (float(data_tipo['Apriv'].min()), float(data_tipo['Apriv'].max())), step=.1)
+    if int(data_tipo['Dorm'].min()) != 0 and int(data_tipo['Dorm'].max()) != 0:
+        dorm_range = st.slider('Dormitórios', int(data_tipo['Dorm'].min()), int(data_tipo['Dorm'].max()), (int(data_tipo['Dorm'].min()), int(data_tipo['Dorm'].max())), step=1)
+    if int(data_tipo['Banheiro'].min()) != 0 and int(data_tipo['Banheiro'].max()) != 0:
+        banho_range = st.slider('Banheiros', int(data_tipo['Banheiro'].min()), int(data_tipo['Banheiro'].max()), (int(data_tipo['Banheiro'].min()), int(data_tipo['Banheiro'].max())), step=1)
+    if int(data_tipo['Vaga'].min()) != 0 and int(data_tipo['Vaga'].max()) != 0:
+        vaga_range = st.slider('Vaga de estacionamento', int(data_tipo['Vaga'].min()), int(data_tipo['Vaga'].max()), (int(data_tipo['Vaga'].min()), int(data_tipo['Vaga'].max())), step=1)
 
     # Add checkboxes for dummy features
-    elev_checkbox = st.checkbox('Elevador')
-    churr_checkbox = st.checkbox('Churrasqueira')
-    esq_checkbox = st.checkbox('Duas ou mais frentes')
+    if int(data_tipo['Elevador'].min()) != 0 and int(data_tipo['Elevador'].max()) != 0:
+        elev_checkbox = st.checkbox('Elevador')
+    if int(data_tipo['Churrasq'].min()) != 0 and int(data_tipo['Churrasq'].max()) != 0:
+        churr_checkbox = st.checkbox('Churrasqueira')
+    if int(data_tipo['Lot_pos'].min()) != 0 and int(data_tipo['Lot_pos'].max()) != 0:
+        esq_checkbox = st.checkbox('Duas ou mais frentes')
 
     # Transform checkbox values into 1s and 0s
     elev_value = 1 if elev_checkbox else 0
     churr_value = 1 if churr_checkbox else 0
     esq_value = 1 if esq_checkbox else 0
 
-    data = data[(data['Tipo'] == selected_tipo) &
-            (data['Atotal'].between(atotal_range[0], atotal_range[1])) &
-            (data['Apriv'].between(apriv_range[0], apriv_range[1])) &
-            (data['Dorm'].between(dorm_range[0], dorm_range[1])) &
-            (data['Banheiro'].between(banho_range[0], banho_range[1])) &
-            (data['Vaga'].between(vaga_range[0], vaga_range[1])) &  
-            (data['Elevador'] == elev_value) &
-            (data['Churrasq'] == churr_value) &
-            (data['Lot_pos'] == esq_value)]
+    data_tipo = data_tipo[(data_tipo['Atotal'].between(atotal_range[0], atotal_range[1])) &
+            (data_tipo['Apriv'].between(apriv_range[0], apriv_range[1])) &
+            (data_tipo['Dorm'].between(dorm_range[0], dorm_range[1])) &
+            (data_tipo['Banheiro'].between(banho_range[0], banho_range[1])) &
+            (data_tipo['Vaga'].between(vaga_range[0], vaga_range[1])) &  
+            (data_tipo['Elevador'] == elev_value) &
+            (data_tipo['Churrasq'] == churr_value) &
+            (data_tipo['Lot_pos'] == esq_value)]
 
 
     # Links to other apps at the bottom of the sidebar
@@ -214,7 +219,7 @@ with st.sidebar:
 
 # Filter data based on the radius
 if selected_coords == 'Direcionada':
-    filtered_data = data[data.apply(lambda x: calculate_distance(x['latitude'], x['longitude'], custom_lat, custom_lon), axis=1) <= radius_in_meters]
+    filtered_data = data_tipo[data_tipo.apply(lambda x: calculate_distance(x['latitude'], x['longitude'], custom_lat, custom_lon), axis=1) <= radius_in_meters]
     filtered_data = filtered_data.dropna()  # Drop rows with NaN values
 
 # Add a custom CSS class to the map container