Update app.py

app.py

@@ -9,7 +9,7 @@ dataset = load_dataset("dwb2023/brain-tumor-image-dataset-semantic-segmentation"
 
 def simplify_segmentation(segmentation, max_points=20):
     if not segmentation or len(segmentation) == 0:
-        return []
+        return segmentation  # Return original if empty
     epsilon = 1.0
     simplified = rdp(np.array(segmentation), epsilon=epsilon)
     while len(simplified) > max_points:
@@ -38,17 +38,13 @@ def draw_annotations(index):
 
         # Draw original segmentation mask
         segmentation = record["segmentation"]
-        original_points = 0
-        simplified_points = 0
         for seg in segmentation:
-            if seg:  # Check if the segmentation is not empty
-                draw.polygon(seg, outline="blue", width=2)
-                original_points += len(seg)
-                
-                # Simplify and draw simplified segmentation
-                simplified_seg = simplify_segmentation(seg)
-                draw.polygon(simplified_seg, outline="green", width=2)
-                simplified_points += len(simplified_seg)
+            draw.polygon(seg, outline="blue", width=2)
+
+        # Draw simplified segmentation
+        simplified_segmentation = [simplify_segmentation(seg) for seg in segmentation]
+        for seg in simplified_segmentation:
+            draw.polygon(seg, outline="green", width=2)
 
         # Prepare additional information
         category_id = record["category_id"]
@@ -59,8 +55,8 @@ def draw_annotations(index):
         info += f"Image ID: {record['id']}\n"
         info += f"Category ID: {category_id}\n"
         info += f"Bounding Box: [{bbox[0]:.2f}, {bbox[1]:.2f}, {bbox[2]:.2f}, {bbox[3]:.2f}]\n"
-        info += f"Original Segmentation Points: {original_points}\n"
-        info += f"Simplified Segmentation Points: {simplified_points}\n"
+        info += f"Original Segmentation Points: {sum(len(seg) for seg in segmentation)}\n"
+        info += f"Simplified Segmentation Points: {sum(len(seg) for seg in simplified_segmentation)}\n"
         info += f"Area: {area:.2f}"
 
         return img, info