Update app.py

app.py

@@ -1,24 +1,26 @@
+import requests
 from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
 from PIL import Image
-import requests
 
-# Fonction pour envoyer l'email via Mailtrap
+# Fonction pour envoyer un email via Mailtrap
 def send_email(api_key, inbox_id, subject, content, from_email, to_email):
-    url = f"https://mailtrap.io/api/v1/inboxes/{inbox_id}/messages"
+    url = f"https://mailtrap.io/api/v1/inboxes/{inbox_id}/messages"  # Assurez-vous que l'URL est correcte
     headers = {
         "Api-Token": api_key,
         "Content-Type": "application/json"
     }
     data = {
         "from": from_email,
-        "to": to_email,
+        "to": [ {"email": to_email} ],
         "subject": subject,
-        "text": content
+        "text": content,
+        "category": "Integration Test"
     }
     response = requests.post(url, headers=headers, json=data)
     
     # Debug: Print the raw response
+    print(f"Email response status code: {response.status_code}")
     print(f"Email response content: {response.text}")
 
     try:
@@ -50,7 +52,8 @@ def generate_text_with_gemini(api_key, prompt):
     response = requests.post(url, headers=headers, params=params, json=data)
     
     # Debug: Print the raw response
-    print(f"Response content: {response.text}")
+    print(f"Gemini response status code: {response.status_code}")
+    print(f"Gemini response content: {response.text}")
 
     try:
         response_json = response.json()
@@ -82,23 +85,27 @@ for score, label, box in zip(results["scores"], results["labels"], results["boxe
     if model.config.id2label[label.item()] == 'person':
         persons += 1
 
-expected_number_of_persons = 40  # Nombre attendu de personnes
+expected_number_of_persons = 30  # Nombre attendu de personnes
 absents = expected_number_of_persons - persons
 
+# Assurez-vous que le nombre d'absents ne soit pas négatif
+if absents < 0:
+    absents = 0
+
 # Générer le contenu de l'email avec Gemini
 prompt = f"Notifier de l'absence: Le nombre de personnes présentes est de {persons}. Le nombre d'absents est de {absents}."
-gemini_api_key = "AIzaSyCJVUr-76I4TL42X98d9U0THbRD3MKlB38"
+gemini_api_key = "AIzaSyCJVUr-76I4TL42X98d9U0THbRD3MKlB38"  # Remplacez par votre clé API Gemini
 status_code, response_json = generate_text_with_gemini(gemini_api_key, prompt)
-email_content = response_json.get('text', 'Content generation failed.')
+email_content = response_json.get('candidates', [{}])[0].get('content', {}).get('parts', [{}])[0].get('text', 'Content generation failed.')
 
 # Définir les variables pour l'email
 email_subject = "Notification d'absents"
-from_email = "your-email@example.com"
-to_email = "recipient@example.com"
+from_email = "mailtrap@example.com"
+to_email = "noahhamidou@hotmail.com"
 
 # Clés API Mailtrap et informations d'email
-mailtrap_api_key = "0dfd557dc980d49b306ae8647a9fa585"
-inbox_id = "3128145"
+mailtrap_api_key = "0dfd557dc980d49b306ae8647a9fa585"  # Remplacez par votre clé API Mailtrap
+inbox_id = "3128145"  # Remplacez par votre ID de boîte de réception
 
 # Envoyer l'email
 status_code, response_json = send_email(mailtrap_api_key, inbox_id, email_subject, email_content, from_email, to_email)