Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
 from PIL import Image, ImageDraw
+import math  # Importer la bibliothèque math pour arrondir au supérieur
 
 # Fonction pour générer du texte avec Gemini
 def generate_text_with_gemini(api_key, prompt):
@@ -77,18 +78,11 @@ uploaded_files = st.file_uploader("Choisissez des images (vous pouvez en ajouter
 # Saisie du seuil de confiance
 confidence_threshold = st.slider("Seuil de confiance pour la détection (0 à 1)", 0.0, 1.0, 0.7, 0.01)
 
-# Saisie du nombre de personnes attendues pour chaque image
+# Traitement des images et calcul de la moyenne
 if uploaded_files:
-    number_of_people = []
-    for i in range(len(uploaded_files)):
-        count = st.number_input(f"Nombre attendu de personnes pour l'image {i + 1}", min_value=1, value=10, step=1, key=f"input_{i}")
-        number_of_people.append(count)
-
-    max_absents = -1
-    best_image = None
-    best_image_name = ""
-    best_absents = 0  # Pour stocker le nombre d'absents de l'image avec le plus d'absents
-
+    total_persons = 0
+    total_images = len(uploaded_files)
+    
     for i, uploaded_file in enumerate(uploaded_files):
         # Lire l'image téléchargée
         image = Image.open(uploaded_file)
@@ -97,32 +91,22 @@ if uploaded_files:
         # Traitement de l'image
         modified_image, detected_persons = process_image(image, confidence_threshold=confidence_threshold)
         
-        # Calculer le nombre d'absents
-        absents = number_of_people[i] - detected_persons
-        if absents < 0:
-            absents = 0
-
+        # Ajouter au total des personnes détectées
+        total_persons += detected_persons
+        
         # Afficher l'image modifiée
         st.image(modified_image, caption=f'Image avec personnes détectées : {uploaded_file.name}', use_column_width=True)
-        
-        # Afficher le nombre d'absents
-        st.write(f"**Nombre d'absents pour l'image {i + 1} ({uploaded_file.name}) : {absents}**")
+        st.write(f"**Personnes détectées sur l'image {i + 1} ({uploaded_file.name}) : {detected_persons}**")
 
-        # Garder une trace de l'image avec le plus grand nombre d'absents
-        if absents > max_absents:
-            max_absents = absents
-            best_image = modified_image
-            best_image_name = uploaded_file.name
-            best_absents = absents  # Stocker le nombre d'absents pour le mail
+    # Calcul de la moyenne et arrondi au supérieur
+    if total_images > 0:
+        average_persons = math.ceil(total_persons / total_images)  # Arrondir au supérieur
+        st.write(f"**Moyenne des personnes détectées sur toutes les images : {average_persons}**")
 
-    # Générer un e-mail avec Gemini basé sur le nombre maximal d'absents
-    prompt = f"Génère moi un mail destiné à l'attaché de promotion d'une école, pour lui signaler que {best_absents} élèves sont absents dans la classe 'ESGI 2'."
+    # Générer un e-mail avec Gemini basé sur la moyenne arrondie
+    prompt = f"Génère-moi un mail destiné à l'attaché de promotion d'une école, pour lui signaler que la moyenne des personnes détectées sur les images est de {average_persons} élèves pour la classe 'ESGI 2'."
     api_key = "AIzaSyCJVUr-76I4TL42X98d9U0THbRD3MKlB38"  # Clé d'API
 
     email_text = generate_text_with_gemini(api_key, prompt)
     st.write("**E-mail généré :**")
     st.write(email_text)
-
-    # Afficher l'image avec le plus grand nombre d'absents
-    if best_image:
-        st.image(best_image, caption=f'Image avec le plus grand nombre d\'absents : {best_image_name}', use_column_width=True)