Update app.py (#11)

- Update app.py (876f50aca38079cd8de848ac48e72005e8960832)


Co-authored-by: Polina Shishenkova <PolinaShishenkova@users.noreply.huggingface.co>

app.py

@@ -83,6 +83,8 @@ def main():
     # Выбор таблицы для отображения с помощью selectbox
     selected_table = st.selectbox("Выберите таблицу", list(tables.keys()))
     if selected_table == 'Комментарии':
+        filtered_data = tables[selected_table]
+        filtered_data['Age'] = pd.to_numeric(filtered_data['Age'], errors='coerce')
         # post_filter_value = st.text_input("Введите ID постов (разделите их запятой)", key='post_ids_input')
         # post_ids = [int(post_id.strip()) for post_id in post_ids_input.split(', ') if post_id.strip()] if post_ids_input else []
 
@@ -93,6 +95,10 @@ def main():
         country_filter = st.sidebar.multiselect("Фильтр по стране", tables[selected_table]['Country'].unique(), default=[])
         city_filter = st.sidebar.multiselect("Фильтр по городу", tables[selected_table]['City'].unique(), default=[])
         sex_filter = st.sidebar.multiselect("Фильтр по полу", tables[selected_table]['Sex'].unique(), default=[])
+        st.sidebar.subheader("Возраст пользователя")
+        check_age = st.sidebar.checkbox("Указать возраст пользователей", key="disabled")
+        if check_age:
+            age_limits = st.sidebar.slider("Выберите диапазон возраста пользователей", 0, 100, (int(np.min(filtered_data['Age'])), 70))
         # post_filter_value = st.text_area("Поиск всех комментариев к определенным постам (столбец 'Пост')")
        
         # date_type = st.radio("Выберите тип фильтрации по дате", ["Диапазон дат", "Конкретная дата", 'Все'], index=2)
@@ -118,7 +124,8 @@ def main():
             filtered_data = filtered_data[filtered_data['Country'].isin(country_filter)]
         if sex_filter:
             filtered_data = filtered_data[filtered_data['Sex'].isin(sex_filter)]
-        
+        if check_age:
+            filtered_data = filtered_data[(filtered_data['Age'] >= age_limits[0]) & (filtered_data['Age'] <= age_limits[1])]
         # post_filter_value = st.number_input("Введите ID постов (столбец 'Пост') (разделите их запятой)",min_value=0)
         # if post_filter_value:
         #     filtered_data = filtered_data[filtered_data['Пост'] == post_filter_value]
@@ -219,7 +226,9 @@ def main():
         country_filter = st.sidebar.multiselect("Фильтр по стране", tables[selected_table]['Country'].unique(), default=[],key = 'user_country')
         city_filter = st.sidebar.multiselect("Фильтр по городу", tables[selected_table]['City'].unique(), default=[],key = 'user_city')
         sex_filter = st.sidebar.multiselect("Фильтр по полу", tables[selected_table]['Sex'].unique(), default=[],key = 'user_sex')
-        age_limits = st.sidebar.slider("Выберите диапазон возраста пользователей", 0, 100, (int(np.min(filtered_data['Age'])), 70))
+        check_age = st.sidebar.checkbox("Указать возраст пользователей", key="disabled")
+        if check_age:
+            age_limits = st.sidebar.slider("Выберите диапазон возраста пользователей", 0, 100, (int(np.min(filtered_data['Age'])), 70))
         if st.button('🗺️ Показать карту географии подписчиков'):
             st.write('Карта загружается...')
             cities = tables[selected_table]['City'].unique()
@@ -256,7 +265,7 @@ def main():
             filtered_data = filtered_data[filtered_data['Country'].isin(country_filter)]
         if sex_filter:
             filtered_data = filtered_data[filtered_data['Sex'].isin(sex_filter)]
-        if age_limits:
+        if check_age:
             filtered_data = filtered_data[(filtered_data['Age'] >= age_limits[0]) & (filtered_data['Age'] <= age_limits[1])]
         filtered_data = filtered_data.reset_index(drop=True)