updates

app.py

@@ -139,9 +139,7 @@ def get_next_image(session_data):
             match = re.search(r'_(\d+)\.', img)
             if match:
                 image_id_2 = match.group(1).lstrip('0')  # Remove leading zeros
-                print(image_id_2)
-                print(image_id_2 in results)
-                print(annotation_counts.get(img, 0))
+
 
                 # Apply the filtering conditions
                 if (img not in annotation_counts or
@@ -151,6 +149,7 @@ def get_next_image(session_data):
 
         # print("Available images:", available_images)  # Debugging line
 
+
         # Check if the user already has an image
         if session_data["current_image"] is None and available_images:
             # Assign a new random image to the user
@@ -203,7 +202,7 @@ def save_annotation(caption, session_data):
         print("Pushed updated dataset")
 
         # Clear user's current image if the validation image has been annotated twice
-        if ("val" not in image_id) or (annotation_count + 1 >= 2):
+        if (split == "train" and annotation_count > 1) or (split == "dev" and annotation_count > 2):
             session_data["current_image"] = None
 
     # Fetch the next image