Upload app.py

app.py

@@ -193,10 +193,10 @@ if st.session_state.page == "evaluation":
             #     st.session_state.page = "detailed_results"
             #     st.rerun()
             
-        # if st.session_state.submitted == True:
-        #     if st.button("See detailed results"):
-        #         st.session_state.page = "detailed_results"
-        #         st.rerun()
+        if st.session_state.submitted == True:
+            if st.button("See detailed results"):
+                st.session_state.page = "detailed_results"
+                st.rerun()
             # st.stop()
             
         st.stop()
@@ -259,32 +259,34 @@ if st.session_state.page == "evaluation":
         st.rerun()
         
 
-# if st.session_state.page == "detailed_results":
-#     st.title("Detailed Results")
-
-#     for i, (pair, result) in enumerate(zip(st.session_state.list_pair_ID, st.session_state.results)):
-#         st.markdown(f"### Pair {i+1}")
-    
-#         col1, col2 = st.columns(2)
-#         with col1:
-#             st.image(pair[0][1], caption=f"Image 1 ({pair[0][0]})", use_container_width=True)
-#         with col2:
-#             st.image(pair[1][1], caption=f"Image 2 ({pair[1][0]})", use_container_width=True)
+if st.session_state.page == "detailed_results":
+    st.title("Detailed Results")
+    results = np.array(st.session_state.results) 
     
-#         # Trouver la vraie image
-#         real_index = 0 if pair[0][0] == "GT" else 1
-#         user_choice_index = 0 if result == 1 else 1  # on suppose que la prédiction est correcte si result==1
-    
-#         # Afficher les infos
-#         st.write(f"**Real image:** Image {real_index+1}")
-#         st.write(f"**Your choice:** Image {user_choice_index+1}")
+    for i in range(NUM_PAIRS):
+        pair = st.session_state.list_pair[i]
         
-#         if result:
-#             st.success("✅ Correct")
-#         else:
-#             st.error("❌ Incorrect")
+        if pair[0][0] == "GT":        
+            imgGT = Image.open(pair[0][1])
+            imgGT = imgGT.convert("L")
+            imgPred = Image.open(pair[1][1])
+            imgPred = imgPred.convert("L")
+        else:       
+            imgGT = Image.open(pair[1][1])
+            imgGT = imgGT.convert("L")
+            imgPred = Image.open(pair[0][1])
+            imgPred = imgPred.convert("L")
         
-#         st.markdown("---")
-
-
-    
+        col1, col2 = st.columns(2)
+        with col1:
+            st.image(imgGT, caption="Real", use_container_width=True)
+        with col2:
+            st.image(imgPred, caption="AI", use_container_width=True)
+        
+        result = results[i]
+        
+        if result:
+            st.success("✅ Correct")
+        else:
+            st.error("❌ Incorrect")
+        st.markdown("---")