updates

app.py

@@ -104,16 +104,22 @@ def get_caption_for_image_id(image_path):
 def get_next_image(session_data):
     with lock:
         # Available images filter
-        available_images = [
-            img for img in image_files
-            print(img)
-            match = re.search(r'_(\d+)\.', img)
-            image_id = match.group(1).lstrip('0')  # Remove leading zeros
+        available_images = []
 
-            if img not in annotation_counts or
-               (image_id in results and annotation_counts.get(img, 0) < 2) or
-               (image_id not in results and annotation_counts.get(img, 0) == 0)
-        ]
+        # Iterate over each image file to apply the filtering logic
+        for img in image_files:
+            print(img)  # Debugging line to print each image filename
+
+            # Match and extract the image_id from the filename
+            match = re.search(r'_(\d+)\.', img)
+            if match:
+                image_id = match.group(1).lstrip('0')  # Remove leading zeros
+
+                # Apply the filtering conditions
+                if (img not in annotation_counts or
+                    (image_id in results and annotation_counts.get(img, 0) < 2) or
+                    (image_id not in results and annotation_counts.get(img, 0) == 0)):
+                    available_images.append(img)
 
         print("Available images:", available_images)  # Debugging line