Update app.py

app.py

@@ -19,6 +19,22 @@ def send_email(api_key, inbox_id, subject, content, from_email, to_email):
     response = requests.post(url, headers=headers, json=data)
     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://api.gemini.com/v1/generate"  # Remplacez par l'URL correcte de l'API Gemini
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "prompt": prompt,
+        "max_tokens": 150,  # Ajustez en fonction de vos besoins
+        "temperature": 0.7  # Ajustez en fonction de vos besoins
+    }
+    response = requests.post(url, headers=headers, json=data)
+    response_data = response.json()
+    return response_data.get('text', '')
+
 # URL de l'image d'exemple
 url = "https://www.saint-genes.com/wp-content/uploads/2019/10/DSC05405.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
@@ -35,31 +51,29 @@ outputs = model(**inputs)
 target_sizes = torch.tensor([image.size[::-1]])
 results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]
 
-# Afficher les résultats
+# Compter le nombre de personnes détectées
 persons = 0
 for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
-    box = [round(i, 2) for i in box.tolist()]
-    print(
-        f"Detected {model.config.id2label[label.item()]} with confidence "
-        f"{round(score.item(), 3)} at location {box}"
-    )
     if model.config.id2label[label.item()] == 'person':
-        persons += 1  # Incrémentation correcte
-
-print(f"Total persons detected: {persons}")
+        persons += 1
 
-# Définir les variables pour l'email
 expected_number_of_persons = 10  # Nombre attendu de personnes
 absents = expected_number_of_persons - persons
+
+# 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"
+email_content = generate_text_with_gemini(gemini_api_key, prompt)
+
+# Définir les variables pour l'email
 email_subject = "Notification d'absents"
-email_content = f"Le nombre de personnes présentes est de {persons}. Le nombre d'absents est de {absents}."
+from_email = "your-email@example.com"
+to_email = "recipient@example.com"
 
 # Clés API Mailtrap et informations d'email
-api_key = "0dfd557dc980d49b306ae8647a9fa585"
+mailtrap_api_key = "0dfd557dc980d49b306ae8647a9fa585"
 inbox_id = "3128145"
-from_email = "your-email@example.com"
-to_email = "recipient@example.com"
 
 # Envoyer l'email
-status_code, response_json = send_email(api_key, inbox_id, email_subject, email_content, from_email, to_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}")