changement de l'interface

app.py

@@ -112,38 +112,37 @@ def predict(image, model_choice):
     return "Modèle non reconnu."
 
 
-# Interface Gradio avec un thème amélioré
 with gr.Blocks(theme='Nymbo/Nymbo_Theme') as interface:
-    gr.Markdown(
-        """
-        # Prédiction d'Âge et de Genre
-        Upload une image de visage pour prédire l'âge et/ou le genre avec différents modèles.
-
-        **Instructions :**  
-        1. Choisissez un modèle dans la liste ci-dessous.  
-        2. Cliquez sur la zone d'upload pour ajouter une image.  
-        3. La prédiction s'affichera à droite.
-        """
-    )
+    gr.Markdown("## 🎭 Prédiction d'Âge et de Genre")
     
     with gr.Row():
         with gr.Column(scale=1):
+            gr.Markdown("### 📷 Upload ton image")
+            img_input = gr.Image(type="pil", label="Image de visage")
+
+            gr.Markdown("### 🧠 Sélectionne un modèle")
             model_choice = gr.Radio(
                 ["Genre", "Âge", "Genre + Âge", "Pré-entraîné"],
-                label="Choisissez un modèle",
+                label="Modèle",
                 value="Pré-entraîné"
             )
-            img_input = gr.Image(type="pil", label="Upload une image de visage")
-        
+
         with gr.Column(scale=1):
-            output_text = gr.Textbox(label="Résultat de la prédiction", interactive=False)
-    
-    # Déclencher la prédiction dès que l'image est uploadée
+            gr.Markdown("### 📊 Résultat de la prédiction")
+            output_text = gr.Textbox(label="Résultat", interactive=False)
+            img_preview = gr.Image(label="Prévisualisation", interactive=False, visible=False)
+
+    # Mise à jour des résultats et de la prévisualisation
+    def process_and_display(image, model_choice):
+        result = predict(image, model_choice)
+        return result, image
+
     img_input.change(
-        fn=predict,
+        fn=process_and_display,
         inputs=[img_input, model_choice],
-        outputs=output_text
+        outputs=[output_text, img_preview]
     )
 
+# Lancer l'application
 if __name__ == "__main__":
     interface.launch()