Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 import numpy as np
 from scipy.spatial.distance import hamming
 
-# Загрузка данных из Excel файлов
+# Загррузка данных из Excel файлов
 vacancy = 'vacancy_new_new.xlsx'
 resume = 'resume_new_new.xlsx'
 file_nkz = 'label_NKZ.xlsx'
@@ -37,7 +37,24 @@ selected_skills = st.multiselect("Выберите навыки", skills_options
 # Выбор файла для поиска совпадений
 file_option = st.selectbox("Выберите тип", ("Вакансия", "Резюме"))
 
-# Кнопки для добавления новой записи и запуска поиска совпадений
+# Выбор файла для записи новой записи
+write_option = "vacancy" if file_option == "Вакансия" else "resume"
+
+# Кнопка для добавления новой записи
+add_entry = st.button("Добавить новую запись")
+if add_entry:
+    if new_name and selected_skills:
+        skills_ids = df_skills[df_skills['skills'].isin(selected_skills)]['id'].values
+        skills_ids_str = ','.join(map(str, skills_ids))
+        new_entry = pd.DataFrame({'name': [new_name], 'NKZ': [selected_nkz], 'skills': [' '.join(selected_skills)], 'arr': [skills_ids_str]})
+        df_to_update = df_vacancy if write_option == "vacancy" else df_resume
+        df_to_update = pd.concat([df_to_update, new_entry], ignore_index=True)
+        df_to_update.to_excel(vacancy if write_option == "vacancy" else resume, index=False)
+        st.success("Новая запись была успешно добавлена.")
+    else:
+        st.error("Пожалуйста, введите имя и выберите навыки для добавления записи.")
+
+# Кнопка для поиска совпадений
 if st.button("Найти совпадения"):
     if new_name and selected_skills:
         # Определение DataFrame для поиска совпадений