Spaces:
Sleeping
Sleeping
| # importing libraries | |
| import streamlit as st | |
| import google.generativeai as genai | |
| from dotenv import load_dotenv | |
| import pdf2image | |
| from PIL import Image | |
| import os | |
| import io | |
| import base64 | |
| # loading environment variables | |
| load_dotenv() | |
| # Google API KEY configuration | |
| genai.configure(api_key=os.getenv('GOOGLE_API_KEY')) | |
| # generating response from Gemini API | |
| def get_gemini_response(input, pdf_content, prompt): | |
| model = genai.GenerativeModel('gemini-pro-vision') | |
| response = model.generate_content([input, pdf_content[0],prompt]) | |
| return response.text | |
| # taking PDF and converting into image format | |
| def input_pdf_setup(uploaded_file): | |
| if uploaded_file is not None: | |
| # convert PDF to image format | |
| images = pdf2image.convert_from_bytes(uploaded_file.read()) | |
| # content from 1st page | |
| first_page = images[0] | |
| # convert to bytes | |
| img_bytes_arr = io.BytesIO() | |
| first_page.save(img_bytes_arr, format='JPEG') | |
| img_bytes_arr = img_bytes_arr.getvalue() | |
| pdf_parts = [{ | |
| "mime_type": "image/jpeg", | |
| "data" : base64.b64encode(img_bytes_arr).decode() # encode to base64 | |
| }] | |
| return pdf_parts | |
| else: | |
| raise FileNotFoundError("No file uploaded") | |
| # streamlit application | |
| st.set_page_config(page_title="ATS Resume Expert") | |
| st.header("ATS Tracking System") | |
| input_text = st.text_area("Job Description: ", key="input") | |
| uploaded_file = st.file_uploader("Upload your resume(PDF)...", type=['pdf']) | |
| if uploaded_file is not None: | |
| st.write("PDF Uploaded Successfully") | |
| # creating buttons | |
| submit1 = st.button("Tell Me About the Resume") | |
| # submit2 = st.button("How Can I Improvise my skills?") | |
| submit3 = st.button("Percentage match") | |
| input_prompt1 = """ | |
| You are an experienced Technical Human Resource Manager,your task is to review the provided resume against the job description. | |
| Please share your professional evaluation on whether the candidate's profile aligns with the role. | |
| Highlight the strengths and weaknesses of the applicant in relation to the specified job requirements. | |
| """ | |
| input_prompt3 = """ | |
| You are an skilled ATS (Applicant Tracking System) scanner with a deep understanding of data science and ATS functionality, | |
| your task is to evaluate the resume against the provided job description. give me the percentage of match if the resume matches | |
| the job description. First the output should come as percentage and then keywords missing and last final thoughts. | |
| """ | |
| # for submit1 | |
| if submit1: | |
| if uploaded_file is not None: | |
| pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(input_prompt1, pdf_content, input_text) | |
| st.subheader("The Response is") | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| # for submit3 | |
| elif submit3: | |
| if uploaded_file is not None: | |
| pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(input_prompt3, pdf_content, input_text) | |
| st.subheader("The Response is") | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |