Update app.py

app.py

@@ -3,34 +3,6 @@ from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
 from PIL import Image
 
-# 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"  # Assurez-vous que l'URL est correcte
-    headers = {
-        "Api-Token": api_key,
-        "Content-Type": "application/json"
-    }
-    data = {
-        "from": from_email,
-        "to": [ {"email": to_email} ],
-        "subject": subject,
-        "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:
-        response_json = response.json()
-    except ValueError:
-        response_json = {}
-        print("Failed to decode JSON response")
-    
-    return response.status_code, response_json
-
 # Fonction pour générer du texte avec Gemini
 def generate_text_with_gemini(api_key, prompt):
     url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent"
@@ -85,7 +57,7 @@ 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 = 30  # Nombre attendu de personnes
+expected_number_of_persons = 10  # Nombre attendu de personnes
 absents = expected_number_of_persons - persons
 
 # Assurez-vous que le nombre d'absents ne soit pas négatif
@@ -94,19 +66,9 @@ if 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"  # Remplacez par votre clé API Gemini
+gemini_api_key = "YOUR_GEMINI_API_KEY"  # Remplacez par votre clé API Gemini
 status_code, response_json = generate_text_with_gemini(gemini_api_key, prompt)
 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 = "mailtrap@example.com"
-to_email = "noahhamidou@hotmail.com"
-
-# Clés API Mailtrap et informations d'email
-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)
-print(f"Email sent with status code {status_code}: {response_json}")
+# Afficher le contenu généré
+print(f"Generated Content: {email_content}")