Update app.py

app.py

@@ -446,26 +446,40 @@ def submit_next_image(
         #     gr.update(),
         # )
     
-    left_after = user_left_count(progress[uid]["seen"], samples)
-    target = user_target_count(samples)
+    # left_after = user_left_count(progress[uid]["seen"], samples)
+    # target = user_target_count(samples)
+    updated_seen = progress[uid]["seen"]
 
-    if left_after == 0:
+    if len(updated_seen) >= TARGET_PER_PERSON:
         status = (
             f"Saved! You’ve completed all {target} images. 🎉 "
             f"Click **Exit** to close this session."
         )
         return (
-                    updated_seen, -1,
-                    None, None, None, None, None,  # Return None to avoid image load errors
-                    gr.update(value="Target reached! Processing..."), 
-                    gr.update(value=""),
-                    5, 5, 5, 5, 5, 
-                    gr.update(value=None)
-                )
-
-    idx_next = pick_next_index(progress[uid]["seen"], samples)
+            updated_seen, -1, 
+            None, None, None, None, None, # No more images to load
+            gr.update(value="Target reached!"), # Status
+            gr.update(value=""), # Clear notes
+            5, 5, 5, 5, 5 # Reset sliders
+        )
+    
+    # if left_after == 0:
+    #     status = (
+    #         f"Saved! You’ve completed all {target} images. 🎉 "
+    #         f"Click **Exit** to close this session."
+    #     )
+    #     return (
+    #                 updated_seen, -1,
+    #                 None, None, None, None, None,  # Return None to avoid image load errors
+    #                 gr.update(value="Target reached! Processing..."), 
+    #                 gr.update(value=""),
+    #                 5, 5, 5, 5, 5, 
+    #                 gr.update(value=None)
+    #             )
+
+    idx_next = pick_next_index(updated_seen, samples)
     if idx_next == -1:
-        return (updated_seen, -1, None, None, None, None, None, "No more images.", "", 5, 5, 5, 5, 5, None)
+        return (updated_seen, -1, None, None, None, None, None, "No more images.", "", 5, 5, 5, 5, 5)
          # Fallback if no images left
          # return (
          #    progress[uid]["seen"], -1,
@@ -493,7 +507,7 @@ def submit_next_image(
         gr.update(value=""),
         gr.update(value=""),
         5, 5, 5, 5, 5,
-        gr.update(value=None)
+        
     )
     # return (
     #     progress[uid]["seen"], idx_next,