Update app.py

app.py

@@ -1,65 +1,50 @@
 import streamlit as st
 import numpy as np
-import time
-import random
-
-# Fonction simulée pour générer une "image" aléatoire sous forme de matrice de pixels
-def generate_random_image_data():
-    return np.random.randint(0, 256, (300, 300, 3), dtype=np.uint8)
-
-# Fonction pour extraire les "caractéristiques" d'une image (simulation)
-def extract_image_features(image_data):
-    avg_color = np.mean(image_data, axis=(0, 1))
-    feature_description = f"Moyenne de couleur RGB: ({int(avg_color[0])}, {int(avg_color[1])}, {int(avg_color[2])})"
-    return feature_description
-
-# Initialisation de session
-if "feedback" not in st.session_state:
-    st.session_state["feedback"] = []
-if "don_data" not in st.session_state:
-    st.session_state["don_data"] = ""
-if "auto_run" not in st.session_state:
-    st.session_state["auto_run"] = True
+from PIL import Image
+import requests
+from io import BytesIO
+
+# Initialisation des données
+if "images" not in st.session_state:
+    st.session_state.images = []
+
+# 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)
+
+# Analyse basique de l'image
+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}")
 
 # Interface Streamlit
-st.title("Générateur d'images unique avec feedback")
-
-# Boutons d'animaux
-animal = st.radio("Choisissez un animal", ["Chat", "Chien", "Cheval"])
-
-# Génération d'une "image" aléatoire chaque 3 secondes (sans API ni stockage d'image)
-if st.session_state["auto_run"]:
-    image_data = generate_random_image_data()
-    st.image(image_data, caption=f"Image générée pour {animal}", use_column_width=True)
-
-# Extraction de "caractéristiques" de l'image (simulation)
-feature_description = extract_image_features(image_data)
-st.write("Caractéristiques extraites de l'image :")
-st.write(feature_description)
+st.title("Générateur d'Images Auto-Entrant (Démonstration)")
+st.write("Choisissez un animal pour générer une image et analyser ses pixels.")
 
-# Boutons de feedback
-col1, col2 = st.columns(2)
-if col1.button("Oui"):
-    st.session_state["feedback"].append({"animal": animal, "features": feature_description})
-    st.write("Feedback positif enregistré.")
-elif col2.button("Non"):
-    st.write("Feedback négatif : L'image ne correspond pas.")
+col1, col2, col3 = st.columns(3)
 
-# Bouton pour enregistrer les données extraites (texte seulement)
-if st.button("Enregistrer"):
-    st.session_state["don_data"] += feature_description + "\n"
-    st.write("Caractéristiques de l'image sauvegardées.")
+if col1.button("Chat"):
+    download_image("Chat")
+if col2.button("Chien"):
+    download_image("Chien")
+if col3.button("Cheval"):
+    download_image("Cheval")
 
-# Champs de texte pour le don manuel de données
-if st.button("Don"):
-    with st.form("form_don"):
-        don_data = st.text_area("Entrez les données manuellement", value=st.session_state["don_data"])
-        if st.form_submit_button("Envoyer"):
-            st.session_state["don_data"] = don_data
-            st.write("Données ajoutées avec succès.")
-            st.session_state["auto_run"] = False  # Arrête l'auto-génération si le don est activé
+# Affichage des images et analyses
+for idx, img in enumerate(st.session_state.images):
+    st.image(img, caption=f"Image {idx+1}", width=300)
 
-# Relance automatique toutes les 3 secondes si le bouton "Don" n'a pas été utilisé
-if st.session_state["auto_run"]:
-    time.sleep(3)
-    st.experimental_rerun()
+# Bouton pour réinitialiser
+if st.button("Redémarrer", key="reset"):
+    st.session_state.images = []