Upload app.py

app.py

@@ -32,20 +32,18 @@ def load_image_pair(index):
     pred_path = os.path.join(IMAGE_DIR, f"{idx_str}_gen0.png")
     return gt_path, pred_path
 
-def open_image(path):
-    """Ouvre l'image avec PIL."""
-    return Image.open(path)
 
-def get_shuffled_pair(index):
-    """Charge la paire d'images et retourne une liste de tuples (label, image) dans un ordre aléatoire.
-       Le label est 'GT' pour ground truth et 'Pred' pour generated.
-    """
-    gt_path, pred_path = load_image_pair(index)
-    img_gt = open_image(gt_path)
-    img_pred = open_image(pred_path)
-    pair = [("GT", img_gt), ("Pred", img_pred)]
-    random.shuffle(pair)
-    return pair
+
+# def get_shuffled_pair(index):
+#     """Charge la paire d'images et retourne une liste de tuples (label, image) dans un ordre aléatoire.
+#        Le label est 'GT' pour ground truth et 'Pred' pour generated.
+#     """
+#     gt_path, pred_path = load_image_pair(index)
+#     img_gt = open_image(gt_path)
+#     img_pred = open_image(pred_path)
+#     pair = [("GT", img_gt), ("Pred", img_pred)]
+#     random.shuffle(pair)
+#     return pair
 
 # -------------------------
 # Navigation via st.session_state
@@ -55,13 +53,15 @@ if "page" not in st.session_state:
 if "user_name" not in st.session_state:
     st.session_state.user_name = ""
 if "current_index" not in st.session_state:
-    st.session_state.current_index = 1
+    st.session_state.current_index = 0
 if "results" not in st.session_state:
     st.session_state.results = []
+if "list_pair" not in st.session_state:
+    st.session_state.list_pair = []
+
 
     
     
-
 # -------------------------
 # Intro page
 # -------------------------
@@ -83,6 +83,17 @@ if st.session_state.page == "intro":
         st.session_state.user_name = name
         st.session_state.page = "evaluation"
         st.rerun()
+    
+    rd_ID_list = random.sample(range(1, 51), NUM_PAIRS)
+
+    for i, index in enumerate(rd_ID_list):
+        
+        gt_path, pred_path = load_image_pair(index)
+        
+        pair = [("GT", gt_path), ("Pred", pred_path)]
+        random.shuffle(pair)
+        st.session_state.list_pair.append(pair)
+    
     st.stop()
 
 
@@ -110,25 +121,30 @@ if st.session_state.page == "evaluation":
         
         st.stop()
     
-    st.write(f"Image Pair {st.session_state.current_index} of {NUM_PAIRS}")
+    st.write(f"Image Pair {st.session_state.current_index+1} of {NUM_PAIRS}")
     
     # Charger et mélanger la paire pour l'index courant
-    pair = get_shuffled_pair(st.session_state.current_index)
+    pair = st.session_state.list_pair[st.session_state.current_index]
+    
+    img1 = Image.open(pair[0][1])
+    img2 = Image.open(pair[1][1])
+    
+    
     # Affichage dans deux colonnes avec radio buttons sous chacune
     col1, col2 = st.columns(2)
     
     with col1:
-        st.image(pair[0][1], caption="Image 1", use_container_width =True)
+        st.image(img1, caption="Image 1", use_container_width =True)
         
     with col2:
-        st.image(pair[1][1], caption="Image 2", use_container_width =True)
+        st.image(img2, caption="Image 2", use_container_width =True)
     
-    choice = st.radio("", options=["1                  ", "2"], index=0, horizontal = True)
+    choice = st.radio("", options=["1", "2"], index=0, horizontal = True)
 
     
     if st.button("Next"):
         
-        if (choice == "1                  " and pair[0][0] == "GT") or (choice == "2" and pair[1][0] == "GT"):
+        if (choice == "1" and pair[0][0] == "GT") or (choice == "2" and pair[1][0] == "GT"):
             correct_guess = 1
         else:
             correct_guess = 0