Spaces:
Sleeping
Sleeping
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_cotent,prompt): | |
model=genai.GenerativeModel('gemini-pro-vision') | |
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 Successfully") | |
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. | |
""" | |
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 Repsonse is") | |
st.write(response) | |
else: | |
st.write("Please uplaod 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 Repsonse is") | |
st.write(response) | |
else: | |
st.write("Please uplaod the resume") | |