save image id storing them into an array

app.py

@@ -7,7 +7,7 @@ import time
 from pathlib import Path
 from huggingface_hub import CommitScheduler, HfApi
 
-from src.utils import load_words, load_image_and_saliency, load_example_images, load_csv_concepts
+from src.utils import load_words, get_random_image_id, load_image_and_saliency, load_example_images, load_csv_concepts
 from src.style import css
 from src.user import UserID
 
@@ -68,8 +68,9 @@ def main():
 
         with gr.Row():
             count = user_state if isinstance(user_state, int) else user_state.value
-            images = load_image_and_saliency(count, data_dir)
-            img_ids = gr.State([images[0]])  
+            random_img_id = get_random_image_id(count, data_dir)
+            img_ids.append(random_img_id)
+            images = load_image_and_saliency(count, data_dir, random_img_id)
             target_img = gr.Image(images[1], elem_classes="main-image delay", visible=False)
             saliency_gradcam = gr.Image(images[2], elem_classes="main-image", visible=False)
             saliency_lime = gr.Image(images[3], elem_classes="main-image", visible=False)
@@ -117,16 +118,16 @@ def main():
         def update_saliencies(dropdown1, dropdown2, dropdown3, dropdown4, user_state, img_ids):
             count = user_state if isinstance(user_state, int) else user_state.value
             if count < config['dataset'][config['dataset']['name']]['n_classes']:
-                images = load_image_and_saliency(count, data_dir)
-                img_ids.append(images[0])  
+                img_id = img_ids[-1] if isinstance(img_ids[-1], int) else img_ids[-1].value
+                images = load_image_and_saliency(count, data_dir, img_id)
                 target_img = gr.Image(images[1], elem_classes="main-image", visible=True)
                 saliency_gradcam = gr.Image(images[2], elem_classes="main-image", visible=True)
                 saliency_lime = gr.Image(images[3], elem_classes="main-image", visible=True)
                 saliency_sidu = gr.Image(images[5], elem_classes="main-image", visible=True)
                 saliency_rise = gr.Image(images[4], elem_classes="main-image", visible=True)
-                return img_ids, target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
+                return target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
             else:
-                return img_ids, target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
+                return target_img, saliency_gradcam, saliency_lime, saliency_rise, saliency_sidu
 
         def update_state(state):
             count = state if isinstance(state, int) else state.value
@@ -136,6 +137,12 @@ def main():
             count = state if isinstance(state, int) else state.value
             return f" Target image: **{class_names[count]}**"
 
+        def update_random_img_id(img_ids):
+            count = user_state if isinstance(user_state, int) else user_state.value
+            random_img_id = get_random_image_id(count, data_dir)
+            img_ids.append(random_img_id)
+            return img_ids
+
         def update_buttons():
             submit_button = gr.Button("Submit", visible=False)
             continue_button = gr.Button("Continue", visible=True)
@@ -202,6 +209,7 @@ def main():
             info_to_push = {
                 "user_id": time.time(), 
                 "answer": {i: answer for i, answer in enumerate(answers)},
+                "img_ids": [img_id for img_id in img_ids],
                 "duration": duration
             }
 
@@ -212,6 +220,7 @@ def main():
                         "user_id": info_to_push["user_id"],
                         "answers": info_to_push["answer"],
                         "duration": info_to_push["duration"],
+                        "img_ids": info_to_push["img_ids"],
                         "datetime": datetime.now().isoformat()
                     }, f)
                     f.write("\n")
@@ -241,6 +250,10 @@ def main():
             add_answer,
             inputs=[dropdown1, dropdown2, dropdown3, dropdown4, answers],
             outputs=answers
+        ).then(
+            update_random_img_id,
+            inputs=img_ids,
+            outputs=img_ids
         ).then(
             update_img_label,
             inputs=user_state,

src/utils.py

@@ -7,19 +7,21 @@ import time
 
 config = yaml.safe_load(open("./config/config.yaml")) 
     
-def load_image_and_saliency(class_idx, data_dir):
+def get_random_image_id(class_idx, data_dir):
     path = os.path.join(data_dir, 'images', str(class_idx))
     images = os.listdir(path)
-    # pick a random image
-    # set random seed usiing time
-    np.random.seed(int(time.time()))
-    id = np.random.randint(0, len(images))
+    return np.random.randint(0, len(images))
+
+def load_image_and_saliency(class_idx, data_dir, img_id):
+    path = os.path.join(data_dir, 'images', str(class_idx))
+    images = os.listdir(path)
+    id = img_id
     image = os.path.join(path, images[id])
     gradcam_image = os.path.join(data_dir, 'saliency', 'gradcam', images[id])
     lime_image = os.path.join(data_dir, 'saliency', 'lime', images[id])
     sidu_image = os.path.join(data_dir, 'saliency', 'sidu', images[id])
     rise_image = os.path.join(data_dir, 'saliency', 'rise', images[id])
-    return id, image, gradcam_image, lime_image, sidu_image, rise_image
+    return image, gradcam_image, lime_image, sidu_image, rise_image
 
 def load_example_images(class_idx, data_dir, max_images=16):
     path = os.path.join(data_dir, 'images', str(class_idx))