Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 import google.generativeai as genai
 import os
@@ -6,50 +5,64 @@ import PyPDF2 as pdf
 from dotenv import load_dotenv
 import json
 
-load_dotenv() ## load all our environment variables
+load_dotenv()  ## Load all environment variables
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-def get_gemini_repsonse(input):
-    model=genai.GenerativeModel('gemini-1.5-pro')
-    response=model.generate_content(input)
+def get_gemini_response(input_text):
+    model = genai.GenerativeModel('gemini-1.5-pro')
+    response = model.generate_content(input_text)
     return response.text
 
 def input_pdf_text(uploaded_file):
-    reader=pdf.PdfReader(uploaded_file)
-    text=""
+    reader = pdf.PdfReader(uploaded_file)
+    text = ""
     for page in range(len(reader.pages)):
-        page=reader.pages[page]
-        text+=str(page.extract_text())
+        extracted_text = reader.pages[page].extract_text()
+        if extracted_text:
+            text += extracted_text
     return text
 
-#Prompt Template
-
-input_prompt="""
-Hey Act Like a skilled or very experience ATS(Application Tracking System)
-with a deep understanding of tech field,software engineering,data science ,data analyst
-and big data engineer. Your task is to evaluate the resume based on the given job description.
-You must consider the job market is very competitive and you should provide 
-best assistance for improving thr resumes. Assign the percentage Matching based 
-on Jd and
-the missing keywords with high accuracy
-resume:{text}
-description:{jd}
-
-I want the response in one single string having the structure
-{{"JD Match":"%","MissingKeywords:[]","Profile Summary":""}}
+# 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}
+
+I want the response in one single JSON string with the following structure:
+{{"JD Match":"%", "MissingKeywords":[], "Profile Summary":""}}
 """
 
-## streamlit app
+## Streamlit app
 st.title("Smart ATS")
-st.text("Improve Your Resume ATS")
-jd=st.text_area("Paste the Job Description")
-uploaded_file=st.file_uploader("Upload Your Resume",type="pdf",help="Please uplaod the pdf")
+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:
-        text=input_pdf_text(uploaded_file)
-        response=get_gemini_repsonse(input_prompt)
-        st.subheader(response)
+    if uploaded_file is not None and jd:
+        text = input_pdf_text(uploaded_file)
+        formatted_prompt = input_prompt.format(text=text, jd=jd)
+        response = get_gemini_response(formatted_prompt)
+        
+        # Try parsing the response as JSON
+        try:
+            parsed_response = json.loads(response)
+            st.subheader("Job Description Match: " + parsed_response["JD Match"])
+            st.write("Missing Keywords: ", parsed_response["MissingKeywords"])
+            st.write("Profile Summary: ", parsed_response["Profile Summary"])
+        except json.JSONDecodeError:
+            st.error("Error: Could not parse the response. Please check the response format.")
+    else:
+        st.warning("Please upload a resume and enter a job description.")