Update app.py

app.py

@@ -3,9 +3,9 @@ import numpy as np
 from PIL import Image
 import random
 
-# Listes pour stocker les pixels validés et non validés
-oe = []
-nn = []
+# Ensembles pour stocker les pixels validés et non validés
+oe = set()  # Pixels validés
+nn = set()  # Pixels non validés
 
 # Fonction pour traiter et redimensionner l'image importée
 def process_uploaded_image(uploaded_file):
@@ -13,7 +13,7 @@ def process_uploaded_image(uploaded_file):
     pixel_data = np.array(image)
     return pixel_data
 
-# Fonction pour générer une image en fonction des pixels voisins
+# Fonction pour générer une image en fonction des pixels voisins avec variation
 def generate_image_from_neighbors(base_pixels):
     height, width, _ = base_pixels.shape
     generated_image = np.zeros((height, width, 3), dtype=np.uint8)
@@ -27,7 +27,7 @@ def generate_image_from_neighbors(base_pixels):
                 for dj in range(-1, 2):  # -1, 0, 1
                     if 0 <= i + di < height and 0 <= j + dj < width:
                         surrounding_pixels.append(base_pixels[i + di, j + dj])
-            
+
             # Calculer la moyenne des pixels environnants
             surrounding_pixels = np.array(surrounding_pixels)
             new_pixel_color = np.mean(surrounding_pixels, axis=0)
@@ -35,6 +35,12 @@ def generate_image_from_neighbors(base_pixels):
             # Ajouter une petite variation aléatoire pour générer de l'unicité
             variation = np.array([random.randint(-20, 20) for _ in range(3)])
             new_pixel_color = np.clip(new_pixel_color + variation, 0, 255).astype(np.uint8)
+
+            # Vérification pour changer aléatoirement le pixel pour plus d'unicité
+            if random.random() < 0.2:  # 20% de chance de changer le pixel
+                new_pixel_color += np.array([random.randint(-50, 50) for _ in range(3)])
+                new_pixel_color = np.clip(new_pixel_color, 0, 255).astype(np.uint8)
+
             generated_image[i, j] = new_pixel_color
 
     return Image.fromarray(generated_image)
@@ -64,10 +70,9 @@ if uploaded_file:
         col1, col2 = st.columns(2)  # Créez deux colonnes pour les boutons
         with col1:
             if st.button("Oui"):
-                # Enregistrer les pixels de l'image générée dans la liste "oe"
+                # Enregistrer les pixels de l'image générée dans l'ensemble "oe"
                 new_pixels = list(generated_image.getdata())
-                oe.extend(new_pixels)  # Ajouter les nouveaux pixels à la liste "oe"
-                oe = list(set(map(tuple, oe)))  # Enlever les doublons
+                oe.update(map(tuple, new_pixels))  # Ajouter les nouveaux pixels à l'ensemble "oe"
                 st.success("Image validée!")
 
                 # Réinitialiser l'image générée après validation
@@ -75,21 +80,20 @@ if uploaded_file:
 
         with col2:
             if st.button("Non"):
-                # Enregistrer les pixels de l'image générée dans la liste "nn"
+                # Enregistrer les pixels de l'image générée dans l'ensemble "nn"
                 new_pixels = list(generated_image.getdata())
-                nn.extend(new_pixels)  # Ajouter les nouveaux pixels à la liste "nn"
-                nn = list(set(map(tuple, nn)))  # Enlever les doublons
+                nn.update(map(tuple, new_pixels))  # Ajouter les nouveaux pixels à l'ensemble "nn"
                 st.error("Image rejetée!")
 
                 # Réinitialiser l'image générée après rejet
                 generated_image = None
 
-    # Affichage des listes de pixels validés et non validés
+    # Affichage des ensembles de pixels validés et non validés
     st.write("Pixels validés (oe):", len(oe), "pixels enregistrés.")
     st.write("Pixels non validés (nn):", len(nn), "pixels enregistrés.")
 
-# Option pour réinitialiser les listes
-if st.button("Réinitialiser les listes"):
+# Option pour réinitialiser les ensembles
+if st.button("Réinitialiser les ensembles"):
     oe.clear()
     nn.clear()
-    st.success("Les listes de pixels ont été réinitialisées.")
+    st.success("Les ensembles de pixels ont été réinitialisés.")