Update app.py

app.py

@@ -90,21 +90,26 @@ def generate_question(prompt, domain, state=None, max_attempts=10):
 
     raise RuntimeError("Failed to generate a unique question after multiple attempts.")
 
-def reset_state(domain, company):
+def reset_state(domain, company, role, year):
     return {
         "domain": domain,
         "company": company,
+        "role": role,
+        "year": year,
         "asked_questions": set(),
         "asked_subtopics": set(),
-        "conversation": []  # List of dicts: {"role": ..., "content": ...}
+        "conversation": []
     }
 
-def start_interview(domain, company):
-    state = reset_state(domain, company)
-    prompt = f"Domain: {domain}. " + (f"Company: {company}. " if company else "") + "Generate the first question:"
-    question = generate_question(prompt, domain, state)
-    state["conversation"].append({"role": "assistant", "content": question})
-    return state["conversation"], state
+def start_interview(domain, company, role, year):
+    state = reset_state(domain, company, role, year)
+    prompt = (
+        f"Domain: {domain}. "
+        f"Company: {company or 'N/A'}. "
+        f"Role: {role or 'N/A'}. "
+        f"Year: {year or 'N/A'}. "
+        "Generate the first question:"
+    )
 
 def submit_response(candidate_response, state):
     state["conversation"].append({"role": "user", "content": candidate_response})
@@ -120,6 +125,8 @@ with gr.Blocks() as demo:
     with gr.Row():
         domain_input = gr.Textbox(label="Domain")
         company_input = gr.Textbox(label="Company (Optional)")
+        role_input = gr.Textbox(label="Role (e.g., Data Analyst, MLE)", placeholder="Enter your target role")
+        year_input = gr.Textbox(label="Year (Optional)", placeholder="e.g., 2024")
 
     start_button = gr.Button("🚀 Start Interview")
     chatbot = gr.Chatbot(label="Interview Conversation", type="messages")
@@ -131,7 +138,11 @@ with gr.Blocks() as demo:
     state = gr.State({})  # Session state holder
 
     # Hook buttons to logic
-    start_button.click(start_interview, inputs=[domain_input, company_input], outputs=[chatbot, state])
+    start_button.click(
+    start_interview,
+    inputs=[domain_input, company_input, role_input, year_input],
+    outputs=[chatbot, state]
+    )
     submit_button.click(submit_response, inputs=[response_input, state], outputs=[chatbot, state]).then(
         lambda: "", inputs=[], outputs=[response_input]  # Clear textbox after submit
     )