Update app.py

app.py

@@ -14,6 +14,11 @@ if "resume_data" not in st.session_state:
 
 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")
@@ -51,7 +56,7 @@ def start():
 
         st.subheader("AI FEEDBACK:")
         st.write(model_feedback.text)
-        st.session_state.resume_data = ""
+        
     else:
         st.info("Please, upload Resume and Provide the Job Description")
 
@@ -66,4 +71,5 @@ 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: setattr(st.session_state, "page_number", 1))
+    st.button("Go to PDF Upload Page", on_click=lambda: set_page_number_and_reset_data())
+