Update app.py

app.py

@@ -81,12 +81,15 @@ def handle_answer(answer, mock_index, score, incorrect_questions):
         result = f"Your score: {score}/{len(data)}\n\n{incorrect_str}"
         return gr.update(value=result), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), mock_index, score, incorrect_questions
     
-    question, ans1, ans2, ans3, ans4, correct = display_mock_question(mock_index, score, incorrect_questions)
-    return gr.update(value=question), gr.update(visible=False), gr.update(value=ans1), gr.update(value=ans2), gr.update(value=ans3), gr.update(value=ans4), mock_index, score, incorrect_questions
+    return gr.update(), gr.update(visible=False), gr.update(), gr.update(), gr.update(), gr.update(), mock_index, score, incorrect_questions
 
 def update_question(mock_index, score, incorrect_questions):
+    data = load_mocktest()
+    if mock_index >= len(data):
+        return gr.update(), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    
     question, ans1, ans2, ans3, ans4, correct = display_mock_question(mock_index, score, incorrect_questions)
-    return gr.update(value=question), gr.update(value=ans1), gr.update(value=ans2), gr.update(value=ans3), gr.update(value=ans4)
+    return gr.update(value=question), gr.update(value=ans1, visible=True), gr.update(value=ans2, visible=True), gr.update(value=ans3, visible=True), gr.update(value=ans4, visible=True)
 
 # Function to restart the test
 def restart_test():