Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import numpy as np
-from PIL import Image, ImageFilter
+from PIL import Image, ImageFilter, ImageEnhance
 from collections import Counter
 
 # Ensembles pour stocker les pixels validés et non validés
@@ -22,17 +22,22 @@ def dominant_color(surrounding_pixels):
 # Fonction pour générer une image en fonction des pixels voisins
 def generate_image_from_neighbors(base_pixels):
     height, width, _ = base_pixels.shape
-    generated_image = np.zeros((height, width, 3), dtype=np.uint8)
+    generated_image = np.zeros((50, 50, 3), dtype=np.uint8)  # Image générée de 50x50 pixels
+
+    # Itération sur chaque pixel de l'image générée
+    for i in range(50):  # Hauteur de l'image générée
+        for j in range(50):  # Largeur de l'image générée
+            # Calculer les indices correspondants dans l'image de base
+            orig_i = int(i * (height / 50))  # Mappage de 50 à 300
+            orig_j = int(j * (width / 50))    # Mappage de 50 à 300
 
-    # Itération sur chaque pixel de l'image
-    for i in range(height):
-        for j in range(width):
             surrounding_pixels = []
             # Collecte des couleurs des pixels environnants
             for di in range(-1, 2):  # -1, 0, 1
                 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])
+                    # Vérifier les limites
+                    if 0 <= orig_i + di < height and 0 <= orig_j + dj < width:
+                        surrounding_pixels.append(base_pixels[orig_i + di, orig_j + dj])
 
             # Déterminer la couleur dominante parmi les pixels environnants
             new_pixel_color = dominant_color(surrounding_pixels)