Added all images report viewer.

aryan-programmer · aryan-programmer · commit bfd1f7443159 · 2025-07-05T12:47:20.000+05:30
Added all images report viewer.
Allows to to see at once which images have issues.
diff --git a/lib/src/data/image_data.dart b/lib/src/data/image_data.dart
@@ -7,7 +7,7 @@ part 'image_data.freezed.dart';
 part 'image_data.g.dart';
 
 @freezed
-abstract class ImageData with _$ImageData {
+sealed class ImageData with _$ImageData {
   ImageData._();
 
   factory ImageData({
@@ -39,7 +39,7 @@ abstract class ImageData with _$ImageData {
 }
 
 @freezed
-abstract class BenchmarkImageData with _$BenchmarkImageData {
+sealed class BenchmarkImageData with _$BenchmarkImageData {
   const BenchmarkImageData._();
 
   factory BenchmarkImageData({
diff --git a/lib/src/data/image_status_tile_data.dart b/lib/src/data/image_status_tile_data.dart
@@ -0,0 +1,37 @@
+import 'package:flutter/material.dart';
+import 'package:freezed_annotation/freezed_annotation.dart';
+
+part 'image_status_tile_data.freezed.dart';
+
+enum ImageStatus {
+  Unchecked,
+  Faulty,
+  Ok;
+
+  (IconData, Color) get iconDataAndColor {
+    late IconData trailingIcon;
+    late Color tileColor;
+    switch (this) {
+      case ImageStatus.Unchecked:
+        trailingIcon = Icons.warning_amber;
+        tileColor = Colors.amberAccent[100]!;
+      case ImageStatus.Faulty:
+        trailingIcon = Icons.error;
+        tileColor = Colors.redAccent[100]!;
+      case ImageStatus.Ok:
+        trailingIcon = Icons.check;
+        tileColor = Colors.greenAccent[100]!;
+    }
+    return (trailingIcon, tileColor);
+  }
+}
+
+@freezed
+sealed class ImageStatusTileData with _$ImageStatusTileData {
+  ImageStatusTileData._();
+
+  factory ImageStatusTileData({
+    required ImageStatus status,
+    required String statusString,
+  }) = _ImageStatusTileData;
+}
diff --git a/lib/src/data/image_status_tile_data.freezed.dart b/lib/src/data/image_status_tile_data.freezed.dart
diff --git a/lib/src/data/project_data.dart b/lib/src/data/project_data.dart
@@ -6,7 +6,7 @@ part 'project_data.freezed.dart';
 part 'project_data.g.dart';
 
 @freezed
-abstract class ProjectData with _$ProjectData {
+sealed class ProjectData with _$ProjectData {
   factory ProjectData({String? benchmarkDataFolder}) = _ProjectData;
 
   factory ProjectData.fromJson(Map<String, dynamic> json) =>
diff --git a/lib/src/models/auto_slicing.dart b/lib/src/models/auto_slicing.dart
@@ -8,7 +8,7 @@ part 'auto_slicing.freezed.dart';
 part 'auto_slicing.g.dart';
 
 @freezed
-abstract class ModelInferenceParameters with _$ModelInferenceParameters {
+sealed class ModelInferenceParameters with _$ModelInferenceParameters {
   factory ModelInferenceParameters({
     required bool keepOriginal,
     required List<SliceInputParams> sliceOptions,
diff --git a/lib/src/store/image_data.store.dart b/lib/src/store/image_data.store.dart
@@ -1,6 +1,7 @@
 import 'package:mobx/mobx.dart';
 import 'package:path/path.dart' as path;
 import 'package:pcb_fault_detection_ui/src/data/image_data.dart';
+import 'package:pcb_fault_detection_ui/src/data/image_status_tile_data.dart';
 import 'package:pcb_fault_detection_ui/src/rust/api/utils.dart';
 import 'package:pcb_fault_detection_ui/src/store/project.store.dart';
 import 'package:pcb_fault_detection_ui/src/utils/components.dart';
@@ -15,7 +16,7 @@ class ImageDataStore extends _ImageDataStore with _$ImageDataStore {
   });
 }
 
-abstract class _ImageDataStore with Store {
+sealed class _ImageDataStore with Store {
   @observable
   ImageData imageData;
 
@@ -54,6 +55,60 @@ abstract class _ImageDataStore with Store {
     return path.join(projectFolder, folderName, IMAGE_FILE_NAME);
   }
 
+  @computed
+  ImageStatusTileData get statusTileData {
+    if (imageData.tracksOnly) {
+      if (imageData.trackDefects.isEmpty) {
+        return ImageStatusTileData(
+          status: ImageStatus.Unchecked,
+          statusString: "Please run inference on this image.",
+        );
+      }
+      final numTrackDefects = imageData.trackDefects
+          .where((v) => v.confidence >= imageData.trackDefectDetectionThreshold)
+          .length;
+      if (numTrackDefects == 0) {
+        return ImageStatusTileData(
+          status: ImageStatus.Ok,
+          statusString: "No track defects found.",
+        );
+      } else {
+        return ImageStatusTileData(
+          status: ImageStatus.Faulty,
+          statusString: "Track defects: $numTrackDefects",
+        );
+      }
+    } else {
+      if (imageData.components.isEmpty) {
+        return ImageStatusTileData(
+          status: ImageStatus.Unchecked,
+          statusString: "Please run inference on this image.",
+        );
+      }
+      if (parent.benchmarkImageData?.components.isEmpty ?? true) {
+        return ImageStatusTileData(
+          status: ImageStatus.Unchecked,
+          statusString: "Please run inference on the benchmark image.",
+        );
+      }
+      int numMissingComponents =
+          nonOverlappingComponents.missingComponents.length;
+      int numExtraComponents = nonOverlappingComponents.extraComponents.length;
+      if (numMissingComponents == 0 && numExtraComponents == 0) {
+        return ImageStatusTileData(
+          status: ImageStatus.Ok,
+          statusString: "No component defects found.",
+        );
+      } else {
+        return ImageStatusTileData(
+          status: ImageStatus.Faulty,
+          statusString:
+              "Missing: $numMissingComponents, Extra: $numExtraComponents",
+        );
+      }
+    }
+  }
+
   @action
   void setComponents(List<YoloEntityOutput> components) {
     imageData = imageData.copyWith(components: components);
diff --git a/lib/src/store/image_data.store.g.dart b/lib/src/store/image_data.store.g.dart
diff --git a/lib/src/store/project.store.dart b/lib/src/store/project.store.dart
@@ -27,7 +27,7 @@ class ProjectStore extends _ProjectStore with _$ProjectStore {
   }
 }
 
-abstract class _ProjectStore with Store {
+sealed class _ProjectStore with Store {
   static final SharedPreferencesAsync prefs = SharedPreferencesAsync();
 
   @observable
diff --git a/lib/src/utils/components.dart b/lib/src/utils/components.dart
@@ -40,10 +40,6 @@ NonOverlappingComponentsResult __getNonOverlappingComponents({
       continue;
     }
     for (YoloEntityOutput test in testComponents) {
-      if (test.confidence <= benchmarkComponentsDetectionThreshold) {
-        benchmarkComponentsSet.remove(benchmark);
-        continue;
-      }
       if (benchmark.boundingBox.iou(box2: test.boundingBox) >=
           benchmarkOverlapThreshold) {
         benchmarkComponentsSet.remove(benchmark);
@@ -60,17 +56,17 @@ NonOverlappingComponentsResult __getNonOverlappingComponents({
 final _getNonOverlappingComponents = memo5(
   (
     List<YoloEntityOutput> benchmarkComponents,
-    double benchmarkComponentsDetectionThreshold,
+    int benchmarkComponentsDetectionThreshold,
     List<YoloEntityOutput> testComponents,
-    double testComponentsDetectionThreshold,
-    double benchmarkOverlapThreshold,
+    int testComponentsDetectionThreshold,
+    int benchmarkOverlapThreshold,
   ) => __getNonOverlappingComponents(
     benchmarkComponents: benchmarkComponents,
     benchmarkComponentsDetectionThreshold:
-        benchmarkComponentsDetectionThreshold,
+        benchmarkComponentsDetectionThreshold / 100,
     testComponents: testComponents,
-    testComponentsDetectionThreshold: testComponentsDetectionThreshold,
-    benchmarkOverlapThreshold: benchmarkOverlapThreshold,
+    testComponentsDetectionThreshold: testComponentsDetectionThreshold / 100,
+    benchmarkOverlapThreshold: benchmarkOverlapThreshold / 100,
   ),
 );
 
@@ -82,8 +78,8 @@ NonOverlappingComponentsResult getNonOverlappingComponents({
   required double benchmarkOverlapThreshold,
 }) => _getNonOverlappingComponents(
   benchmarkComponents,
-  benchmarkComponentsDetectionThreshold,
+  (benchmarkComponentsDetectionThreshold * 100).round(),
   testComponents,
-  testComponentsDetectionThreshold,
-  benchmarkOverlapThreshold,
+  (testComponentsDetectionThreshold * 100).round(),
+  (benchmarkOverlapThreshold * 100).round(),
 );
diff --git a/lib/src/widgets/annotated_image.dart b/lib/src/widgets/annotated_image.dart
@@ -34,7 +34,7 @@ class AnnotatedImage extends StatelessWidget {
     this.onRemoveBlue,
     double? minBoxWidth,
     double? maxBoxWidth,
-  }) : minBoxWidth = minBoxWidth ?? 1.5,
+  }) : minBoxWidth = minBoxWidth ?? 2,
        maxBoxWidth = maxBoxWidth ?? 6 {
     boxWidthDiff = this.maxBoxWidth - this.minBoxWidth;
   }
@@ -99,7 +99,7 @@ class AnnotatedImage extends StatelessWidget {
     final padding = width;
     return [
       Positioned(
-        key: ValueKey((imagePath, color, box)),
+        key: ValueKey((#annotatedImageBoundingBoxPositioned, imagePath, color, box)),
         left: bx - padding,
         top: by - padding,
         width: bw + padding * 2,
diff --git a/lib/src/widgets/pcb_accordion_panel.dart b/lib/src/widgets/pcb_accordion_panel.dart
diff --git a/lib/src/widgets/project_view.dart b/lib/src/widgets/project_view.dart

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ class ProjectStore extends _ProjectStore with _$ProjectStore {`
`27`	`27`	`}`
`28`	`28`	`}`
`29`	`29`
`30`		`-abstract class _ProjectStore with Store {`
	`30`	`+sealed class _ProjectStore with Store {`
`31`	`31`	`static final SharedPreferencesAsync prefs = SharedPreferencesAsync();`
`32`	`32`
`33`	`33`	`@observable`