Update app.py

app.py

@@ -67,7 +67,7 @@ def analyze():
     all_percentages = {}
 
     for model_name, model in models.items():
-        # Set confidence threshold per model
+        # Confidence per model
         conf = 0.03 if model_name == "newpig" else default_conf_threshold
 
         results = model(img, conf=conf, imgsz=imgsz)
@@ -79,7 +79,7 @@ def analyze():
         for i, cls_id in enumerate(boxes_cls):
             cls_name = model.names.get(cls_id, str(cls_id))
 
-            # Skip pores from newpig model
+            # Skip pores from newpig
             if model_name == "newpig" and cls_name.lower() == "pores":
                 continue
 
@@ -90,13 +90,23 @@ def analyze():
                 if intersection.area > 0:
                     class_polygons[cls_id].append(intersection)
 
-        # Calculate percentages
-        skin_percentages = {}
+        # Pre-fill with 0 for all classes
+        skin_percentages = {
+            name: 0.0 for name in model.names.values()
+        }
+
+        # Remove pores from newpig
+        if model_name == "newpig" and "pores" in skin_percentages:
+            skin_percentages.pop("pores")
+
+        # Fill detected percentages
         for cls_id, polys in class_polygons.items():
             union_poly = unary_union(polys)
             pixels = union_poly.area
             percentage = (pixels / face_area) * 100 if face_area > 0 else 0.0
             cls_name = model.names.get(cls_id, str(cls_id))
+            if model_name == "newpig" and cls_name.lower() == "pores":
+                continue
             skin_percentages[cls_name] = round(percentage, 2)
 
         all_percentages[model_name] = skin_percentages