Update app.py

app.py

@@ -15,29 +15,25 @@ def process_uploaded_image(uploaded_file):
     pixel_data = np.array(image)
     return pixel_data
 
-# Fonction pour générer une image basée sur les pixels environnants
-def generate_image_based_on_pixels(base_pixels, label, avoid_data=None):
+# Fonction pour générer une image avec variations basées sur les pixels environnants
+def generate_image_with_variation(base_pixels, label, avoid_data=None):
     width, height = 300, 300
     generated_image = np.zeros((height, width, 3), dtype=np.uint8)
 
-    # Parcourir les pixels pour générer l'image
     for i in range(height):
         for j in range(width):
-            # Choisir des valeurs de pixels similaires aux pixels environnants de l'image importée
-            if i > 0 and j > 0:
-                # Prendre en compte le pixel précédent, celui au-dessus et en diagonale
-                avg_pixel = (
-                    base_pixels[i-1, j] + base_pixels[i, j-1] + base_pixels[i-1, j-1]
-                ) // 3
-            else:
-                # Premier pixel arbitraire
-                avg_pixel = base_pixels[i, j]
+            # Utiliser le pixel de référence et ajouter des variations aléatoires
+            base_pixel = base_pixels[i, j]
+            
+            # Ajouter une variation aléatoire aux composants RGB
+            variation = np.array([random.randint(-20, 20) for _ in range(3)])
+            new_pixel = np.clip(base_pixel + variation, 0, 255)
 
-            # Éviter les valeurs problématiques des erreurs enregistrées
+            # Éviter les combinaisons déjà marquées comme erreur
             if avoid_data and (i, j) in avoid_data:
-                avg_pixel = (avg_pixel + np.array([random.randint(0, 50) for _ in range(3)])) % 255
-
-            generated_image[i, j] = avg_pixel
+                new_pixel = (new_pixel + np.array([random.randint(0, 50) for _ in range(3)])) % 255
+            
+            generated_image[i, j] = new_pixel
 
     return Image.fromarray(generated_image)
 
@@ -57,7 +53,7 @@ if uploaded_file:
     # Choisir l'animal et générer une nouvelle image basée sur cette référence
     animal = st.selectbox("Sélectionnez un animal :", ["chat", "chien", "cheval"])
     if st.button("Générer une image"):
-        generated_image = generate_image_based_on_pixels(base_pixels, animal, avoid_data=error_data.get(animal))
+        generated_image = generate_image_with_variation(base_pixels, animal, avoid_data=error_data.get(animal))
         st.image(generated_image, caption=f"Image générée pour un {animal}", use_column_width=True)
 
         # Options de validation de l'image générée