49 lines
1.4 KiB
Dart
49 lines
1.4 KiB
Dart
import "package:flutter/material.dart";
|
|
import "package:logging/logging.dart";
|
|
import "package:photos/models/file.dart";
|
|
import "package:photos/services/object_detection/object_detection_service.dart";
|
|
import "package:photos/ui/components/buttons/chip_button_widget.dart";
|
|
import "package:photos/ui/components/info_item_widget.dart";
|
|
import "package:photos/utils/thumbnail_util.dart";
|
|
|
|
class ObjectsItemWidget extends StatelessWidget {
|
|
final File file;
|
|
const ObjectsItemWidget(this.file, {super.key});
|
|
|
|
@override
|
|
Widget build(BuildContext context) {
|
|
return InfoItemWidget(
|
|
key: const ValueKey("Objects"),
|
|
leadingIcon: Icons.image_search_outlined,
|
|
subtitleSection: _objectTags(file),
|
|
hasChipButtons: true,
|
|
);
|
|
}
|
|
|
|
Future<List<ChipButtonWidget>> _objectTags(File file) async {
|
|
try {
|
|
final chipButtons = <ChipButtonWidget>[];
|
|
var objectTags = <String>[];
|
|
final thumbnail = await getThumbnail(file);
|
|
if (thumbnail != null) {
|
|
objectTags = await ObjectDetectionService.instance.predict(thumbnail);
|
|
}
|
|
if (objectTags.isEmpty) {
|
|
return const [
|
|
ChipButtonWidget(
|
|
"No results",
|
|
noChips: true,
|
|
)
|
|
];
|
|
}
|
|
for (String objectTag in objectTags) {
|
|
chipButtons.add(ChipButtonWidget(objectTag));
|
|
}
|
|
return chipButtons;
|
|
} catch (e, s) {
|
|
Logger("ObjctsItemWidget").info(e, s);
|
|
return [];
|
|
}
|
|
}
|
|
}
|