Create app.py

app.py

@@ -0,0 +1,60 @@
+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