Upload 2 files

app.py

@@ -0,0 +1,82 @@
+import io
+from dotenv import load_dotenv
+
+load_dotenv()
+
+import streamlit as st
+import os
+from PIL import Image
+import pdf2image
+import base64
+import google.generativeai as genai
+
+genai.configure(api_key=os.getenv("API_KEY"))
+
+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
+
+def input_pdf_setup(upload_file):
+    if upload_file is not None:
+                images=pdf2image.convert_from_bytes(upload_file.read())
+
+                first_page = images[0]
+
+                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()
+                }
+                ]
+                return pdf_parts
+    else:
+         raise FileNotFoundError("No File uploaded")
+
+st.set_page_config(page_title="Resume Parser")
+st.header("Mock ATS")
+input_text = st.text_area("Paste the job description here: ",key="input")
+uploaded_file = st.file_uploader("Upload your resume (only PDFs are supported)...", type="pdf")
+
+if uploaded_file is not None:
+    st.write("Resume uploaded successfully")
+
+    submit1=st.button("Tell me about the resume")
+
+    submit2=st.button("Percentage match with the job description")
+
+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_prompt2 = """
+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 submit2:
+    if uploaded_file is not None:
+        pdf_content=input_pdf_setup(uploaded_file)
+        response=get_gemini_response(input_prompt2,pdf_content,input_text)
+        st.subheader("The Repsonse is")
+        st.write(response)
+    else:
+        st.write("Please uplaod the resume")
+
+    

requirements.txt

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