updates

app.py

@@ -7,6 +7,8 @@ from huggingface_hub import login
 import json
 import re
 
+reset = True
+
 # Authenticate with Hugging Face
 token = os.getenv("HUGGINGFACE_TOKEN")
 if token:
@@ -52,6 +54,8 @@ try:
             annotation_counts[image_id] = count
 
     print("Annotation counts:", annotation_counts)
+
+
 except Exception as e:
     print(f"Error loading dataset: {e}")
     # Create an empty dataset if it doesn't exist
@@ -102,9 +106,12 @@ def get_next_image(session_data):
         # Available images filter
         available_images = [
             img for img in image_files
+            match = re.search(r'_(\d+)\.', img)
+            image_id = match.group(1).lstrip('0')  # Remove leading zeros
+
             if img not in annotation_counts or
-               (split.get(img, 0) == "dev" and annotation_counts.get(img, 0) < 2) or
-               (split.get(img, 0) == "train" and annotation_counts.get(img, 0) == 0)
+               (image_id in results and annotation_counts.get(img, 0) < 2) or
+               (image_id not in results and annotation_counts.get(img, 0) == 0)
         ]
 
         print("Available images:", available_images)  # Debugging line