Update app.py
Browse files
app.py
CHANGED
@@ -39,63 +39,73 @@ with st.sidebar:
|
|
39 |
|
40 |
''')
|
41 |
|
42 |
-
st.subheader("Candidate Profile 1", divider = "green")
|
43 |
-
|
44 |
-
txt = st.text_area("
|
45 |
job = pd.Series(txt, name="Text")
|
46 |
|
47 |
|
48 |
if 'upload_count' not in st.session_state:
|
49 |
st.session_state['upload_count'] = 0
|
50 |
|
51 |
-
max_attempts =
|
52 |
|
53 |
if st.session_state['upload_count'] < max_attempts:
|
54 |
uploaded_files = st.file_uploader(
|
55 |
-
"Upload your resume
|
56 |
)
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
-
|
68 |
-
|
69 |
-
|
70 |
-
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
|
|
|
|
|
|
|
|
76 |
values='score', color='label')
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
|
|
|
|
86 |
|
87 |
-
|
|
|
|
|
|
|
88 |
x=['Resume 1', 'Jon Description'],
|
89 |
y=['Resume 1', 'Job Description'])
|
90 |
-
|
91 |
|
92 |
-
|
93 |
-
|
94 |
-
|
|
|
95 |
|
96 |
else:
|
97 |
-
st.warning(f"You have reached the maximum
|
98 |
-
|
99 |
if 'upload_count' in st.session_state and st.session_state['upload_count'] > 0:
|
100 |
st.info(f"Files uploaded {st.session_state['upload_count']} time(s).")
|
101 |
|
|
|
39 |
|
40 |
''')
|
41 |
|
42 |
+
st.subheader ("Candidate Profile 1", divider = "green")
|
43 |
+
|
44 |
+
txt = st.text_area("Paste the job description and then press Ctrl + Enter", key = "text 1")
|
45 |
job = pd.Series(txt, name="Text")
|
46 |
|
47 |
|
48 |
if 'upload_count' not in st.session_state:
|
49 |
st.session_state['upload_count'] = 0
|
50 |
|
51 |
+
max_attempts = 3
|
52 |
|
53 |
if st.session_state['upload_count'] < max_attempts:
|
54 |
uploaded_files = st.file_uploader(
|
55 |
+
"Upload your resume", accept_multiple_files=True, type="pdf", key="candidate 1"
|
56 |
)
|
57 |
+
|
58 |
+
if uploaded_files:
|
59 |
+
st.session_state['upload_count'] += 1
|
60 |
+
for uploaded_file in uploaded_files:
|
61 |
+
pdf_reader = PdfReader(uploaded_file)
|
62 |
+
text_data = ""
|
63 |
+
for page in pdf_reader.pages:
|
64 |
+
text_data += page.extract_text()
|
65 |
+
data = pd.Series(text_data, name = 'Text')
|
66 |
+
|
67 |
+
|
68 |
+
frames = [job, data]
|
69 |
+
result = pd.concat(frames)
|
70 |
+
|
71 |
+
|
72 |
+
model = GLiNER.from_pretrained("urchade/gliner_base")
|
73 |
+
labels = ["person", "country","organization", "date", "time", "role", "skills", "year"]
|
74 |
+
entities = model.predict_entities(text_data, labels)
|
75 |
+
df = pd.DataFrame(entities)
|
76 |
+
|
77 |
+
|
78 |
+
st.subheader("Profile of candidate 1")
|
79 |
+
fig1 = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
|
80 |
values='score', color='label')
|
81 |
+
fig1.update_layout(margin = dict(t=50, l=25, r=25, b=25))
|
82 |
+
st.plotly_chart(fig1, key = "figure 1")
|
83 |
+
|
84 |
+
|
85 |
+
|
86 |
+
vectorizer = TfidfVectorizer()
|
87 |
+
tfidf_matrix = vectorizer.fit_transform(result)
|
88 |
+
tfidf_df = pd.DataFrame(tfidf_matrix.toarray(), columns=vectorizer.get_feature_names_out())
|
89 |
+
|
90 |
+
cosine_sim_matrix = cosine_similarity(tfidf_matrix)
|
91 |
+
cosine_sim_df = pd.DataFrame(cosine_sim_matrix)
|
92 |
|
93 |
+
|
94 |
+
|
95 |
+
st.subheader("Measuring similarity between keywords of candidate profile 1 and job description")
|
96 |
+
fig2 = px.imshow(cosine_sim_df, text_auto=True, labels=dict(x="Keyword similarity", y="Resumes", color="Productivity"),
|
97 |
x=['Resume 1', 'Jon Description'],
|
98 |
y=['Resume 1', 'Job Description'])
|
99 |
+
st.plotly_chart(fig2, key = "figure 2")
|
100 |
|
101 |
+
|
102 |
+
for i, similarity_score in enumerate(cosine_sim_matrix[0][1:]):
|
103 |
+
st.write(f"Similarity of job description with candidate profile 1. {i + 1}: {similarity_score:.4f}")
|
104 |
+
st.write("A score closer to 1 (0.80, 0.90) means higher similarity between candidate profile 1 and job description. A score closer to 0 (0.20, 0.30) means lower similarity between candidate profile 1 and job description.")
|
105 |
|
106 |
else:
|
107 |
+
st.warning(f"You have reached the maximum URL attempts ({max_attempts}).")
|
108 |
+
|
109 |
if 'upload_count' in st.session_state and st.session_state['upload_count'] > 0:
|
110 |
st.info(f"Files uploaded {st.session_state['upload_count']} time(s).")
|
111 |
|