Update app.py

app.py

@@ -1,83 +1,81 @@
-import streamlit as st
-import google.generativeai as genai
-import os
-import PyPDF2 as pdf
 from dotenv import load_dotenv
-import json
 
-load_dotenv()  ## Load all environment variables
+load_dotenv()
+import base64
+import streamlit as st
+import os
+import io
+from PIL import Image
+import pdf2image
+import google.generativeai as genai
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Define the prompt template
-input_prompt = """
-Hey, act like a skilled and very experienced ATS (Application Tracking System)
-with a deep understanding of tech fields such as software engineering, data science, 
-data analysis, and big data engineering. Your task is to evaluate the resume 
-based on the given job description.
-
-You must consider that the job market is very competitive, and you should provide 
-the best assistance for improving resumes. Assign the percentage match based on JD 
-and identify the missing keywords with high accuracy.
-
-resume: {text}
-description: {jd}
-
-Return ONLY a JSON response in the following format:
-{{
-  "JD Match": "xx%",
-  "MissingKeywords": ["keyword1", "keyword2", ...],
-  "Profile Summary": "A short summary here"
-}}
-
-DO NOT include any additional text, explanation, or greetings—only return the JSON object.
+def get_gemini_response(input_text, pdf_content, prompt):
+    model = genai.GenerativeModel('gemini-pro-vision')
+    response = model.generate_content([input_text, pdf_content, prompt])
+    return response.text
+
+def input_pdf_setup(uploaded_file):
+    if not uploaded_file:
+        st.warning("Please upload a PDF file.")
+        st.stop()
+
+    # Convert PDF to images
+    images = pdf2image.convert_from_bytes(uploaded_file.read())
+    first_page = images[0]
+
+    # Convert first page to bytes
+    img_byte_arr = io.BytesIO()
+    first_page.save(img_byte_arr, format='JPEG')
+    img_byte_arr = img_byte_arr.getvalue()
+
+    # Encode as base64
+    pdf_parts = [{
+        "mime_type": "image/jpeg",
+        "data": base64.b64encode(img_byte_arr).decode()
+    }]
+    return pdf_parts
+
+# 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:
+    st.success("PDF Uploaded Successfully")
+
+submit1 = st.button("Tell Me About the Resume")
+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.
 """
 
-def get_gemini_response(resume_text, jd_text):
-    """Generates a response from Gemini API based on the resume and job description."""
-    formatted_prompt = input_prompt.format(text=resume_text, jd=jd_text)  # Format the prompt
-    model = genai.GenerativeModel('gemini-1.5-pro')
-    response = model.generate_content(formatted_prompt)
-
-    # Ensure we get only the text response
-    response_text = response.text.strip()
-
-    # Attempt to extract JSON from the response
-    try:
-        parsed_response = json.loads(response_text)
-        return parsed_response  # Return parsed JSON
-    except json.JSONDecodeError:
-        return None  # Return None if response is not valid JSON
-
-def input_pdf_text(uploaded_file):
-    """Extracts text from the uploaded PDF resume."""
-    reader = pdf.PdfReader(uploaded_file)
-    text = ""
-    for page in range(len(reader.pages)):
-        extracted_text = reader.pages[page].extract_text()
-        if extracted_text:
-            text += extracted_text
-    return text
-
-# Streamlit app
-st.title("Smart ATS")
-st.text("Improve Your Resume for ATS Systems")
-
-jd = st.text_area("Paste the Job Description")
-uploaded_file = st.file_uploader("Upload Your Resume", type="pdf", help="Please upload a PDF file")
-
-submit = st.button("Submit")
-
-if submit:
-    if uploaded_file is not None and jd:
-        text = input_pdf_text(uploaded_file)
-        response = get_gemini_response(text, jd)  # Pass formatted resume & JD
+input_prompt3 = """
+You are a skilled ATS (Applicant Tracking System) scanner with expertise in data science and ATS functionality. 
+Your task is to evaluate the resume against the provided job description. Provide a percentage match if the resume aligns with the job description.
+First, output the percentage match. Then, list missing keywords, followed by final thoughts.
+"""
 
-        if response:
-            st.subheader(f"Job Description Match: {response['JD Match']}")
-            st.write("Missing Keywords: ", response["MissingKeywords"])
-            st.write("Profile Summary: ", response["Profile Summary"])
-        else:
-            st.error("Error: Gemini did not return valid JSON. Please try again.")
+if submit1 and not submit3:
+    if uploaded_file:
+        pdf_content = input_pdf_setup(uploaded_file)
+        response = get_gemini_response(input_prompt1, pdf_content, input_text)
+        st.subheader("Response:")
+        st.write(response)
+    else:
+        st.warning("Please upload the resume.")
+
+elif submit3 and not submit1:
+    if uploaded_file:
+        pdf_content = input_pdf_setup(uploaded_file)
+        response = get_gemini_response(input_prompt3, pdf_content, input_text)
+        st.subheader("Response:")
+        st.write(response)
     else:
-        st.warning("Please upload a resume and enter a job description.")
+        st.warning("Please upload the resume.")