Update src/app_job_copy_1.py

src/app_job_copy_1.py

@@ -410,28 +410,34 @@ def display_job_selection(jobs_df, candidates_df, sh):
     # Clear previous job state when a new job is selected
     if selected_job_index != st.session_state.last_selected_job_index:
         old_job_key = st.session_state.last_selected_job_index
-
-        # Clear job-specific session state
         job_processed_key = f"job_{old_job_key}_processed_successfully"
         job_is_processing_key = f"job_{old_job_key}_is_currently_processing"
 
+        # Remove old job flags
         for key in [job_processed_key, job_is_processing_key, 'stop_processing_flag', 'total_input_tokens', 'total_output_tokens']:
-            if key in st.session_state:
-                del st.session_state[key]
+            st.session_state.pop(key, None)
 
-        # Clear selected candidates for the old job
-        if 'Selected_Candidates' in st.session_state and old_job_key in st.session_state.Selected_Candidates:
-            del st.session_state.Selected_Candidates[old_job_key]
+        # Clear selected candidates for old job if they exist
+        if 'Selected_Candidates' in st.session_state:
+            st.session_state.Selected_Candidates.pop(old_job_key, None)
 
-        # Clear cached data
+        # Clear cache to avoid old data in UI
         st.cache_data.clear()
 
         # Update last selected job index
         st.session_state.last_selected_job_index = selected_job_index
 
-        # Force rerun to refresh UI
+        # Rerun to refresh UI and prevent stale data
         st.rerun()
 
+    # Ensure Selected_Candidates is initialized for the new job
+    if 'Selected_Candidates' not in st.session_state:
+        st.session_state.Selected_Candidates = {}
+
+    if selected_job_index not in st.session_state.Selected_Candidates:
+        st.session_state.Selected_Candidates[selected_job_index] = []
+
+    # Proceed with job details
     job_row = jobs_df.iloc[selected_job_index]
     job_row_stack = parse_tech_stack(job_row["Tech Stack"])
 
@@ -528,13 +534,13 @@ def display_job_selection(jobs_df, candidates_df, sh):
             headers = existing_candidates_from_sheet[0]
             for row in existing_candidates_from_sheet[1:]:
                 cand = {headers[i]: row[i] if i < len(row) else None for i in range(len(headers))}
-                try: cand['Fit Score'] = float(cand.get('Fit Score',0))
+                try: cand['Fit Score'] = float(cand.get('Fit Score', 0))
                 except: cand['Fit Score'] = 0.0
                 final_candidates.append(cand)
-            final_candidates.sort(key=lambda x: x.get('Fit Score',0.0), reverse=True)
+            final_candidates.sort(key=lambda x: x.get('Fit Score', 0.0), reverse=True)
 
     if should_display:
-        col_title, col_copyall = st.columns([3,1])
+        col_title, col_copyall = st.columns([3, 1])
         with col_title:
             st.subheader("Selected Candidates")
         with col_copyall:
@@ -559,8 +565,8 @@ def display_job_selection(jobs_df, candidates_df, sh):
             display_token_usage()
 
         for i, candidate in enumerate(final_candidates):
-            score = candidate.get('Fit Score',0.0)
-            score_display = f"{score:.3f}" if isinstance(score,(int,float)) else score
+            score = candidate.get('Fit Score', 0.0)
+            score_display = f"{score:.3f}" if isinstance(score, (int, float)) else score
             exp_title = f"{i+1}. {candidate.get('Name','N/A')} (Score: {score_display})"
             with st.expander(exp_title):
                 text_copy = f"Candidate: {candidate.get('Name','N/A')}\nLinkedIn: {candidate.get('LinkedIn','N/A')}\n"