hope

app.py

@@ -139,7 +139,7 @@ def main():
             count = state if isinstance(state, int) else state.value
             return f" Target image: **{class_names[count]}**"
 
-        def update_random_img_id(img_ids):
+        def update_random_img_id(img_ids, user_state):
             count = user_state if isinstance(user_state, int) else user_state.value
             random_img_id = get_random_image_id(count, data_dir)
             img_ids.append(random_img_id)
@@ -254,7 +254,7 @@ def main():
             outputs=answers
         ).then(
             update_random_img_id,
-            inputs=img_ids,
+            inputs=[img_ids, user_state],
             outputs=img_ids
         ).then(
             update_img_label,