Update app.py

app.py

@@ -1,60 +1,65 @@
 import streamlit as st
 import numpy as np
-import matplotlib.pyplot as plt
-from PIL import Image, ImageDraw
-
-# Fonction pour générer une image simple et stylisée d'un animal
-def generate_animal_image(animal):
-    img_size = (300, 300)
-    image = Image.new("RGB", img_size, (255, 255, 255))
-    draw = ImageDraw.Draw(image)
-    
-    # Paramètres pour dessiner chaque animal
-    if animal == "Chat":
-        # Corps du chat (cercle simple + oreilles)
-        draw.ellipse((50, 100, 250, 300), fill=(200, 160, 200), outline="black")
-        draw.polygon([(80, 100), (100, 60), (120, 100)], fill=(200, 160, 200), outline="black") # Oreille gauche
-        draw.polygon([(180, 100), (200, 60), (220, 100)], fill=(200, 160, 200), outline="black") # Oreille droite
-
-    elif animal == "Chien":
-        # Corps du chien (forme ovale + oreilles pendantes)
-        draw.ellipse((50, 100, 250, 300), fill=(160, 100, 60), outline="black")
-        draw.rectangle([(60, 60), (100, 120)], fill=(160, 100, 60), outline="black")  # Oreille gauche
-        draw.rectangle([(200, 60), (240, 120)], fill=(160, 100, 60), outline="black")  # Oreille droite
-
-    elif animal == "Cheval":
-        # Corps du cheval (corps allongé + oreilles en pointe)
-        draw.rectangle([(80, 150), (220, 300)], fill=(120, 70, 50), outline="black")
-        draw.polygon([(100, 150), (110, 100), (130, 150)], fill=(120, 70, 50), outline="black")  # Oreille gauche
-        draw.polygon([(170, 150), (180, 100), (200, 150)], fill=(120, 70, 50), outline="black")  # Oreille droite
-
-    return image
-
-# Initialisation de l'interface Streamlit
-st.title("Générateur d'Images d'Animaux")
-
-# Variables de session pour stocker l'état des images générées
-if "animal_image" not in st.session_state:
-    st.session_state.animal_image = None
-
-# Boutons pour sélectionner l'animal
-col1, col2, col3 = st.columns(3)
-with col1:
-    if st.button("Chat"):
-        st.session_state.animal_image = generate_animal_image("Chat")
-
-with col2:
-    if st.button("Chien"):
-        st.session_state.animal_image = generate_animal_image("Chien")
-
-with col3:
-    if st.button("Cheval"):
-        st.session_state.animal_image = generate_animal_image("Cheval")
-
-# Affichage de l'image générée
-if st.session_state.animal_image:
-    st.image(st.session_state.animal_image, caption="Image Générée", use_column_width=False, width=300)
-
-# Bouton de réinitialisation
-if st.button("Réinitialiser", key="reset", help="Cliquez pour générer une nouvelle image"):
-    st.session_state.animal_image = None
+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
+
+# 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)
+
+# 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.")
+
+# 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.")
+
+# 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é
+
+# 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()