Update app.py

app.py

@@ -42,7 +42,11 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 def set_type(choice, user_profile):
     user_profile.update({
         "interview_type": choice,
-        "state": "idle"
+        "state": "idle",
+        "questions": [],
+        "current_q": 0,
+        "user_answers": [],
+        "field": ""
     })
     return "Great! What’s your background and what field/role are you aiming for?", user_profile
 
@@ -53,71 +57,72 @@ def save_background(info, user_profile):
 
 # ---- Main Respond Logic ----
 def respond(message, chat_history, user_profile):
-    message = message.strip().lower()
+    msg = message.strip().lower()
+    state = user_profile.get("state", "idle")
 
-    # Safety: Ensure interview type + field are set
+    # Safety: ensure setup
     if not user_profile.get("interview_type") or not user_profile.get("field"):
         chat_history.append((message, "Please finish steps 1 and 2 first."))
         return chat_history
 
-    state = user_profile.get("state", "idle")
-
-    # Handle start
-    if message == 'start' and state == 'idle':
-        interview_type = user_profile['interview_type']
-        selected_questions = questions_by_type.get(interview_type, [])
+    # START command
+    if msg == "start" and state == "idle":
+        selected_questions = questions_by_type.get(user_profile['interview_type'], [])
 
         if not selected_questions:
             chat_history.append((message, "No questions found for this interview type."))
             return chat_history
 
-        user_profile.update({
-            'questions': selected_questions,
-            'current_q': 0,
-            'user_answers': [],
-            'state': 'running'
-        })
+        user_profile["questions"] = selected_questions
+        user_profile["current_q"] = 0
+        user_profile["user_answers"] = []
+        user_profile["state"] = "intro"
 
-        intro_msg = f"Great, let's begin your {interview_type} interview for the {user_profile['field']} role.\nI’ll be asking you a few questions to assess your knowledge. Take your time!"
-        first_q = selected_questions[0]['question']
+        intro_msg = f"Great, let's begin your {user_profile['interview_type']} interview for the {user_profile['field']} role.\nI'll be asking you several questions to assess your knowledge. Feel free to take your time answering."
         chat_history.append((message, intro_msg))
-        chat_history.append(("", f"First question: {first_q}"))
+        chat_history.append(("", "Type 'next' when you're ready to begin."))
         return chat_history
 
-    # Handle stop
-    if message == 'stop' and state == 'running':
+    # INTRO → waiting for user to say "next"
+    if msg == "next" and state == "intro":
+        user_profile["state"] = "interviewing"
+        first_question = user_profile['questions'][0]['question']
+        chat_history.append((message, f"First question: {first_question}"))
+        return chat_history
+
+    # STOP anytime during interviewing
+    if msg == "stop" and state in ["intro", "interviewing"]:
         user_profile["state"] = "stopped"
-        chat_history.append((message, "Interview stopped. Type 'start' to begin again or 'feedback' if you'd like an analysis (if available)."))
+        chat_history.append((message, "Interview stopped. You may type 'start' to begin again."))
         return chat_history
 
-    # Handle feedback
-    if message == 'feedback':
-        if state != 'completed':
+    # FEEDBACK command
+    if msg == "feedback":
+        if state != "completed":
             chat_history.append((message, "Feedback is only available after completing the interview."))
             return chat_history
-
         feedback = generate_feedback(user_profile)
         chat_history.append((message, feedback))
         return chat_history
 
-    # Handle interview question flow
-    if state == 'running':
-        q_index = user_profile['current_q']
-        questions = user_profile['questions']
+    # INTERVIEWING STATE LOGIC
+    if state == "interviewing":
+        q_index = user_profile["current_q"]
+        questions = user_profile["questions"]
 
         if q_index < len(questions):
-            user_profile['user_answers'].append(message)
-            user_profile['current_q'] += 1
+            user_profile["user_answers"].append(message)
+            user_profile["current_q"] += 1
 
-            if user_profile['current_q'] < len(questions):
-                next_question = questions[user_profile['current_q']]['question']
-                chat_history.append((message, f"Next question: {next_question}"))
+            if user_profile["current_q"] < len(questions):
+                next_q = questions[user_profile["current_q"]]["question"]
+                chat_history.append((message, f"Next question: {next_q}"))
             else:
-                user_profile['state'] = 'completed'
-                chat_history.append((message, "✅ Interview complete! You can type 'feedback' to receive an analysis."))
+                user_profile["state"] = "completed"
+                chat_history.append((message, "✅ Interview complete! Type 'feedback' if you'd like a performance analysis."))
             return chat_history
 
-    # Handle idle/stopped fallback small talk using LLM
+    # ANY OTHER MESSAGE (idle, stopped, etc) → small talk fallback using LLM
     messages = [
         {"role": "system", "content": f"You are a professional interviewer conducting a {user_profile['interview_type']} interview for a candidate in the {user_profile['field']} field."}
     ]
@@ -154,7 +159,10 @@ def generate_feedback(user_profile):
 
 # ---- Gradio UI ----
 with gr.Blocks() as demo:
-    user_profile = gr.State({"interview_type": "", "field": "", "state": "idle"})
+    user_profile = gr.State({
+        "interview_type": "", "field": "", "state": "idle",
+        "questions": [], "current_q": 0, "user_answers": []
+    })
     chat_history = gr.State([])
 
     gr.Markdown("# 🎤 Welcome to Intervu")