ats.py

# in this project we are taking a pdf(Resume), converting it into text using poppler-windows + pdf2image library and then extracting bytes from that image

from dotenv import load_dotenv
load_dotenv()
import streamlit as st
import os
import io
import base64
import google.generativeai as genai
from PIL import Image
import pdf2image # this pdf2image requires poppler-windows set up


genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
# gemini=pr-vision is used because we want to convert pdf to image

def get_response(input, pdf_content, prompt):
    model = genai.GenerativeModel("gemini-1.5-flash")
    response = model.generate_content([input, pdf_content, prompt])
    return response.text

def input_pdf_setup(uploaded_file):
    if uploaded_file is not None:
        # Convert the PDF to image
        images = pdf2image.convert_from_bytes(uploaded_file.read())

        # Convert the first page to bytes
        first_page = images[0]
        img_byte_arr = io.BytesIO()
        first_page.save(img_byte_arr, format='JPEG')
        img_byte_arr = img_byte_arr.getvalue()

        # Encode the bytes to base64
        pdf_content = base64.b64encode(img_byte_arr).decode()

        return pdf_content
    else:
        raise FileNotFoundError("No file uploaded")

# streamlit app
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!")

submit1=st.button("Tell me About the Resume")
submit2=st.button("How can I Improvise my Skills")
# submit3=st.button("What are the Keywords that are missing")
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.
"""

if submit1:
    if uploaded_file is not None:
        pdf_content = input_pdf_setup(uploaded_file)
        response = get_response(input_prompt1, pdf_content, input_text)
        st.subheader("The Response is")
        st.write(response)
    else:
        st.write("Please upload the resume")

elif submit3:
    if uploaded_file is not None:
        pdf_content=input_pdf_setup(uploaded_file)
        response=get_response(input_prompt3,pdf_content,input_text)
        st.subheader("The Repsonse is")
        st.write(response)
    else:
        st.write("Please uplaod the resume")