Update app.py

app.py

@@ -3,48 +3,124 @@ import numpy as np
 from PIL import Image
 import requests
 from io import BytesIO
+import random
+import time
 
 # Initialisation des données
 if "images" not in st.session_state:
     st.session_state.images = []
+if "validated_data" not in st.session_state:
+    st.session_state.validated_data = []
+if "animal_data" not in st.session_state:
+    st.session_state.animal_data = {
+        "chat": [],
+        "chien": [],
+        "cheval": []
+    }
 
-# Téléchargement d'images basiques selon le bouton cliqué
-def download_image(animal):
-    url = {
-        "Chat": "https://i.ibb.co/2FnDthw/IMG-6419.jpg",  # Lien mis à jour pour le chat
-        "Chien": "https://example.com/sample-dog-image.jpg",
-        "Cheval": "https://example.com/sample-horse-image.jpg"
-    }.get(animal, None)
-    if url:
-        response = requests.get(url)
-        img = Image.open(BytesIO(response.content)).resize((300, 300))
-        st.session_state.images.append(img)
-        analyze_image(img)
+# Fonction pour télécharger une image aléatoire
+def download_random_image(animal):
+    # Liste d'URLs d'images fictives à remplacer par de vraies images
+    animal_images = {
+        "chat": [
+            "https://i.ibb.co/2FnDthw/IMG-6419.jpg",
+            "https://placekitten.com/300/300"  # Exemple d'URL fictive
+        ],
+        "chien": [
+            "https://placedog.net/300/300",  # Exemple d'URL fictive
+            "https://dog.ceo/api/breeds/image/random"
+        ],
+        "cheval": [
+            "https://placehorse.com/300/300"  # Exemple d'URL fictive
+        ]
+    }
+    
+    url = random.choice(animal_images[animal])
+    response = requests.get(url)
+    img = Image.open(BytesIO(response.content)).resize((300, 300))
+    return img
 
-# Analyse basique de l'image
+# Fonction pour analyser l'image (simulée)
 def analyze_image(img):
-    st.write("Analyse basique de l'image")
     arr = np.array(img)
     avg_color = arr.mean(axis=(0, 1))
-    st.write(f"Couleur moyenne: {avg_color}")
+    return avg_color
+
+# Fonction pour redémarrer le processus
+def reset_process():
+    st.session_state.images = []
+    st.session_state.validated_data = []
+    st.session_state.animal_data = {
+        "chat": [],
+        "chien": [],
+        "cheval": []
+    }
 
 # Interface Streamlit
-st.title("Générateur d'Images Auto-Entrant (Démonstration)")
+st.title("Générateur d'Images Auto-Entrant")
 st.write("Choisissez un animal pour générer une image et analyser ses pixels.")
 
+# Boutons pour les animaux
 col1, col2, col3 = st.columns(3)
 
-if col1.button("Chat"):
-    download_image("Chat")
-if col2.button("Chien"):
-    download_image("Chien")
-if col3.button("Cheval"):
-    download_image("Cheval")
+with col1:
+    if st.button("Chat"):
+        img = download_random_image("chat")
+        st.session_state.images.append(img)
+        avg_color = analyze_image(img)
+        st.write(f"Couleur moyenne : {avg_color}")
+
+with col2:
+    if st.button("Chien"):
+        img = download_random_image("chien")
+        st.session_state.images.append(img)
+        avg_color = analyze_image(img)
+        st.write(f"Couleur moyenne : {avg_color}")
+
+with col3:
+    if st.button("Cheval"):
+        img = download_random_image("cheval")
+        st.session_state.images.append(img)
+        avg_color = analyze_image(img)
+        st.write(f"Couleur moyenne : {avg_color}")
 
-# Affichage des images et analyses
+# Affichage des images générées
 for idx, img in enumerate(st.session_state.images):
-    st.image(img, caption=f"Image {idx+1}", width=300)
+    st.image(img, caption=f"Image {idx + 1}", width=300)
 
-# Bouton pour réinitialiser
-if st.button("Redémarrer", key="reset"):
-    st.session_state.images = []
+# Boutons de validation
+if st.button("Oui"):
+    if st.session_state.images:
+        avg_color = analyze_image(st.session_state.images[-1])
+        st.session_state.validated_data.append(avg_color)
+        st.session_state.animal_data["chat" if "chat" in st.session_state.images[-1].filename else "chien" if "chien" in st.session_state.images[-1].filename else "cheval"].append(avg_color)
+
+if st.button("Non"):
+    if st.session_state.images:
+        st.session_state.images.pop()  # Ignore l'image
+        # Modifier les données précédentes pour éviter la même erreur
+        # Implémentez votre logique ici pour ajuster les données
+
+# Champ de données manuelles
+if st.button("Don"):
+    st.session_state.don_field_visible = True
+
+if "don_field_visible" in st.session_state and st.session_state.don_field_visible:
+    don_text = st.text_area("Entrez vos données manuelles :")
+    if st.button("Envoyer"):
+        if don_text:
+            # Traitement des données du champ "Don"
+            st.session_state.animal_data["chat"].append(don_text)  # Exemple d'utilisation
+            st.session_state.don_field_visible = False
+
+# Bouton pour redémarrer le processus
+if st.button("Redémarrer"):
+    reset_process()
+
+# Démarrer automatiquement après 3 secondes si le champ "Don" n'a pas été rempli
+if "don_field_visible" not in st.session_state:
+    st.session_state.don_field_visible = False
+
+if not st.session_state.don_field_visible:
+    time.sleep(3)
+    # Vous pouvez ajouter ici le code pour relancer le processus ou une action de votre choix