Update app.py

app.py

@@ -160,6 +160,8 @@ with st.sidebar:
     dorm_range = (int(data_tipo['Dorm'].min()), int(data_tipo['Dorm'].max()))
     banho_range = (int(data_tipo['Banheiro'].min()), int(data_tipo['Banheiro'].max()))
     vaga_range = (int(data_tipo['Vaga'].min()), int(data_tipo['Vaga'].max()))
+    test_range = (int(data_tipo['Test'].min()), int(data_tipo['Test'].max()))
+    
 
     # Add sliders to filter data based
     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 if data_tipo['Atotal'].min() != data_tipo['Atotal'].max() else 0.1)
@@ -167,12 +169,14 @@ with st.sidebar:
 
     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 data_tipo['Dorm'].min() != data_tipo['Dorm'].max() else 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 data_tipo['Banheiro'].min() != data_tipo['Banheiro'].max() else 1)
+    if int(data_tipo['Banh'].min()) != 0 and int(data_tipo['Banh'].max()) != 0:
+        banho_range = st.slider('Banheiros', int(data_tipo['Banh'].min()), int(data_tipo['Banh'].max()), (int(data_tipo['Banh'].min()), int(data_tipo['Banh'].max())), step=1 if data_tipo['Banheiro'].min() != data_tipo['Banheiro'].max() else 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 if data_tipo['Vaga'].min() != data_tipo['Vaga'].max() else 1)
+    if int(data_tipo['Test'].min()) != 0 and int(data_tipo['Test'].max()) != 0:
+        test_range = st.slider('Testada', int(data_tipo['Test'].min()), int(data_tipo['Test'].max()), (int(data_tipo['Test'].min()), int(data_tipo['Test'].max())), step=1 if data_tipo['Test'].min() != data_tipo['Test'].max() else 1)
     
-    # Initialize checkbox variables
+        # Initialize checkbox variables
     elev_checkbox = False
     churr_checkbox = False
     esq_checkbox = False
@@ -180,23 +184,20 @@ with st.sidebar:
         # Add checkboxes for dummy features
     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_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['Banh'].between(banho_range[0], banho_range[1])) &
+            (data_tipo['Vaga'].between(vaga_range[0], vaga_range[1])) &
+            (data_tipo['Test'].between(test_range[0], test_range[1])) &
             (data_tipo['Elevador'] == elev_value) &
-            (data_tipo['Churrasq'] == churr_value) &
             (data_tipo['Lot_pos'] == esq_value)]