Update app.py

app.py

@@ -91,8 +91,8 @@ if uploaded_file:
         with col1:
             if st.button("Oui"):
                 # Enregistrer les pixels de l'image générée dans l'ensemble "oe"
-                new_pixels = list(generated_image.getdata())
-                oe.update(map(tuple, new_pixels))  # Ajouter les nouveaux pixels à l'ensemble "oe"
+                new_pixels = set(map(tuple, np.array(generated_image.getdata())))  # Récupérer et convertir les pixels
+                oe.update(new_pixels)  # Ajouter les nouveaux pixels à l'ensemble "oe"
                 st.success("Image validée ! Les pixels sont enregistrés.")
 
                 # Réinitialiser l'image générée après validation
@@ -101,8 +101,8 @@ if uploaded_file:
         with col2:
             if st.button("Non"):
                 # Enregistrer les pixels de l'image générée dans l'ensemble "nn"
-                new_pixels = list(generated_image.getdata())
-                nn.update(map(tuple, new_pixels))  # Ajouter les nouveaux pixels à l'ensemble "nn"
+                new_pixels = set(map(tuple, np.array(generated_image.getdata())))  # Récupérer et convertir les pixels
+                nn.update(new_pixels)  # Ajouter les nouveaux pixels à l'ensemble "nn"
                 st.error("Image rejetée ! Les pixels sont enregistrés.")
 
                 # Réinitialiser l'image générée après rejet