Update app.py

app.py

@@ -49,10 +49,10 @@ def next_question():
         full_text = f"{context} {question}"
         question_audio = speak_text(full_text)
         current_question += 1
-        return gr.update(value=question_audio, visible=True), gr.update(visible=True), questions[current_question-1]["label"], gr.update(visible=True), gr.update(visible=False)
+        return gr.update(value=question_audio, visible=True), gr.update(visible=True), questions[current_question-1]["label"], gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
     else:
         final_results = evaluate_responses()
-        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value=final_results, visible=True)
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(value=final_results, visible=True), gr.update(visible=True)
 
 def save_response(audio):
     transcription = transcribe_audio(audio)
@@ -66,6 +66,12 @@ def evaluate_responses():
         result += f"<b>Q:</b> {question['question']}<br><b>Your Answer:</b> {user_response}<br><br>"
     return result
 
+def restart():
+    global current_question, responses
+    current_question = 0
+    responses = []
+    return load_first_question()
+
 with gr.Blocks() as demo:
     gr.Markdown("### Interactive Questions")
 
@@ -73,14 +79,16 @@ with gr.Blocks() as demo:
     audio_input = gr.Audio(type="filepath", label="Your answer", visible=True)
     transcription_output = gr.Textbox(label="Transcription", visible=True)
     btn_next = gr.Button("Next", visible=True)
+    btn_restart = gr.Button("Restart", visible=False)
     final_output = gr.HTML(visible=False)
 
     def load_first_question():
         return next_question()
 
-    demo.load(load_first_question, outputs=[question_audio, audio_input, transcription_output, btn_next, final_output])
+    demo.load(load_first_question, outputs=[question_audio, audio_input, transcription_output, btn_next, final_output, btn_restart])
 
-    btn_next.click(next_question, outputs=[question_audio, audio_input, transcription_output, btn_next, final_output])
+    btn_next.click(next_question, outputs=[question_audio, audio_input, transcription_output, btn_next, final_output, btn_restart])
     audio_input.change(save_response, inputs=audio_input, outputs=transcription_output)
+    btn_restart.click(restart, outputs=[question_audio, audio_input, transcription_output, btn_next, final_output, btn_restart])
 
     demo.launch()