Update app.py

app.py

@@ -436,15 +436,16 @@ def submit_next_image(
             s1, s2, s3, s4, s5,
             q1_notes
         )
-    except gr.Error:
-        return (
-            user_seen, idx,
-            gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
-            gr.update(), 
-            gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
-            gr.update(),
-        )
-
+    except gr.Error as e:
+        raise e
+        # return (
+        #     user_seen, idx,
+        #     gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
+        #     gr.update(), 
+        #     gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),
+        #     gr.update(),
+        # )
+    
     left_after = user_left_count(progress[uid]["seen"], samples)
     target = user_target_count(samples)
 
@@ -454,26 +455,26 @@ def submit_next_image(
             f"Click **Exit** to close this session."
         )
         return (
-            progress[uid]["seen"], idx,
-            gr.update(), gr.update(), gr.update(), gr.update(), gr.update(),       
-            gr.update(),                                               
-            gr.update(value=status),                                   
-            gr.update(value=""),                                       
-            gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5), 
-            gr.update(value=None),                                     
-        )
+                    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)
     if idx_next == -1:
+        return (updated_seen, -1, None, None, None, None, None, "No more images.", "", 5, 5, 5, 5, 5, None)
          # Fallback if no images left
-         return (
-            progress[uid]["seen"], -1,
-            None, None, None, None, None,
-            gr.update(value="No more images."),
-            gr.update(value=""),
-            gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5),
-            gr.update(value=None),
-         )
+         # return (
+         #    progress[uid]["seen"], -1,
+         #    None, None, None, None, None,
+         #    gr.update(value="No more images."),
+         #    gr.update(value=""),
+         #    gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5),
+         #    gr.update(value=None),
+         # )
 
     sample_next = samples[idx_next]
 
@@ -483,17 +484,29 @@ def submit_next_image(
     #     f"Next sample: {sample_next.sample_id}"
     # )
     return (
-        progress[uid]["seen"], idx_next,
+        updated_seen, idx_next,
         load_image(sample_next.masked_gt_path),
         load_image(sample_next.unmasked_gt_path),
         load_image(sample_next.sr_path),
         load_image(sample_next.original_path),
         load_image(sample_next.image_5_path),
-        gr.update(value=status),
-        gr.update(value=""),                                          
-        gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5),
-        gr.update(value=None),                                        
+        gr.update(value=""),
+        gr.update(value=""),
+        5, 5, 5, 5, 5,
+        gr.update(value=None)
     )
+    # return (
+    #     progress[uid]["seen"], idx_next,
+    #     load_image(sample_next.masked_gt_path),
+    #     load_image(sample_next.unmasked_gt_path),
+    #     load_image(sample_next.sr_path),
+    #     load_image(sample_next.original_path),
+    #     load_image(sample_next.image_5_path),
+    #     gr.update(value=status),
+    #     gr.update(value=""),                                          
+    #     gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5), gr.update(value=5),
+    #     gr.update(value=None),                                        
+    # )
 
 
 def to_thanks(name: str, user_seen: List[str], samples: List[Sample]):