Update app.py

app.py

@@ -9,10 +9,11 @@ Select an exam from the dropdown menu below and start testing your skills. You c
 
 # --- FUNCTION DEFINITIONS ---
 
-def start_exam(exam_choice, audio_enabled, selected_questions_state):
-    """Starts the exam by selecting questions, setting up UI."""
-    selected_questions = select_exam_vce(exam_choice)  # Fetch new questions specific to the exam choice
-    selected_questions_state = selected_questions  # Store the selected questions in the state
+def start_exam(audio_enabled, selected_exam_state, selected_questions_state):
+    """Starts the exam by selecting questions based on the selected exam, setting up UI."""
+    selected_exam = selected_exam_state  # Get the user-selected exam
+    selected_questions = select_exam_vce(selected_exam)  # Fetch new questions for the selected exam
+    selected_questions_state = selected_questions  # Store the selected questions in the session state
 
     question, options, audio_path = display_question(0, audio_enabled, selected_questions_state)
     return (
@@ -36,19 +37,6 @@ def display_question(index, audio_enabled, selected_questions_state):
     audio_path = text_to_speech(question_text_ + " " + " ".join(choices_options)) if audio_enabled else None
     return question_text, choices_options, audio_path
 
-def show_explanation(index, selected_questions_state):
-    """Shows the explanation for the current question and hides previous results."""
-    selected_questions = selected_questions_state
-    if 0 <= index < len(selected_questions):
-        explanation = selected_questions[index].get('explanation', 'No explanation available for this question.')
-        return (
-            f"**Explanation:** {explanation}", 
-            gr.update(visible=True),  # Show explanation_text
-            gr.update(visible=True)   # Show result_text
-        )
-    else:
-        return "No explanation available for this question.", gr.update(visible=False), gr.update(visible=False)
-
 def check_answer(index, answer, selected_questions_state):
     """Checks the given answer against the correct answer."""
     selected_questions = selected_questions_state
@@ -65,22 +53,19 @@ def update_question(index, audio_enabled, selected_questions_state):
 
 def handle_answer(index, answer, audio_enabled, current_audio, selected_questions_state):
     """Handles answer submission, provides feedback, and generates audio."""
-    # Handle the case when no answer is selected
     if answer is None:
         return "Please select an option before submitting.", None, None
-    
-    # Stop the current question audio before playing the answer audio
+
     stop_audio = True if current_audio else False
     result = check_answer(index, answer, selected_questions_state)
     answer_audio_path = text_to_speech(result) if audio_enabled else None
     return result, answer_audio_path, stop_audio
 
-
 def handle_next(index, audio_enabled, selected_questions_state):
     """Moves to the next question and updates the UI."""
     new_index = min(index + 1, len(selected_questions_state) - 1)
     question, options, new_index, audio_path = update_question(new_index, audio_enabled, selected_questions_state)
-    return question, options, new_index, "", audio_path, gr.update(visible=False)  # Hide explanation
+    return question, options, new_index, "", audio_path, gr.update(visible=False)
 
 def handle_previous(index, audio_enabled, selected_questions_state):
     """Moves to the previous question and updates the UI."""
@@ -91,15 +76,24 @@ def handle_previous(index, audio_enabled, selected_questions_state):
 def return_home():
     """Returns to the home screen."""
     return (
-        # Show start screen elements
         gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),
-        gr.update(visible=True), # Show the audio_checkbox
-        # Hide quiz elements
+        gr.update(visible=True),  # Show the audio_checkbox
         gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
         gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), "", "", gr.update(visible=False), gr.update(visible=False),
-        gr.update(visible=False) # Hide explain button
+        gr.update(visible=False)  # Hide explain button
     )
-
+def show_explanation(index, selected_questions_state):
+    """Shows the explanation for the current question and hides previous results."""
+    selected_questions = selected_questions_state
+    if 0 <= index < len(selected_questions):
+        explanation = selected_questions[index].get('explanation', 'No explanation available for this question.')
+        return (
+            f"**Explanation:** {explanation}", 
+            gr.update(visible=True),  # Show explanation_text
+            gr.update(visible=True)   # Show result_text
+        )
+    else:
+        return "No explanation available for this question.", gr.update(visible=False), gr.update(visible=False)
 with gr.Blocks() as demo:
     # Home page elements
     title = gr.Markdown(value="**AWS Exam Simulator (Quiz)**")
@@ -110,8 +104,9 @@ with gr.Blocks() as demo:
 
     # Quiz elements (initially hidden)
     question_state = gr.State(0)
-    current_audio_state = gr.State(None)  # State to track the current audio playing
-    selected_questions_state = gr.State([])  # State to store selected questions for the user session
+    current_audio_state = gr.State(None)
+    selected_questions_state = gr.State([])  # Store the selected questions for each user session
+    selected_exam_state = gr.State("")  # State to store the selected exam for each user session
     question_text = gr.Markdown(visible=False, elem_id="question-text")
     choices = gr.Radio(visible=False, label="Options")
     result_text = gr.Markdown(visible=True)
@@ -150,16 +145,19 @@ with gr.Blocks() as demo:
     with gr.Row():
         gr.Column([home_button])
     
+    # Store the selected exam when chosen
+    exam_selector.change(fn=lambda exam: exam, inputs=[exam_selector], outputs=[selected_exam_state])
+
     # Connect the start button to start the exam
     start_button.click(
         fn=start_exam, 
-        inputs=[exam_selector, audio_checkbox, selected_questions_state], 
+        inputs=[audio_checkbox, selected_exam_state, selected_questions_state], 
         outputs=[
             title, description, exam_selector, start_button,
-            audio_checkbox,  # Ensure the checkbox visibility is updated
+            audio_checkbox,
             question_text, question_text, choices, answer_button, 
             next_button, prev_button, home_button, question_state, result_text, question_audio,
-            explain_button, selected_questions_state, current_audio_state  # Add selected_questions_state to store per-user questions
+            explain_button, selected_questions_state, current_audio_state
         ]
     )
     
@@ -172,10 +170,9 @@ with gr.Blocks() as demo:
 
     home_button.click(fn=return_home, inputs=None, outputs=[
         title, description, exam_selector, start_button,
-        audio_checkbox,  # Ensure the checkbox visibility is updated
+        audio_checkbox,
         question_text, question_text, choices, answer_button, 
         next_button, prev_button, home_button, question_state, result_text, explanation_text, explain_button
     ])
 
-
 demo.queue(max_size=100).launch(debug=True, max_threads=75)