Upload app.py

app.py

@@ -17,7 +17,7 @@ import numpy as np
 # -------------------------
 IMAGE_DIR = "images"         # Folder containing images
 NUM_PAIRS = 10               # Total number of pairs to be assessed
-# RESULTS_FILE = "results.csv" # CSV file for saving responses
+RESULTS_FILE = "results.csv" # CSV file for saving responses
 
 # -------------------------
 # Helper functions
@@ -32,6 +32,9 @@ def load_image_pair(index):
     pred_path = os.path.join(IMAGE_DIR, f"{idx_str}_gen0.png")
     return gt_path, pred_path
 
+def form_callback(name, imgID, correct):    
+    with open(RESULTS_FILE, 'a+') as f:    #Append & read mode
+        f.write(f"{name},{imgID},{correct}\n")
 
 
 # -------------------------
@@ -109,9 +112,11 @@ if st.session_state.page == "evaluation":
         
         st.success(f"Number of correct answers: {nb_correct}/{NUM_PAIRS}")
         
-        d = {'ID_img': st.session_state.list_pair_ID, 'Correct': correct_guess}
-        df = pd.DataFrame(data=d)
-        df.to_csv(st.session_state.user_name+'.csv')
+        # d = {'ID_img': st.session_state.list_pair_ID, 'Correct': correct_guess}
+        # df = pd.DataFrame(data=d)
+        # df.to_csv(st.session_state.user_name+'.csv')
+        
+        form_callback(st.session_state.user_name, st.session_state.list_pair_ID, correct_guess)
         
         st.stop()