Update app.py

app.py

@@ -16,37 +16,43 @@ with open('dop_table.json', 'r') as json_file:
     data_dop = json.load(json_file)
 df2= pd.DataFrame(data_dop)
 
+
+df1['Навыки'] = [i.lower().replace('.', '').replace('-', '') for i in df1['Навыки']]
+df2['Навыки'] = [i.lower().replace('.', '').replace('-', '')  for i in df2['Навыки']]
+
 dohod = 0
 i = 0
 
 with st.sidebar:
     sphere = st.selectbox(
-        "Выберите сферу:", options = list(df1['Сфера'].unique()), key=1)
+        "Выбери сферу:", options = list(df1['Сфера'].unique()), key=1)
     st.write("Выбрано:", sphere)
 
     job = st.selectbox(
-        "Выберите профессию:", options=list(df1['Профессия'].unique()), key=2)
+        "Выбери сферу:", options=list(df1['Профессия'].unique()), key=2)
     st.write("Выбрано:", job)
 
     # st.write(df.columns)
 
-    dohod = st.number_input(
-        "Введите уровень дохода:",
+    age = st.number_input(
+        "Введите возраст",
         min_value=0,  # Минимальное значение
         step=1,  # Шаг единицы
-        format='%d',  # Формат для целых чисел
-        # max_value=(df1['ЗП в вакансии'].max() - 50000)
+        format='%d'  # Формат для целых чисел
         # ("IT", "Продажи")
     )
-    st.write("Выбрано:", dohod)
 
-    age = st.number_input(
-        "Введите возраст:",
+    dohod = st.number_input(
+        "Введите уровень дохода",
         min_value=0,  # Минимальное значение
         step=1,  # Шаг единицы
-        format='%d'  # Формат для целых чисел
+        format='%d',  # Формат для целых чисел
+        # max_value=(df1['ЗП в вакансии'].max() - 50000)
         # ("IT", "Продажи")
     )
+    st.write("Выбрано:", dohod)
+
+
     st.write("Введено:", age)
 
     # df2 = df[df['ЗП в вакансии'] >= dohod]
@@ -67,38 +73,44 @@ with st.sidebar:
     st.write("Выбрано:", gotov_uchitsia)
     st.header('Выберите навыки, которые у Вас есть')
 
-    python = st.checkbox("Python", key=f"{i+1}mlr")
-    sql = st.checkbox("SQL", key=f"{i+1}mlrq")
-    html = st.checkbox("HTML", key=f"{i+1}mlrw")
-    java = st.checkbox("Java", key=f"{i+1}mlre")
-    figma = st.checkbox("Figma", key=f"{i+1}mlrr")
-    power_bi = st.checkbox("Power BI", key=f"{i+1}mlrt")
-    prodaji = st.checkbox("Продажи", key=f"{i+1}mlry")
-    analys_prodaj = st.checkbox("Анализ продаж", key=f"{i+1}mlru")
-    work_with_clients = st.checkbox("Работа с клиентами", key=f"{i+1}mlri")
-    riteil = st.checkbox("Ритейл", key=f"{i+1}mlrl")
-    i+=1
-    but = st.button('Данные введены в полном объеме', key=f"{i+1}mlurl")
-    list_skills=[]
-    if but:
-        if python:
-            list_skills.append('Python')
-        if sql:
-            list_skills.append('SQL')
-        if html:
-            list_skills.append('HTML')
-        if java:
-            list_skills.append('Java')
-        if figma:
-            list_skills.append('Figma')
-        if power_bi:
-            list_skills.append('Power BI')
-        if prodaji:
-            list_skills.append('Продажи')
-        if work_with_clients:
-            list_skills.append('Работа с клиентами')
-        if riteil:
-            list_skills.append('ритейл')
+    list_skills = []
+    for i in sorted(set(df1['Навыки'])):
+        skill = st.checkbox(i, key=f"{i}")
+        if skill:
+            list_skills.append(i)
+
+    # python = st.checkbox("Python", key=f"{i+1}mlr")
+    # sql = st.checkbox("SQL", key=f"{i+1}mlrq")
+    # html = st.checkbox("HTML", key=f"{i+1}mlrw")
+    # java = st.checkbox("Java", key=f"{i+1}mlre")
+    # figma = st.checkbox("Figma", key=f"{i+1}mlrr")
+    # power_bi = st.checkbox("Power BI", key=f"{i+1}mlrt")
+    # prodaji = st.checkbox("Продажи", key=f"{i+1}mlry")
+    # analys_prodaj = st.checkbox("Анализ продаж", key=f"{i+1}mlru")
+    # work_with_clients = st.checkbox("Работа с клиентами", key=f"{i+1}mlri")
+    # riteil = st.checkbox("Ритейл", key=f"{i+1}mlrl")
+    # i+=1
+    # but = st.button('Данные введены в полном объеме', key=f"{i+1}mlurl")
+    # list_skills=[]
+    # if but:
+    #     if python:
+    #         list_skills.append('Python')
+    #     if sql:
+    #         list_skills.append('SQL')
+    #     if html:
+    #         list_skills.append('HTML')
+    #     if java:
+    #         list_skills.append('Java')
+    #     if figma:
+    #         list_skills.append('Figma')
+    #     if power_bi:
+    #         list_skills.append('Power BI')
+    #     if prodaji:
+    #         list_skills.append('Продажи')
+    #     if work_with_clients:
+    #         list_skills.append('Ра��ота с клиентами')
+    #     if riteil:
+    #         list_skills.append('ритейл')
 
 
 def table(sphere: str, list_with_skills: list, dohod, sphere_change, gotov_uchitsia):
@@ -138,8 +150,9 @@ def table(sphere: str, list_with_skills: list, dohod, sphere_change, gotov_uchit
 df_fin = table(sphere, list_skills, dohod, sphere_change, gotov_uchitsia)
 
 st.header('Рост Вашего дохода')
-labels = ['Доход сейчас', 'Доход через год', 'Доход через 2 года']
-values = [dohod, dohod*1.1, dohod*1.2]
+
+labels = [age, age+1, age+2, age+3, age+4, age+5, age+6, age+7, age+8, age+9]
+values = [dohod, dohod*1.05, dohod*1.1, dohod*1.11, dohod*1.12, dohod*1.13, dohod*1.14, dohod*1.14, dohod*1.14, dohod*1.14,]
 fig, ax = plt.subplots()
 
 # fig.set_size_inches(10, 5)
@@ -153,6 +166,7 @@ ax.bar(labels, values)
 fig, ax = plt.subplots(figsize=(10, 5))  # Размер графика 30 на 30 дюймов
 ax.plot(labels, values, marker='o', linewidth=2)  # Линейный график с маркерами
 ax.set_ylabel('Сумма, руб')
+ax.set_xlabel('Ваш возраст, лет')
 ax.set_title('Сравнение доходов')
 ax.grid(True)  # Добавление сетки
 
@@ -162,7 +176,7 @@ st.pyplot(fig)
 if gotov_uchitsia == "Нет":
     count_course = 0
 else:
-    count_course = 5
+    count_course = 3
 
 if sphere_change == "Да":
     mask_sphere = df_fin['Сфера'] != 0
@@ -178,3 +192,4 @@ df_fin_sort = df_fin.sort_values(by=['ЗП в вакансии'], ascending=[Fal
 
 st.header('Вакансии, которые вам будут интересны')
 st.dataframe(df_fin_sort, width=1100, height=500)
+