updates

app.py

@@ -149,6 +149,7 @@ def get_next_image(session_data):
 
         # print("Available images:", available_images)  # Debugging line
 
+        random.shuffle(available_images)
 
         # Check if the user already has an image
         if session_data["current_image"] is None and available_images:
@@ -201,9 +202,9 @@ def save_annotation(caption, session_data):
         dataset.push_to_hub(dataset_name)
         print("Pushed updated dataset")
 
-        # Clear user's current image if the validation image has been annotated twice
-        if (split == "train" and annotation_count > 1) or (split == "dev" and annotation_count > 2):
-            session_data["current_image"] = None
+        # # Clear user's current image if the validation image has been annotated twice
+        # if (split == "train" and annotation_count > 1) or (split == "dev" and annotation_count > 2):
+        #     session_data["current_image"] = None
 
     # Fetch the next image
     next_image = get_next_image(session_data)