Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from dotenv import load_dotenv | |
| from utils import * | |
| import uuid | |
| #Creating session variables | |
| if 'unique_id' not in st.session_state: | |
| st.session_state['unique_id'] ='' | |
| def main(): | |
| load_dotenv() | |
| st.set_page_config(page_title="Resume Screening Assistance", page_icon="📝") | |
| st.title("HR - Resume Screening Assistance...") | |
| # st.subheader("I can help you in resume screening process") | |
| job_description = st.text_area("Please paste the 'JOB DESCRIPTION' here...", key="desc") | |
| document_count = st.text_input("Enter the no.of resumes to return",key="count") | |
| # Upload the Resumes (pdf files) | |
| pdf = st.file_uploader("Upload resumes here, only PDF files allowed", type=["pdf"], accept_multiple_files=True) | |
| submit=st.button("Screen") | |
| if submit: | |
| with st.spinner('Wait for it...'): | |
| #Creating a unique ID, so that we can use to query and get only the user uploaded documents from PINECONE vector store | |
| st.session_state['unique_id']=uuid.uuid4().hex | |
| #Create a documents list out of all the user uploaded pdf files | |
| final_docs_list=create_docs(pdf,st.session_state['unique_id']) | |
| #Displaying the count of resumes that have been uploaded | |
| st.write("*Resumes uploaded* :"+str(len(final_docs_list))) | |
| #Create embeddings instance | |
| embeddings=create_embeddings_load_data() | |
| #Push data to PINECONE | |
| # using already created index automatic-ticket-tool | |
| table = push_to_lancedb(embeddings) | |
| #Fecth relavant documents from LanceDB | |
| relavant_docs = similar_docs_lancedb(job_description,table,embeddings,final_docs_list) | |
| #st.write(relavant_docs) | |
| #Introducing a line separator | |
| st.write(":heavy_minus_sign:" * 30) | |
| st.success(f"Find below the {document_count} Resumes") | |
| # For each item in relavant docs - we are displaying some info of it on the UI | |
| for idx, resume in enumerate(relavant_docs[:int(document_count)]): | |
| st.subheader("👉 Resume "+str(idx+1)) | |
| # st.write(resume.page_content) | |
| #Gets the summary of the current item using 'get_summary' function that we have created which uses LLM & Langchain chain | |
| summary = get_summary(relavant_docs[idx]) | |
| print([relavant_docs[idx]]) | |
| st.write("**Summary** : "+summary) | |
| st.success("Hope I was able to save your time❤️") | |
| #Invoking main function | |
| if __name__ == '__main__': | |
| main() | |