Upload app.py

app.py

@@ -14,6 +14,8 @@ import numpy as np
 import gspread
 from oauth2client.service_account import ServiceAccountCredentials
 from skimage.exposure import match_histograms
+import time
+from streamlit_autorefresh import st_autorefresh
 
 # -------------------------
 # Global parameters
@@ -52,9 +54,9 @@ def match_brightness(source_img, target_img):
 
 def match_histograms_pil(img_reference, img_to_adjust):
     """
-    Calque l'histogramme de `img_reference` sur `img_to_adjust`
-    (les deux images sont des objets PIL.Image).
-    Retourne une image PIL avec histogramme ajusté.
+    Layer the histogram of `img_reference` on `img_to_adjust`
+    (both images are PIL.Image objects).
+    Returns a PIL image with adjusted histogram.
     """
     # Convertir les deux images en tableaux numpy
     ref_array = np.array(img_reference)
@@ -130,6 +132,11 @@ if st.session_state.page == "evaluation":
     st.title("AI Wood Generation Evaluation")
     # st.write(f"User: **{st.session_state.user_name}**")
     
+    if "start_time" not in st.session_state or st.session_state.page_changed:
+        st.session_state.start_time = time.time()
+        st.session_state.page_changed = False
+    st_autorefresh(interval=1000, key=f"timer_{st.session_state.current_index}")
+    
     # If all pairs have been evaluated, display a message and save the results
     if st.session_state.current_index+1 > NUM_PAIRS:
 
@@ -194,13 +201,27 @@ if st.session_state.page == "evaluation":
     # else:
     #     img1 = match_brightness(img2, img1)
     
+    elapsed = time.time() - st.session_state.start_time
+    placeholder = Image.new("L", img1.size, 128)
+    
     col1, col2 = st.columns(2)
     
-    with col1:
-        st.image(img1, caption="Image 1", use_container_width =True)
+    # with col1:
+    #     st.image(img1, caption="Image 1", use_container_width =True)
         
+    # with col2:
+    #     st.image(img2, caption="Image 2", use_container_width =True)
+    
+    with col1:
+        if elapsed < 10:
+            st.image(img1, caption="Image 1", use_container_width=True)
+        else:
+            st.image(placeholder, caption="Time’s up!", use_container_width=True)
     with col2:
-        st.image(img2, caption="Image 2", use_container_width =True)
+        if elapsed < 10:
+            st.image(img2, caption="Image 2", use_container_width=True)
+        else:
+            st.image(placeholder, caption="Time’s up!", use_container_width=True)
     
     choice = st.radio("Select the real image: ", options=["1", "2"], horizontal = True) #, index=None
 
@@ -216,6 +237,8 @@ if st.session_state.page == "evaluation":
         
         # Passer à la paire suivante
         st.session_state.current_index += 1
+        st.session_state.start_time = time.time()
+        st.session_state.page_changed = True
         st.rerun()