Update app.py

app.py

@@ -78,6 +78,9 @@ location = st.selectbox("Sélectionnez la localisation de la photo :", ["École"
 # Champ pour entrer le nom du destinataire
 recipient_name = st.text_input("Entrez le nom du destinataire de l'e-mail :")
 
+# Champ pour entrer le nom de l'expéditeur
+sender_name = st.text_input("Entrez votre nom (expéditeur) :")
+
 # 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)
 
@@ -85,7 +88,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 and recipient_name:  # S'assurer que des images sont téléchargées et qu'un nom est saisi
+if uploaded_files and recipient_name and sender_name:  # S'assurer que des images sont téléchargées, qu'un destinataire et un expéditeur sont fournis
     total_persons = 0
     total_images = len(uploaded_files)
     
@@ -113,7 +116,8 @@ if uploaded_files and recipient_name:  # S'assurer que des images sont télécha
     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."
+        f"Le mail doit être prêt à l'envoi, clair, concis et adapté à un contexte professionnel. "
+        f"L'e-mail doit être signé par {sender_name}."
     )
     api_key = "AIzaSyCJVUr-76I4TL42X98d9U0THbRD3MKlB38"  # Clé d'API
 
@@ -121,4 +125,4 @@ if uploaded_files and recipient_name:  # S'assurer que des images sont télécha
     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.")
+    st.write("Veuillez télécharger des images, entrer le nom du destinataire et votre nom (expéditeur).")