Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
 from PIL import Image
-import io
 
 # Fonction pour générer du texte avec Gemini
 def generate_text_with_gemini(api_key, prompt):
@@ -56,7 +55,7 @@ def process_image(image):
         if model.config.id2label[label.item()] == 'person':
             persons += 1
 
-    expected_number_of_persons = 20  # 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
@@ -65,7 +64,7 @@ def process_image(image):
 
     # 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 = "AIzaSyCJVUr-76I4TL42X98d9U0THbRD3MKlB38"  # 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.')
 
@@ -86,4 +85,6 @@ if uploaded_file is not None:
     
     # Afficher le contenu généré
     st.write("**Contenu généré :**")
-    st.write(email_content)
+    
+    # Utilisation de Markdown pour un rendu graphique plus riche
+    st.markdown(email_content)