Update app.py

app.py

@@ -1,27 +1,41 @@
+
 import gradio as gr
-from PIL import Image, ImageEnhance
-import numpy as np
+from PIL import Image, ImageEnhance, ImageOps
 
-# Fonction pour modifier les couleurs de l'image
+# Fonction pour modifier les couleurs de l'image (ajouter de la saturation)
 def modify_image_color(image):
-    # Convertir l'image PIL en format RGB
     img = image.convert("RGB")
-
-    # Augmenter la saturation de l'image
     enhancer = ImageEnhance.Color(img)
-    img = enhancer.enhance(2)  # Valeur > 1 augmente la saturation, < 1 la diminue
-
+    img = enhancer.enhance(2)  # Augmente la saturation
     return img
 
-# Interface Gradio
+# Fonction pour convertir l'image en noir et blanc
+def convert_to_black_and_white(image):
+    return ImageOps.grayscale(image)
+
+# Interface Gradio avec plusieurs fonctionnalités
 iface = gr.Interface(
-    fn=modify_image_color,  # Fonction à appeler
-    inputs=gr.Image(type="pil", label="Télécharger une image"),  # Champ pour télécharger une image
-    outputs=gr.Image(label="Image modifiée"),  # Afficher l'image modifiée
-    title="Modification des couleurs d'image",
-    description="Téléchargez une image et modifiez ses couleurs en augmentant la saturation."
+    fn=None,  # La fonction principale sera None ici
+    inputs=gr.Image(type="pil", label="Télécharger une image"),  # Télécharger une image
+    outputs="image",  # Sortie image
+    title="Modification d'image",
+    description="Choisissez une option ci-dessous pour modifier l'image.",
+    live=True,  # Permet d'avoir des mises à jour en temps réel
+    layout="vertical",  # Disposition verticale des composants
+
+    # Composants supplémentaires pour chaque fonctionnalité
+    examples=[["image1.jpg"], ["image2.jpg"]],  # Exemples d'images
+    allow_flagging="never"  # Désactive le flagging (optionnelle)
+)
+
+# Ajout de plusieurs options avec des boutons pour activer chaque fonction
+iface.add_component(
+    gr.Button("Augmenter la saturation", elem_id="saturation_button").click(modify_image_color, inputs="image", outputs="image")
+)
+
+iface.add_component(
+    gr.Button("Convertir en noir et blanc", elem_id="bw_button").click(convert_to_black_and_white, inputs="image", outputs="image")
 )
 
 # Lancer l'interface
 iface.launch()
-