Update app.py

добавление вакансий
фикс ошибки с доходом

app.py

@@ -21,19 +21,30 @@ i = 0
 
 with st.sidebar:
     sphere = st.selectbox(
-        "Выбери сферу:", options = list(df1['Сфера'].unique()), key=i)
-    # st.write("Выбрано:", sphere)
+        "Выбери сферу:", options = list(df1['Сфера'].unique()), key=1)
+    st.write("Выбрано:", sphere)
+
+    job = st.selectbox(
+        "Выбери сферу:", options=list(df1['Профессия'].unique()), key=2)
+    st.write("Выбрано:", job)
 
     # st.write(df.columns)
 
     dohod = st.number_input(
-        "Введите уровень дохода", key=f"number_input_{i}"
+        "Введите уровень дохода",
+        min_value=0,  # Минимальное значение
+        step=1,  # Шаг единицы
+        format='%d',  # Формат для целых чисел
+        max_value=(df1['ЗП в вакансии'].max() - 50000)
         # ("IT", "Продажи")
     )
     st.write("Выбрано:", dohod)
 
     age = st.number_input(
-        "Введите возраст"
+        "Введите возраст",
+        min_value=0,  # Минимальное значение
+        step=1,  # Шаг единицы
+        format='%d'  # Формат для целых чисел
         # ("IT", "Продажи")
     )
     st.write("Введено:", age)
@@ -90,13 +101,17 @@ with st.sidebar:
             list_skills.append('ритейл')
 
 
-def table(sphere: str, list_with_skills: list, dohod: int, sphere_change, gotov_uchitsia):
+def table(sphere: str, list_with_skills: list, dohod, sphere_change, gotov_uchitsia):
     new_df = pd.merge(df1, df2, on='Навыки', how='outer')
     if sphere_change == "Да":
         mask_sphere = new_df['Сфера'] != 0
     else:
         mask_sphere = new_df['Сфера'] == sphere
-    new_df2 = new_df[new_df['ЗП в вакансии'] >= dohod][mask_sphere]
+    try:
+        new_df2 = new_df[new_df['ЗП в вакансии'] >= dohod][mask_sphere]
+    except:
+        new_df2 = new_df[mask_sphere][new_df['ЗП в вакансии'] >= (dohod - 50000)]
+
     new_df2['Есть_в_списке'] = new_df2['Навыки'].apply(lambda x: 1 if x in list_with_skills else 0)
     result = new_df2.groupby(['Профессия', 'Сфера', 'ЗП в вакансии']).agg(
         Skill=('Навыки', set),
@@ -109,22 +124,16 @@ def table(sphere: str, list_with_skills: list, dohod: int, sphere_change, gotov_
     def calculate_difference(row):
         cell_set = set(row['Skill'])  # Преобразуем список в набор
         test_set = set(list_with_skills)  # Преобразуем list_test в набор
-        return list(cell_set - test_set)
+        return (cell_set - test_set)
+
 
     result['Недостающие навыки'] = result.apply(calculate_difference, axis=1)
     result['Требуемые навыки для позиции'] = result['Skill']
     result['Наименования доступных курсов'] = result['Cours']
     result['Ссылки на курс'] = result['Links']
 
-    if gotov_uchitsia == "Нет":
-        mask_ucheba = (result['On_list'] != {0}) &  (result['On_list'] != {0, 1})
-    else:
-        mask_ucheba = result['On_list'] != {5}
 
-    # try:
-    result = result[mask_ucheba]
-    # except:
-    #     result =result
+
 
     result2 = result[['Сфера', 'Профессия', 'ЗП в вакансии', 'Требуемые навыки для позиции', 'Недостающие навыки',
                       'Наименования доступных курсов',
@@ -153,12 +162,19 @@ ax.set_ylabel('Сумма, руб')
 ax.set_title('Сравнение доходов')
 ax.grid(True)  # Добавление сетки
 
+count_course = 3
 # Отображение графика
 st.pyplot(fig)
+if gotov_uchitsia == "Нет":
+    count_course = 0
+else:
+    count_course = 3
 
-
+mask_len = df_fin['Недостающие навыки'].apply(lambda x: len(x) <= count_course)
+df_fin = df_fin[mask_len]
 df_fin_sort = df_fin.sort_values(by=['ЗП в вакансии'], ascending=[False]).reset_index(drop=True)
 
 st.header('Вакансии, которые вам будут интересны')
 st.dataframe(df_fin_sort, width=1100, height=500)
 
+