Update app.py

app.py

@@ -46,30 +46,43 @@ st.write("Choisissez un animal pour générer une image ou téléchargez une ima
 # Téléchargement de l'image importée
 uploaded_file = st.file_uploader("Téléchargez une image de référence (JPG ou PNG)", type=["jpg", "jpeg", "png"])
 
+# Initialisation d'une image générée
+generated_image = None
+
 # 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)
 
-    # Choisir l'animal et générer une nouvelle image basée sur cette référence
+    # 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)
 
-        # Options de validation de l'image générée
-        if st.button("Oui"):
-            # Enregistrer les pixels de l'image générée dans la liste "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
-            st.success("Image validée!")
-
-        elif st.button("Non"):
-            # Enregistrer les pixels de l'image générée dans la liste "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
-            st.error("Image rejetée!")
+    # Options de validation de l'image générée
+    if generated_image is not None:
+        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"
+                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
+                st.success("Image validée!")
+
+                # Réinitialiser l'image générée après validation
+                generated_image = None
+
+        with col2:
+            if st.button("Non"):
+                # Enregistrer les pixels de l'image générée dans la liste "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
+                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
     st.write("Pixels validés (oe):", len(oe), "pixels enregistrés.")