app.py

from dotenv import load_dotenv

load_dotenv()

import base64
import streamlit as st
import os
import io
from PIL import Image
import pdf2image
import google.generativeai as genai

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

def get_gemini_response(input,pdf_content,prompt):
    model = genai.GenerativeModel('gemini-1.5-flash')
    response = model.generate_content([input,pdf_content[0],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())

        first_page =images[0]

        ## Convert to bytes
        img_byte_arr = io.BytesIO()
        first_page.save(img_byte_arr,format='JPEG')
        img_byte_arr =img_byte_arr.getvalue()

        pdf_parts =[ 
            {
                "mime_type":"image/jpeg",
                "data":base64.b64encode(img_byte_arr).decode()  #encode to base64
            }            
        ]

        return pdf_parts
    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 Succesfully")

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 Techncal Human Resource Manager the field of any one job role from Data Science,Full stack
Web development, Big Data Engineering,DEVOPS,Data Analyst  your task is to review 
the provided resume against the job description for theses profiles.
Please share your professional evaluation on whether the candidate's
profile aligns with 
Highlighting the strengths and the weakness of the applicant in relation to the specific
"""
## input prompt1 always covers the submit button 2

input_prompt3 ="""
You are a skilled ATS (Applicant Tracking System) scanner with a deep understanding ofData Science,Full stack
Web development, Big Data Engineering,DEVOPS,Data Analyst and deep 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 a 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_gemini_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_gemini_response(input_prompt3,pdf_content,input_text)
        st.subheader("The response is")
        st.write(response)
    else:
        st.write("Please upload the resume")