Reset Start VIVA button when searching new topic

app.py

@@ -785,18 +785,20 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
     def handle_query(query):
         """Handle learning mode query and store topic/image."""
         img, info, error = process_anatomy_query(query)
-        return img, info, error, query, img  # Return topic and image for VIVA mode
+        # Reset Start VIVA button
+        viva_btn_update = gr.update(value="🎯 Start VIVA Training", interactive=True)
+        return img, info, error, query, img, viva_btn_update  # Return topic, image, and button update
     
     submit_btn.click(
         fn=handle_query,
         inputs=[query_input],
-        outputs=[image_output, info_output, error_output, current_topic, current_image_state]
+        outputs=[image_output, info_output, error_output, current_topic, current_image_state, start_viva_btn]
     )
     
     query_input.submit(
         fn=handle_query,
         inputs=[query_input],
-        outputs=[image_output, info_output, error_output, current_topic, current_image_state]
+        outputs=[image_output, info_output, error_output, current_topic, current_image_state, start_viva_btn]
     )
     
     # Start VIVA Mode