Update app.py

app.py

@@ -49,7 +49,7 @@ job = pd.Series(txt, name="Text")
 if 'upload_count' not in st.session_state:
     st.session_state['upload_count'] = 0
 
-max_attempts = 2
+max_attempts = 3
 
 if st.session_state['upload_count'] < max_attempts:
     uploaded_files = st.file_uploader(
@@ -94,6 +94,7 @@ if st.session_state['upload_count'] < max_attempts:
                 for i, similarity_score in enumerate(cosine_sim_matrix[0][1:]):
                     st.write(f"Similarity with Candidate Profile {i + 1}: {similarity_score:.4f}")
 
+
 else:
     st.warning(f"You have reached the maximum upload attempts ({max_attempts})")
 
@@ -104,58 +105,8 @@ if 'upload_count' in st.session_state and st.session_state['upload_count'] > 0:
 
     
                
-st.subheader("Candidate Profile 2", divider = "green")
-             
-txt = st.text_area("Job description", key = "text 2")
-job = pd.Series(txt, name="Text")
-if 'upload_count' not in st.session_state:
-    st.session_state['upload_count'] = 0
-max_attempts = 2
-if st.session_state['upload_count'] < max_attempts:
-    uploaded_files = st.file_uploader(
-        "Upload your resume in .pdf format", type="pdf", key="candidate 2"
-    )
-    if uploaded_files:
-        st.session_state['upload_count'] += 1
-        for uploaded_file in uploaded_files:
-            pdf_reader = PdfReader(uploaded_file)
-            text_data = ""
-            for page in pdf_reader.pages:
-                text_data += page.extract_text()
-                data = pd.Series(text_data, name = 'Text')
-                frames = [job, data]
-                result = pd.concat(frames)
-                
-                
-                model = GLiNER.from_pretrained("urchade/gliner_base")
-                labels = ["person", "country","organization", "date", "time", "role", "skills", "year"]
-                entities = model.predict_entities(text_data, labels)
-                df = pd.DataFrame(entities)
-                
-                
-                fig1 = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
-                values='score', color='label')
-                fig1.update_layout(margin = dict(t=50, l=25, r=25, b=25))
-                st.plotly_chart(fig1, key = "figure 3")
-                
-                vectorizer = TfidfVectorizer()
-                tfidf_matrix = vectorizer.fit_transform(result)
-                tfidf_df = pd.DataFrame(tfidf_matrix.toarray(), columns=vectorizer.get_feature_names_out())
-                cosine_sim_matrix = cosine_similarity(tfidf_matrix)
-                cosine_sim_df = pd.DataFrame(cosine_sim_matrix)
+
                 
-        
-                fig2 = px.imshow(cosine_sim_df, text_auto=True, labels=dict(x="Keyword similarity", y="Resumes", color="Productivity"),
-                        x=['Resume 1', 'Jon Description'],
-                        y=['Resume 1', 'Job Description'])
-                st.plotly_chart(fig2, key = "figure 4")
-                st.subheader("Cosine Similarity Scores (Job Description vs. Resumes):")
-                for i, similarity_score in enumerate(cosine_sim_matrix[0][1:]):
-                    st.write(f"Similarity with Candidate Profile {i + 1}: {similarity_score:.4f}")
-else:
-    st.warning(f"You have reached the maximum upload attempts ({max_attempts})")
-if 'upload_count' in st.session_state and st.session_state['upload_count'] > 0:
-    st.info(f"Files uploaded {st.session_state['upload_count']} time(s).")