Update app.py

app.py

@@ -1,34 +1,51 @@
-from sentence_transformers import SentenceTransformer
 import streamlit as st
-import pandas as pd
 from PyPDF2 import PdfReader
-import nltk
-nltk.download('punkt_tab')
-from nltk.corpus import stopwords
-nltk.download('stopwords')
-from nltk.tokenize import word_tokenize
-
-from sentence_transformers import SentenceTransformer
-import streamlit as st
 import pandas as pd
-from PyPDF2 import PdfReader
-import nltk
-from nltk.corpus import stopwords
-from nltk.tokenize import word_tokenize
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
+from gliner import GLiNER
+import plotly.express as px
+import time
+
+with st.sidebar:
+    st.button("DEMO APP", type="primary")
+   
 
+    expander = st.expander("**Important notes on the AI Resume Analysis based on Keywords App**")
+    expander.write('''
+    
+    
+     **Supported File Formats**
+    This app accepts files in .pdf formats.
+    
+    **How to Use**
+    Paste the job description first. Then, upload the resume of each applicant to retrieve the results.  
+    
+    **Usage Limits**
+    For each applicant you can upload their resume and request results once (1 request per applicant's resume).
+    At the bottom of the app, you can also upload (up to 50 times) each applicant's resume to visualize their profile as a treemap chart as well as the results in a matrix heatmap. If you hover over the interactive graphs, an icon will appear to download them.
+    
+     **Subscription Management**
+    This demo app offers a one-day subscription, expiring after 24 hours. If you are interested in building your own AI Resume Analysis based on Keywords Web App, we invite you to explore our NLP Web App Store on our website. You can select your desired features, place your order, and we will deliver your custom app within five business days. If you wish to delete your Account with us, please contact us at [email protected]
+    
+    **Customization**
+    To change the app's background color to white or black, click the three-dot menu on the right-hand side of your app, go to Settings and then Choose app theme, colors and fonts.
+    
+    **File Handling and Errors**
+    The app may display an error message if your file is corrupt, or has other errors.
+    
+    
+    For any errors or inquiries, please contact us at [email protected]
+   
+    
+    
+''')
 
-stop_words = set(stopwords.words('english'))
 
 model = SentenceTransformer("all-mpnet-base-v2")
 st.title("AI Resume Analysis based on Keywords App")
 st.divider()
-job_desc_raw = st.text_area("Paste the job description and then press Ctrl + Enter", key="job_desc")
-
-# Process job description for stop words
-job_desc_tokens = word_tokenize(job_desc_raw.lower())
-job_desc_filtered = [word for word in job_desc_tokens if not word in stop_words]
-job_desc_processed = " ".join(job_desc_filtered)
-st.write("Processed Job Description:", job_desc_processed)
+job_desc = st.text_area("Paste the job description and then press Ctrl + Enter", key="job_desc")
 
 if 'applicant_data' not in st.session_state:
     st.session_state['applicant_data'] = {}
@@ -48,23 +65,22 @@ for i in range(1, 51):  # Looping for 50 applicants
         if st.session_state['applicant_data'][applicant_key]['uploaded_file'] and not st.session_state['applicant_data'][applicant_key]['analysis_done']:
             try:
                 pdf_reader = PdfReader(st.session_state['applicant_data'][applicant_key]['uploaded_file'])
-                text_data_raw = ""
+                text_data = ""
                 for page in pdf_reader.pages:
-                    text_data_raw += page.extract_text()
+                    text_data += page.extract_text()
                 with st.expander(f"See Applicant's {i} resume"):
-                    st.write(text_data_raw)
+                    st.write(text_data)
 
-                # Process resume text for stop words
-                text_tokens = word_tokenize(text_data_raw.lower())
-                text_data_filtered = [word for word in text_tokens if not word in stop_words]
-                text_data_processed = " ".join(text_data_filtered)
-                st.write("Processed Resume:", text_data_processed)
+                # Encode the job description and resume text separately
+                job_embedding = model.encode([job_desc])
+                resume_embedding = model.encode([text_data])
 
-                # Encode the processed job description and resume text
-                job_embedding = model.encode([job_desc_processed])
-                resume_embedding = model.encode([text_data_processed])
                 # Calculate the cosine similarity between the two embeddings
                 similarity_score = model.similarity(job_embedding, resume_embedding)[0][0]
+
+                with st.spinner("Wait for it...", show_time=True):
+                    time.sleep(2)
+
                 with st.popover(f"See Result for Applicant {i}"):
                     st.write(f"Similarity between Applicant's resume and job description based on keywords: {similarity_score:.2f}")
                     st.info(
@@ -75,4 +91,58 @@ for i in range(1, 51):  # Looping for 50 applicants
     else:
         st.warning(f"Maximum upload attempts has been reached ({max_attempts}).")
         if st.session_state['applicant_data'][applicant_key]['upload_count'] > 0:
-            st.info(f"Files uploaded for Applicant {i}: {st.session_state['applicant_data'][applicant_key]['upload_count']} time(s).")
+            st.info(f"Files uploaded for Applicant {i}: {st.session_state['applicant_data'][applicant_key]['upload_count']} time(s).")
+
+
+
+
+st.divider()
+st.subheader("Visualise", divider="blue")
+
+if 'upload_count' not in st.session_state:
+    st.session_state['upload_count'] = 0
+    max_attempts = 1
+
+if st.session_state['upload_count'] < max_attempts:
+    uploaded_files = st.file_uploader("Upload Applicant's resume", type="pdf", key="applicant 1")
+    if uploaded_files:
+        st.session_state['upload_count'] += 1
+        with st.spinner("Wait for it...", show_time=True):
+            time.sleep(2)
+            pdf_reader = PdfReader(uploaded_files)
+            text_data = ""
+            for page in pdf_reader.pages:
+                text_data += page.extract_text()
+
+            data = pd.Series(text_data, name='Text')
+            frames = [job_desc, data]
+            result = pd.concat(frames)
+            model = GLiNER.from_pretrained("urchade/gliner_base")
+            labels = ["person", "country", "organization", "role", "skills"]
+            entities = model.predict_entities(text_data, labels)
+            df = pd.DataFrame(entities)
+            st.subheader("Applicant's Profile", divider = "orange")
+            fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
+                             values='score', color='label')
+            fig.update_layout(margin=dict(t=50, l=25, r=25, b=25))
+            st.plotly_chart(fig, key="figure 1")
+
+            
+            job_embedding = model.encode([job_desc])
+            resume_embedding = model.encode([text_data])
+
+                
+            similarity_score = model.similarity(job_embedding, resume_embedding)[0][0]
+            
+                  
+            st.subheader("Similarity between Applicant's Profile and Job Description", divider = "orange")
+
+            fig = px.imshow(similarity_score, text_auto=True,
+                            labels=dict(x="Keyword similarity", y="Resumes", color="Productivity"),
+                            x=['Resume', 'Jon Description'],
+                            y=['Resume', 'Job Description'])
+            st.plotly_chart(fig, key="figure 2")
+else:
+    st.warning(f"Maximum upload attempts has been reached ({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).")