Update app.py

app.py

@@ -72,6 +72,12 @@ def process_image(image, confidence_threshold=0.7):
 # Interface Streamlit
 st.title("Analyse d'Images pour Détection de Personnes")
 
+# Ajout d'une boîte de sélection pour la localisation
+location = st.selectbox("Sélectionnez la localisation de la photo :", ["École", "Manifestation", "Meeting", "Autre"])
+
+# Champ pour entrer le nom du destinataire
+recipient_name = st.text_input("Entrez le nom du destinataire de l'e-mail :")
+
 # Téléchargement des images
 uploaded_files = st.file_uploader("Choisissez des images (vous pouvez en ajouter plusieurs)", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
@@ -79,7 +85,7 @@ uploaded_files = st.file_uploader("Choisissez des images (vous pouvez en ajouter
 confidence_threshold = st.slider("Seuil de confiance pour la détection (0 à 1)", 0.0, 1.0, 0.7, 0.01)
 
 # Traitement des images et calcul de la moyenne
-if uploaded_files:
+if uploaded_files and recipient_name:  # S'assurer que des images sont téléchargées et qu'un nom est saisi
     total_persons = 0
     total_images = len(uploaded_files)
     
@@ -103,10 +109,16 @@ if uploaded_files:
         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 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'."
+    # Générer un e-mail avec Gemini basé sur la moyenne arrondie et les détails du contexte
+    prompt = (
+        f"Génère-moi un mail destiné à {recipient_name}, attaché de promotion, pour lui signaler que la moyenne "
+        f"des personnes détectées sur les images est de {average_persons} élèves dans le cadre d'une {location}. "
+        "Le mail doit être prêt à l'envoi, clair, concis et adapté à un contexte professionnel."
+    )
     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)
+else:
+    st.write("Veuillez télécharger des images et entrer le nom du destinataire.")