updates

app.py

@@ -73,7 +73,7 @@ def get_next_image(session_data):
         available_images = [
             img for img in image_files
             if img not in annotated_images or
-               ("val" in img and annotated_images[img] < 5) or
+               ("val" in img and annotated_images[img] < 2) or
                ("val" not in img and annotated_images[img] == 0)
         ]
 
@@ -81,6 +81,8 @@ def get_next_image(session_data):
         if session_data["current_image"] is None and available_images:
             # Assign a new random image to the user
             session_data["current_image"] = random.choice(available_images)
+            print("Current image_id:", session_data["current_image"])  # Print the current image_id
+
     return os.path.join(image_folder, session_data["current_image"]) if session_data["current_image"] else None
 
 
@@ -139,7 +141,7 @@ def initialize_interface(session_data):
     next_image = get_next_image(session_data)
     if next_image:
         next_caption = get_caption_for_image_id(os.path.basename(next_image))  # Retrieve caption for initial image
-        print(next_caption)
+        print("Initial image_id:", os.path.basename(next_image))  # Print the initial image_id
         return gr.update(value=next_image), gr.update(value=next_caption or "")
     else:
         return gr.update(visible=False), gr.update(value="All images have been annotated!")