Spaces:
Sleeping
Sleeping
File size: 2,486 Bytes
d20782d c8c71f5 d20782d 5460684 d20782d c8c71f5 d20782d c8c71f5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
# Import necessary libraries
from convert import ExtractPDFText
from ATS_score import calculateATSscore
from model import modelFeedback
import streamlit as st
import time
if "page_number" not in st.session_state:
st.session_state.page_number = 1
if "resume_data" not in st.session_state:
st.session_state.resume_data = ""
if "jobdescription" not in st.session_state:
st.session_state.jobdescription = ""
def set_page_number_and_reset_data():
st.session_state.page_number = 1
st.session_state.resume_data = ""
def page1():
st.title("AI-Powered ATS Screening")
if not st.session_state.resume_data:
pdf = st.file_uploader(label="Upload your resume", type="pdf")
st.write("No Resume Yet? Create one [here](https://www.overleaf.com/latex/templates/tagged/cv)")
if pdf:
st.success("Resume uploaded successfully.")
st.session_state.resume_data = ExtractPDFText(pdf)
def page2():
st.title("AI-Powered ATS Screening: Job Description")
st.session_state.jobdescription = st.text_area("Job Description: ")
st.info("You can just copy paste from the job portal")
submit = st.button("Submit")
if submit:
start()
def page3():
st.title("Your Resume data: ")
if st.session_state.resume_data:
st.write(st.session_state.resume_data)
else:
st.error("Please upload your resume to view the extracted data")
def start():
if st.session_state.resume_data and st.session_state.jobdescription:
with st.spinner("Hold on, we're calculating your ATS Score..."):
ATS_score = calculateATSscore(st.session_state.resume_data, st.session_state.jobdescription)
model_feedback = modelFeedback(ATS_score, st.session_state.resume_data)
# time.sleep(5)
st.subheader("AI FEEDBACK:")
st.write(model_feedback.text)
else:
st.info("Please, upload Resume and Provide the Job Description")
if st.session_state.page_number == 1:
page1()
elif st.session_state.page_number == 2:
page2()
elif st.session_state.page_number == 3:
page3()
if st.session_state.page_number == 1:
st.button("View your Extracted Resume data", on_click = lambda: setattr(st.session_state,"page_number", 3))
st.button("Go to Job Description Page", on_click=lambda: setattr(st.session_state, "page_number", 2))
else:
st.button("Go to PDF Upload Page", on_click=lambda: set_page_number_and_reset_data())
|