Update app.py

app.py

@@ -134,9 +134,10 @@ def create_heatmap_overlay(image, box, score):
     width = x2 - x1
     height = y2 - y1
     
+    # Fond semi-transparent avec dégradé
     steps = 30
     for i in range(steps):
-        alpha = int(255 * (1 - (i / steps)) * 0.7)
+        alpha = int(100 * (1 - (i / steps)))  # Réduit l'opacité maximale à 100
         base_color = get_temp_color(score)
         color = base_color + (alpha,)
         
@@ -149,8 +150,9 @@ def create_heatmap_overlay(image, box, score):
             outline=None
         )
     
+    # Bordure plus visible
     border_color = get_temp_color(score) + (200,)
-    draw.rectangle([x1, y1, x2, y2], outline=border_color, width=2)
+    draw.rectangle([x1, y1, x2, y2], outline=border_color, width=3)
     
     return overlay
 
@@ -163,21 +165,27 @@ def draw_boxes(image, predictions):
         box = pred['box']
         score = pred['score']
         
+        # Ajout de l'overlay avec heatmap
         heatmap = create_heatmap_overlay(image, box, score)
         result_image = Image.alpha_composite(result_image, heatmap)
         
+        # Ajout du texte
         draw = ImageDraw.Draw(result_image)
-        temp = 36.5 + (score * 2.5)
+        temp = 36.5 + (score * 2.5)  # Température simulée
         label = f"{translate_label(pred['label'])} ({score:.1%}) • {temp:.1f}°C"
         
+        # Fond noir semi-transparent pour le texte
         text_bbox = draw.textbbox((box['xmin'], box['ymin']-25), label)
         padding = 3
         text_bbox = (
-            text_bbox[0]-padding, text_bbox[1]-padding,
-            text_bbox[2]+padding, text_bbox[3]+padding
+            text_bbox[0]-padding, 
+            text_bbox[1]-padding,
+            text_bbox[2]+padding, 
+            text_bbox[3]+padding
         )
         draw.rectangle(text_bbox, fill="#000000CC")
         
+        # Texte en blanc avec contour noir
         draw.text(
             (box['xmin'], box['ymin']-25),
             label,