Update app.py

app.py

@@ -180,14 +180,18 @@ df_fin = table(sphere, list_skills, dohod, sphere_change, gotov_uchitsia)
 # """Внедрение модели"""
 if sphere == 'ИТ':
     job_sphere = 'IT'
+    change_sphere = 'Sales'
 else:
     job_sphere = 'Sales'
+    change_sphere = 'IT'
 
 st.header('Рост Вашего дохода')
 labels = [age, age+3]
 dohod_2 = predictor.predict(dohod, float(age), str(job_sphere))
+dohod_change = predictor.predict(dohod, float(age), str(change_sphere))
  # 200_000.0, 25.0, "IT"
 values = [dohod, dohod_2]
+values_change = [dohod, dohod_change]
 st.write(f"Ваш доход через 3 года составит: {round(dohod_2)} руб.")
 
 
@@ -212,7 +216,8 @@ ax.bar(labels, values)
 
 # Построение линейного графика
 fig, ax = plt.subplots(figsize=(10, 5))  # Размер графика 30 на 30 дюймов
-ax.plot(labels, values, marker='o', linewidth=2)  # Линейный график с маркерами
+ax.plot(labels, values, marker='o', linewidth=2, color='blue', label=f'Прогноз дохода в текущей сфере')  # Линейный график с маркерами
+ax.plot(labels, values_change, marker='o', linewidth=2, color='red', linestyle='--', label=f'Прогноз дохода при смене сферы на {change_sphere}')  # Линейный график с маркерами
 ax.set_ylabel('Сумма, руб')
 ax.set_xlabel('Ваш возраст, лет')
 ax.set_title('Сравнение доходов')