Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import numpy as np
-from PIL import Image
-import random
+from PIL import Image, ImageFilter
 from collections import Counter
 
 # Ensembles pour stocker les pixels validés et non validés
@@ -43,6 +42,14 @@ def generate_image_from_neighbors(base_pixels):
 
     return Image.fromarray(generated_image)
 
+# Fonction pour appliquer un filtre à l'image
+def apply_filter(image, filter_type):
+    if filter_type == "Flou":
+        return image.filter(ImageFilter.BLUR)
+    elif filter_type == "Contraste":
+        return ImageEnhance.Contrast(image).enhance(1.5)
+    return image
+
 # Interface utilisateur pour télécharger et afficher l'image importée
 st.title("Générateur d'images d'animaux basé sur l'image importée")
 st.write("Choisissez un animal pour générer une image ou téléchargez une image comme référence.")
@@ -53,15 +60,27 @@ uploaded_file = st.file_uploader("Téléchargez une image de référence (JPG ou
 # Initialisation d'une image générée
 generated_image = None
 
+# Historique des images générées
+image_history = []
+
 # Génération d'image selon la référence téléchargée
 if uploaded_file:
     base_pixels = process_uploaded_image(uploaded_file)
     st.image(Image.fromarray(base_pixels), caption="Image importée de référence", use_column_width=True)
 
+    # Options de filtres
+    filter_option = st.selectbox("Choisissez un filtre à appliquer :", ["Aucun", "Flou", "Contraste"])
+    if filter_option != "Aucun":
+        base_image = Image.fromarray(base_pixels)
+        base_image = apply_filter(base_image, filter_option)
+        st.image(base_image, caption=f"Image avec filtre : {filter_option}", use_column_width=True)
+        base_pixels = np.array(base_image)  # Mettez à jour les pixels de base
+
     # Bouton pour générer une image
     if st.button("Générer une image"):
         generated_image = generate_image_from_neighbors(base_pixels)
         st.image(generated_image, caption="Image générée", use_column_width=True)
+        image_history.append(generated_image)  # Ajoutez l'image générée à l'historique
 
     # Options de validation de l'image générée
     if generated_image is not None:
@@ -71,7 +90,7 @@ if uploaded_file:
                 # 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"
-                st.success("Image validée!")
+                st.success("Image validée ! Les pixels sont enregistrés.")
 
                 # Réinitialiser l'image générée après validation
                 generated_image = None
@@ -81,7 +100,7 @@ if uploaded_file:
                 # 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"
-                st.error("Image rejetée!")
+                st.error("Image rejetée ! Les pixels sont enregistrés.")
 
                 # Réinitialiser l'image générée après rejet
                 generated_image = None
@@ -90,6 +109,11 @@ if uploaded_file:
     st.write("Pixels validés (oe):", len(oe), "pixels enregistrés.")
     st.write("Pixels non validés (nn):", len(nn), "pixels enregistrés.")
 
+    # Affichage de l'historique des images générées
+    st.write("Historique des images générées:")
+    for img in image_history:
+        st.image(img, use_column_width=True)
+
 # Option pour réinitialiser les ensembles
 if st.button("Réinitialiser les ensembles"):
     oe.clear()