Randomly shuffle versus images

app.py

@@ -8,6 +8,7 @@ import plotly.graph_objects as go
 import hashlib
 from PIL import Image
 import json
+import random
 
 os.environ["PYTHONHASHSEED"] = "42"
 
@@ -291,17 +292,6 @@ country_to_center_coords = {
     "United States": (39.7837304, -100.445882),
 }
 
-INTIAL_VERSUS_IMAGE = "versus_images/Europe_Germany_49.069183_10.319444_im2gps3k.jpg"
-INITAL_VERSUS_STATE = {
-    "image": INTIAL_VERSUS_IMAGE,
-    "continent": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[0],
-    "country": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[1],
-    "lat": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[2],
-    "lon": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[3],
-    "score": {"HUMAN": 0, "AI": 0},
-    "idx": 0,
-}
-
 
 def predict(input_img):
     inputs = processor(
@@ -504,11 +494,7 @@ def get_example_images(dir):
 
 
 def next_versus_image(versus_state):
-    images = get_example_images("versus_images")
-    versus_state["idx"] += 1
-    if versus_state["idx"] > len(images):
-        versus_state["idx"] = 0
-    versus_image = images[versus_state["idx"]]
+    versus_image = random.sample(versus_state["images"], 1)[0]
     versus_state["continent"] = versus_image.split("/")[-1].split("_")[0]
     versus_state["country"] = versus_image.split("/")[-1].split("_")[1]
     versus_state["lat"] = versus_image.split("/")[-1].split("_")[2]
@@ -528,6 +514,20 @@ for img_path in example_images:
         "lon": img_path.split("/")[-1].split("_")[3],
     }
 
+def set_up_intial_state():
+    INTIAL_VERSUS_IMAGE = "versus_images/Europe_Germany_49.069183_10.319444_im2gps3k.jpg"
+    INITAL_VERSUS_STATE = {
+        "image": INTIAL_VERSUS_IMAGE,
+        "continent": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[0],
+        "country": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[1],
+        "lat": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[2],
+        "lon": INTIAL_VERSUS_IMAGE.split("/")[-1].split("_")[3],
+        "score": {"HUMAN": 0, "AI": 0},
+        "images": get_example_images("versus_images")
+    }
+    return INITAL_VERSUS_STATE
+
+
 demo = gr.Blocks(title="Thesis Demo")
 with demo:
     gr.HTML(
@@ -573,10 +573,10 @@ with demo:
     )
 
     with gr.Tab("Versus Mode"):
-        versus_state = gr.State(value=INITAL_VERSUS_STATE)
+        versus_state = gr.State(value=set_up_intial_state())
         with gr.Row():
             with gr.Column():
-                versus_image = gr.Image(INITAL_VERSUS_STATE["image"], interactive=False, show_download_button=False, show_share_button=False, show_fullscreen_button=True)
+                versus_image = gr.Image(versus_state.value["image"], interactive=False, show_download_button=False, show_share_button=False, show_fullscreen_button=True)
                 continent_selection = gr.Radio(
                     continents,
                     label="Continents",