Yerzhxn commited on
Commit
561c472
·
verified ·
1 Parent(s): f2959e1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +103 -0
app.py ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import streamlit as st
3
+
4
+ # Загрузка данных из четырех Excel файлов
5
+ vacancy = 'vacancy_new_new.xlsx'
6
+ resume = 'resume_new_new.xlsx'
7
+ file_nkz = 'label_NKZ.xlsx'
8
+ file_skills = 'label_skills.xlsx'
9
+
10
+ df_vacancy = pd.read_excel(vacancy)
11
+ df_resume = pd.read_excel(resume)
12
+ df_nkz = pd.read_excel(file_nkz)
13
+ df_skills = pd.read_excel(file_skills)
14
+
15
+ # Проверка, что в файлах есть нужные колонки
16
+ if 'skills' not in df_vacancy.columns or 'skills' not in df_resume.columns or 'name' not in df_vacancy.columns or 'name' not in df_resume.columns or 'NKZ' not in df_nkz.columns or 'skills' not in df_skills.columns:
17
+ raise ValueError("Файлы должны содержать колонки 'skills', 'name' и 'NKZ'")
18
+
19
+ # Streamlit интерфейс
20
+ st.title("Поиск совпадений по навыкам")
21
+
22
+ # Ввод текста для добавления в колонку 'name'
23
+ new_name = st.text_input("Введите название")
24
+
25
+ # Выбор из колонки 'NKZ' из файла label_NKZ
26
+ nkz_options = df_nkz['name'].unique().tolist()
27
+ selected_nkz = st.selectbox("Выберите профессию по НКЗ", nkz_options)
28
+
29
+ # Множественный выбор из колонки 'skills' из файла label_skills
30
+ skills_options = sorted(df_skills[df_skills['NKZ'] == df_nkz[df_nkz['name'] == selected_nkz]['NKZ'].values[0]]['skills'].unique().tolist())
31
+
32
+ selected_skills = st.multiselect("Выберите навыки", skills_options)
33
+
34
+ # Выбор файла для поиска совпадений
35
+ file_option = st.selectbox("Выберите тип", ("Вакансия", "Резюме"))
36
+
37
+ # Выбор файла для записи новой записи
38
+ write_option = "resume" if file_option == "Вакансия" else "vacancy"
39
+
40
+ # Кнопки для добавления новой записи и запуска поиска совпадений
41
+ add_entry = st.button("Добавить новую запись")
42
+ if add_entry:
43
+ if new_name and selected_skills:
44
+ new_entry = pd.DataFrame({'name': [new_name], 'NKZ': [selected_nkz], 'skills': [' '.join(selected_skills)]})
45
+ if write_option == "vacancy":
46
+ df_vacancy = pd.concat([df_vacancy, new_entry], ignore_index=True)
47
+ df_vacancy.to_excel(vacancy, index=False)
48
+ else:
49
+ df_resume = pd.concat([df_resume, new_entry], ignore_index=True)
50
+ df_resume.to_excel(resume, index=False)
51
+ st.success("Новая запись была успешно добавлена.")
52
+ else:
53
+ st.error("Пожалуйста, введите имя и выберите навыки для добавления записи.")
54
+
55
+ if st.button("Найти совпадения"):
56
+ if new_name and selected_skills:
57
+ # Добавление новой записи в противоположный файл
58
+ new_entry = pd.DataFrame({'name': [new_name], 'NKZ': [selected_nkz], 'skills': [' '.join(selected_skills)]})
59
+ if write_option == "vacancy":
60
+ df_vacancy = pd.concat([df_vacancy, new_entry], ignore_index=True)
61
+ df_vacancy.to_excel(vacancy, index=False)
62
+ else:
63
+ df_resume = pd.concat([df_resume, new_entry], ignore_index=True)
64
+ df_resume.to_excel(resume, index=False)
65
+
66
+ # Определение DataFrame для поиска совпадений
67
+ if file_option == "Вакансия":
68
+ skills_to_compare = df_vacancy['skills'].tolist()
69
+ names_to_compare = df_vacancy['name'].tolist()
70
+ else:
71
+ skills_to_compare = df_resume['skills'].tolist()
72
+ names_to_compare = df_resume['name'].tolist()
73
+
74
+ # Поиск совпадений по навыкам
75
+ matching_results = []
76
+ for i, skills_text in enumerate(skills_to_compare):
77
+ match_count = sum(skill in skills_text for skill in selected_skills)
78
+ if match_count >= len(selected_skills) / 2:
79
+ matching_results.append({
80
+ 'Name': names_to_compare[i],
81
+ 'Matching Skills': [skill for skill in selected_skills if skill in skills_text],
82
+ 'Match Count': match_count
83
+ })
84
+
85
+ # Вывод результатов, где совпадение как минимум половины выбранных навыков
86
+ st.write("Результаты с совпадением как минимум половины выбранных навыков:")
87
+ if matching_results:
88
+ for result in matching_results:
89
+ st.markdown(
90
+ f"""
91
+ <div style="border: 1px solid #ddd; padding: 10px; border-radius: 5px; margin-bottom: 10px;">
92
+ <h4 style="margin: 0;">{result['Name']}</h4>
93
+ <p><strong>Совпадающие навыки:</strong><br>
94
+ {'<br>'.join(['• ' + skill for skill in result['Matching Skills']])}</p>
95
+ <p><strong>Количество совпадений:</strong> {result['Match Count']}</p>
96
+ </div>
97
+ """,
98
+ unsafe_allow_html=True
99
+ )
100
+ else:
101
+ st.write("Нет совпадений с как минимум половиной выбранных навыков.")
102
+ else:
103
+ st.error("Пожалуйста, введите имя и выберите навыки для сравнения.")