Update app.py

app.py

@@ -1,85 +1,89 @@
+
+from dotenv import load_dotenv
+
+load_dotenv()
+import base64
 import streamlit as st
-import PyPDF2
-import re
-
-# Function to extract text from a PDF
-def extract_text_from_pdf(uploaded_file):
-    pdf_reader = PyPDF2.PdfReader(uploaded_file)
-    text = " ".join([page.extract_text() for page in pdf_reader.pages if page.extract_text()])
-    return text
-
-# Function to extract skills from text (reference resumes and user resumes)
-def extract_skills_from_text(text):
-    # A simple way to extract skills based on words in the resume.
-    # This can be more advanced by integrating NER (Named Entity Recognition) or using predefined skill lists.
-    skills = re.findall(r'\b[A-Za-z]+\b', text.lower())  # Extract words
-    skills = set(skills)  # Remove duplicates
-    return skills
-
-# Function to evaluate user's resume based on job description
-def evaluate_resume(candidate_text, job_description_text):
-    # Extract skills dynamically from job description
-    job_description_skills = extract_skills_from_text(job_description_text)
-
-    # Extract skills from candidate's resume
-    candidate_skills = extract_skills_from_text(candidate_text)
-
-    # Calculate matching skills
-    matching_skills = job_description_skills.intersection(candidate_skills)
-    missing_skills = job_description_skills.difference(candidate_skills)
-
-    # Calculate improvement skills (skills present in the JD but not in candidate's resume)
-    improvement_skills = job_description_skills.difference(matching_skills)
-
-    # Calculate matching percentage (based on skills)
-    matching_percentage = (len(matching_skills) / len(job_description_skills)) * 100 if len(job_description_skills) > 0 else 0
-
-    result = {
-        "matching_skills": list(matching_skills),
-        "missing_skills": list(missing_skills),
-        "improvement_skills": list(improvement_skills),
-        "matching_percentage": round(matching_percentage, 2)
-    }
-
-    return result
-
-# Streamlit UI
-st.title("📄 Resume Screening with Job Description Matching")
-st.write("Upload a candidate's resume and provide a job description to check the matching percentage based on skills.")
-
-# Upload job description
-st.subheader("Upload Job Description")
-job_description_file = st.file_uploader("Upload Job Description (PDF)", type=["pdf"])
-
-# Upload candidate resume
-st.subheader("Upload Candidate Resume")
-uploaded_file = st.file_uploader("Upload Candidate Resume (PDF)", type=["pdf"])
-
-if uploaded_file and job_description_file:
-    st.write("🔍 **Processing resume and job description...**")
-
-    # Extract text from candidate resume
-    candidate_text = extract_text_from_pdf(uploaded_file)
-
-    # Extract text from job description
-    job_description_text = extract_text_from_pdf(job_description_file)
-
-    # Evaluate resume based on job description
-    evaluation = evaluate_resume(candidate_text, job_description_text)
-
-    # Display results
-    st.subheader("📌 Evaluation Result:")
-    st.write(f"**🔢 Matching Percentage:** {evaluation['matching_percentage']}%")
-    # st.write(f"**✅ Matching Skills:** {', '.join(evaluation['matching_skills']) or 'None'}")
-    # st.write(f"**❌ Missing Skills:** {', '.join(evaluation['missing_skills']) or 'None'}")
-    st.write(f"**💡 Improvement Skills (Required but Missing):** {', '.join(evaluation['improvement_skills']) or 'None'}")
-
-
-    if evaluation["matching_percentage"] > 70:
-        st.success("✅ Resume is a strong match for the job description!")
-    elif evaluation["matching_percentage"] > 40:
-        st.warning("⚠️ Resume is a partial match for the job description.")
+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.error("❌ Resume does not match well with the job description.")
-else:
-    st.write("Please upload both the job description and candidate resume.")
+        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")
+