Update src/streamlit_app.py

src/streamlit_app.py

@@ -277,40 +277,55 @@ if st.session_state.stage == 'start':
     uploaded_file = st.file_uploader("Upload your Resume (PDF)", type=["pdf"])
     
     if uploaded_file:
-        with st.spinner("Analyzing your resume... This may take a moment."):
-            try:
-                resume_text = read_resume(uploaded_file)
-                if not resume_text:
-                    st.error("Could not extract text from the resume. Please try another file.")
-                    st.session_state.stage = 'start'
-                else:
-                    # 1. Generate Questions
-                    st.session_state.questions = generate_questions_from_resume(resume_text, gen_q_model)
-                    if not st.session_state.questions:
-                        st.warning("No AI-generated questions returned; using fallback questions.")
-                        st.session_state.questions = generate_questions_from_resume(resume_text, None)
-
-                    # 2. Get AI Introduction
-                    intro_output = get_introduction(intro_model)
-                    st.session_state.current_question = getattr(intro_output, 'question', "Can you introduce yourself?")
-
-                    # 3. Move to next stage and display intro
-                    st.session_state.stage = 'awaiting_intro'
-                    
-                    # --- MODIFIED: Display text directly ---
-                    text_to_speech_and_display(getattr(intro_output, 'intro', "Hello, I'm Interviewer.AI. Please introduce yourself."))
-                    text_to_speech_and_display(getattr(intro_output, 'question', "Can you introduce yourself?"))
-                    
-                    st.rerun()
-            except Exception as e:
-                st.error(f"An error occurred while processing the resume. Using fallback behaviour. Error: {e}")
-                fallback_qs = ["Tell me about your most significant project.", "Describe a challenging bug you fixed.", "How do you design for scalability?", "Which technologies are you most comfortable with?"]
-                st.session_state.questions = fallback_qs
-                st.session_state.stage = 'asking_question'
+        try:
+            # 1. ONLY read the file and save the text
+            resume_text = read_resume(uploaded_file)
+            if not resume_text:
+                st.error("Could not extract text from the resume. Please try another file.")
+                st.session_state.stage = 'start'
+            else:
+                # 2. Save text to session state and move to a NEW stage
+                st.session_state.resume_text = resume_text
+                st.session_state.stage = 'processing_resume' # <-- NEW STAGE
                 st.rerun()
+        except Exception as e:
+            st.error(f"Error reading PDF: {e}")
+            st.session_state.stage = 'start'
+
+# --- NEW STAGE 0.5: Process Resume (runs *after* file upload) ---
+elif st.session_state.stage == 'processing_resume':
+    with st.spinner("Analyzing your resume... This may take a moment."):
+        try:
+            resume_text = st.session_state.resume_text
+            
+            # 1. Generate Questions
+            st.session_state.questions = generate_questions_from_resume(resume_text, gen_q_model)
+            if not st.session_state.questions:
+                st.warning("No AI-generated questions returned; using fallback questions.")
+                st.session_state.questions = generate_questions_from_resume(resume_text, None)
+
+            # 2. Get AI Introduction
+            intro_output = get_introduction(intro_model)
+            st.session_state.current_question = getattr(intro_output, 'question', "Can you introduce yourself?")
+
+            # 3. Move to next stage and display intro
+            st.session_state.stage = 'awaiting_intro'
+            
+            text_to_speech_and_display(getattr(intro_output, 'intro', "Hello, I'm Interviewer.AI. Please introduce yourself."))
+            text_to_speech_and_display(getattr(intro_output, 'question', "Can you introduce yourself?"))
+            
+            # Clean up the resume text from session state
+            del st.session_state.resume_text
+            st.rerun()
+        
+        except Exception as e:
+            # NOW we will catch the REAL error if it's from Python
+            st.error(f"An error occurred during AI processing: {e}")
+            st.info("This might be a Google API error. Check your key or quota.")
+            st.session_state.stage = 'start'
 
 # --- Main Interview Area (Stages > 0) ---
-if st.session_state.stage != 'start':
+if st.session_state.stage not in ['start', 'processing_resume']:
     
     # --- Chat History Display ---
     st.subheader("Interview Transcript")