Create app.py

app.py

@@ -0,0 +1,84 @@
+import streamlit as st
+import os 
+import io 
+import base64
+from PIL import Image
+import pdf2image 
+import google.generativeai as genai 
+
+genai.configure(api_key = os.getenv('GEN_AI_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(uploaded_file):
+    if uploaded_file is not None:
+        ## convert pdf to image
+        images = pdf2image.convert_from_bytes(uploaded_file.read())
+        first_page = images[0]
+        # take image and convert into 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()
+            }
+        ]
+        return pdf_parts
+    else:
+    raise FileNotFoundError("No File uploaded")
+
+
+# Streamlit app
+st.set_page_config(page_title = "ATS Resume Analyser")
+st.header("ATS Tracker")
+input_text = st.text_area("Job Description: ", key = "input")
+uploaded_file = st.file_uploader("Upload your resume in pdf format...", type = ["pdf"])
+
+if uploaded_fileis not None:
+    st.write("PDF uploaded successfully ")
+
+submit1 = st.button("Tell me about the resume")
+# submit2 = st.button("How can I improve my skills")
+# submit3 = st.button("What are the missing keywords in my resume")
+submit3 = st.button("Percentage match to Job description")
+
+input_prompt1 = """
+You are an experienced Technical Human Resource Manager with tech experience with experience in data science, LLMs, 
+machine learning, deep learning, devops and data analyst,
+your task is to review the provided resume and share your professional evaluation on 
+whether the candidate's profile aligns with the Job description. Highlight the strengths and weakness of the applicant
+in relation to the specified job description
+
+"""
+input_prompt3 = """
+You are skilled ATS (Application Tracking System) scanner with a deep understanding of any one job role data science, LLMs, 
+machine learning, deep learning, devops, data analyst and deep ATS functionality. Your task is to evaluate the given resume against provided job description
+Give me a percentage match if the resume matches the job description and should come as a percentage followed by keywords missing in the resume.
+
+"""
+if submit1:
+    if uploaded_file is not None:
+        pdf_content = input_pdf_setup(uploaded_file)
+        response = get_gemini_response(input_text, pdf_content, input_prompt1)
+        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_text, pdf_content,input_prompt3)
+        st.subheader("The response is: ")
+        st.write(response)
+    else:
+        st.write("Please upload the resume")
+
+
+