Upload 3 files

app.py

@@ -0,0 +1,56 @@
+import streamlit as st
+from prompt import PROMPT_RESUME_ABOUT, PROMPT_ACCEPTANCE_PERCENTAGE, PROMPT_SKILL_IMPROVEMENTS
+from utils import *
+
+
+def main():
+    st.set_page_config(
+        page_title="Resume (ATS) Review Using Gemini AI",
+        page_icon="📋",
+        layout="wide"
+    )
+
+    # st.write(css, unsafe_allow_html=True)
+    st.title("Resume (ATS) Review Using Gemini AI 📋")
+
+    col1, col2 = st.columns([1, 3])
+    
+    with col1:
+        uploaded_file = st.file_uploader("Please Upload your Resume", type=["pdf"])
+
+        if uploaded_file is None:
+            st.warning("Please upload your image")
+        else:
+            st.success("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")
+
+    with col2:
+        input_text = st.text_area("Job Description:", key= "input")
+        with st.spinner('Wait for it...'):
+            if submit1:
+                if uploaded_file is not None:
+                    pdf_content = input_pdf_setup(uploaded_file)
+                    response = get_gemini_response(PROMPT_RESUME_ABOUT, pdf_content, input_text)
+                    st.subheader("Resume Evaluation:")
+                    st.write(response)
+            elif submit2:
+                if uploaded_file is not None:
+                    pdf_content = input_pdf_setup(uploaded_file)
+                    response = get_gemini_response(PROMPT_SKILL_IMPROVEMENTS, pdf_content, input_text)
+                    st.subheader("Skill Improvement Recommendations:")
+                    st.write(response)
+            elif submit3:
+                if uploaded_file is not None:
+                    pdf_content = input_pdf_setup(uploaded_file)
+                    response = get_gemini_response(PROMPT_ACCEPTANCE_PERCENTAGE, pdf_content, input_text)
+                    st.subheader("ATS System Evaluation:")
+                    st.write(response)
+            st.success('Done!')
+
+if __name__ == "__main__":
+    main()

prompt.py

@@ -0,0 +1,41 @@
+PROMPT_RESUME_ABOUT = """
+As an experienced Technical HR Manager, assess the candidate's resume against the job description. Provide a detailed analysis:
+1. **Strengths and Weaknesses:**
+   - Highlight relevant skills, experience, and qualifications.
+   - Identify areas that may need improvement.
+
+2. **Alignment Feedback:**
+   - Offer constructive feedback on how well the resume aligns with role expectations.
+
+Your thorough assessment guides an informed decision on the candidate's suitability for the position.
+"""
+
+PROMPT_SKILL_IMPROVEMENTS = """
+As a seasoned Technical HR Manager, guide individuals seeking skill enhancement for a specific technical role. Provide insights on:
+
+1. **Job-Specific Technical Skills:**
+   - Analyze and advise on skills directly related to job requirements.
+
+2. **Relevant Learning Resources:**
+   - Recommend courses and resources aligning with technical needs.
+
+3. **Customized Skill Development:**
+   - Propose personalized plans for skill acquisition and reinforcement.
+
+Tailor advice to job specifics for targeted skill improvement aligned with career goals.
+"""
+
+PROMPT_ACCEPTANCE_PERCENTAGE = """
+As an adept ATS scanner, evaluate resumes against job descriptions:
+
+1. **Percentage of Match:**
+   - Calculate alignment percentage between resume and job description.
+
+2. **Keywords Missing:**
+   - Identify and list absent keywords or skills from the job description.
+
+3. **Final Thoughts:**
+   - Provide constructive feedback on strengths and suggestions for improvement.
+
+Get a comprehensive evaluation with a clear match percentage, identified missing keywords, and valuable insights for both the candidate and the ATS system.
+"""

utils.py

@@ -0,0 +1,45 @@
+import os
+import google.generativeai as genai
+import pdf2image
+import io
+import base64
+from dotenv import load_dotenv
+
+load_dotenv()
+
+genai.configure(api_key= os.environ['GOOGLE_API_KEY'])
+
+def get_gemini_response(prompt, pdf_content, input):
+    model = genai.GenerativeModel('gemini-pro-vision')
+    data = [prompt]+ pdf_content  + [input]
+    response = model.generate_content(data)
+    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(), 
+           poppler_path= os.environ["PATH_POPPLER"]# for windows if resulting error
+        )
+        # first_page=images[0]
+        pdf_parts = []
+
+        for page_num, image in enumerate(images):
+            # Convert to bytes
+            img_byte_arr = io.BytesIO()
+            image.save(img_byte_arr, format='JPEG')
+            img_byte_arr = img_byte_arr.getvalue()
+
+            pdf_part = {
+                # "page_number": page_num + 1,  # Page numbers usually start from 1
+                "mime_type": "image/jpeg",
+                "data": base64.b64encode(img_byte_arr).decode()
+            }
+            data = [pdf_part]
+
+            pdf_parts+=data
+
+        return pdf_parts
+    else:
+        raise FileNotFoundError("No file uploaded")