Update app.py

app.py

@@ -71,16 +71,16 @@ st.markdown("""
     
     .stButton > button {
         width: 200px;
-        background-color: #0066cc !important;
-        color: white !important;
-        border: none !important;
+        background-color: #f8f9fa !important;
+        color: #1a1a1a !important;
+        border: 1px solid #e9ecef !important;
         padding: 0.5rem 1rem !important;
         border-radius: 5px !important;
         transition: all 0.3s ease !important;
     }
     
     .stButton > button:hover {
-        background-color: #0052a3 !important;
+        background-color: #e9ecef !important;
         transform: translateY(-1px);
     }
     
@@ -119,79 +119,53 @@ def create_heatmap_overlay(image, box, score):
     overlay = Image.new('RGBA', image.size, (0, 0, 0, 0))
     draw = ImageDraw.Draw(overlay)
     
-    def get_temp_color(value):
-        if value > 0.8:
-            return (255, 0, 0)  # Rouge vif
-        elif value > 0.6:
-            return (255, 69, 0)  # Rouge-orange
-        elif value > 0.4:
-            return (255, 165, 0)  # Orange
-        else:
-            return (255, 255, 0)  # Jaune
-    
     x1, y1 = box['xmin'], box['ymin']
     x2, y2 = box['xmax'], box['ymax']
-    width = x2 - x1
-    height = y2 - y1
     
-    # Fond semi-transparent avec dégradé
-    steps = 30
-    for i in range(steps):
-        alpha = int(100 * (1 - (i / steps)))  # Réduit l'opacité maximale à 100
-        base_color = get_temp_color(score)
-        color = base_color + (alpha,)
-        
-        shrink_x = (i * width) / (steps * 2)
-        shrink_y = (i * height) / (steps * 2)
-        
-        draw.rectangle(
-            [x1 + shrink_x, y1 + shrink_y, x2 - shrink_x, y2 - shrink_y],
-            fill=color,
-            outline=None
-        )
+    # Couleur basée sur le score
+    if score > 0.8:
+        fill_color = (255, 0, 0, 100)  # Rouge
+        border_color = (255, 0, 0, 255)
+    elif score > 0.6:
+        fill_color = (255, 165, 0, 100)  # Orange
+        border_color = (255, 165, 0, 255)
+    else:
+        fill_color = (255, 255, 0, 100)  # Jaune
+        border_color = (255, 255, 0, 255)
+    
+    # Rectangle semi-transparent
+    draw.rectangle([x1, y1, x2, y2], fill=fill_color)
     
-    # Bordure plus visible
-    border_color = get_temp_color(score) + (200,)
-    draw.rectangle([x1, y1, x2, y2], outline=border_color, width=3)
+    # Bordure
+    draw.rectangle([x1, y1, x2, y2], outline=border_color, width=2)
     
     return overlay
 
 def draw_boxes(image, predictions):
     result_image = image.copy().convert('RGBA')
     
-    sorted_predictions = sorted(predictions, key=lambda x: x['score'])
-    
-    for pred in sorted_predictions:
+    for pred in 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)
+        # Création de l'overlay
+        overlay = create_heatmap_overlay(image, box, score)
+        result_image = Image.alpha_composite(result_image, overlay)
         
         # Ajout du texte
         draw = ImageDraw.Draw(result_image)
-        temp = 36.5 + (score * 2.5)  # Température simulée
-        label = f"{translate_label(pred['label'])} ({score:.1%}) • {temp:.1f}°C"
+        temp = 36.5 + (score * 2.5)
+        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
-        )
-        draw.rectangle(text_bbox, fill="#000000CC")
+        # Fond noir pour le texte
+        text_bbox = draw.textbbox((box['xmin'], box['ymin']-20), label)
+        draw.rectangle(text_bbox, fill=(0, 0, 0, 180))
         
-        # Texte en blanc avec contour noir
+        # Texte en blanc
         draw.text(
-            (box['xmin'], box['ymin']-25),
+            (box['xmin'], box['ymin']-20),
             label,
-            fill="#FFFFFF",
-            stroke_width=1,
-            stroke_fill="#000000"
+            fill=(255, 255, 255, 255)
         )
     
     return result_image