Update app.py

app.py

@@ -34,7 +34,11 @@ def generate_text_with_gemini(api_key, prompt):
         response_json = {}
         print("Failed to decode JSON response")
     
-    return response.status_code, response_json
+    # Renvoyer le texte généré ou un message d'erreur
+    if response.status_code == 200 and "candidates" in response_json:
+        return response_json["candidates"][0]["content"]["parts"][0]["text"]
+    else:
+        return "Erreur lors de la génération du texte."
 
 # Fonction de traitement de l'image et d'encadrement des personnes détectées
 def process_image(image, confidence_threshold=0.7):
@@ -73,17 +77,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 = ""
-
+    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)
@@ -92,23 +90,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}**")
+
+    # Calcul de la moyenne
+    if total_images > 0:
+        average_persons = total_persons / total_images
+        st.write(f"**Moyenne des personnes détectées sur toutes les images : {average_persons:.2f}**")
 
-        # 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
+    # Générer un e-mail avec Gemini basé sur la moyenne
+    prompt = f"Génère moi un mail destiné à l'attaché de promotion d'une école, pour lui signaler qu'en moyenne, {average_persons:.2f} élèves ont été détectés dans les images de la classe 'ESGI 2'."
+    api_key = "AIzaSyCJVUr-76I4TL42X98d9U0THbRD3MKlB38"  # Clé d'API
 
-    # 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)
+    email_text = generate_text_with_gemini(api_key, prompt)
+    st.write("**E-mail généré :**")
+    st.write(email_text)