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):
@@ -83,6 +87,7 @@ if uploaded_files:
     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
 
     for i, uploaded_file in enumerate(uploaded_files):
         # Lire l'image téléchargée
@@ -108,6 +113,15 @@ if uploaded_files:
             max_absents = absents
             best_image = modified_image
             best_image_name = uploaded_file.name
+            best_absents = absents  # Stocker le nombre d'absents pour le mail
+
+    # 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'."
+    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: