Upload 2 files

app.py

@@ -0,0 +1,102 @@
+# 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")
+
+
+
+
+
+
+
+
+
+

requirements.txt

@@ -0,0 +1,5 @@
+google-generativeai
+streamlit
+python-dotenv
+pdf2image
+