Selaa lähdekoodia

Resolve object tags for local files as well

vishnukvmd 2 vuotta sitten
vanhempi
commit
fb0042ae03
2 muutettua tiedostoa jossa 13 lisäystä ja 2 poistoa
  1. 2 2
      lib/ui/viewer/file/object_tags_widget.dart
  2. 11 0
      lib/utils/thumbnail_util.dart

+ 2 - 2
lib/ui/viewer/file/object_tags_widget.dart

@@ -14,8 +14,8 @@ class ObjectTagsWidget extends StatelessWidget {
   @override
   @override
   Widget build(BuildContext context) {
   Widget build(BuildContext context) {
     return FutureBuilder<List<String>>(
     return FutureBuilder<List<String>>(
-      future: getThumbnailFromServer(file).then((data) {
-        return ObjectDetectionService.instance.predict(data);
+      future: getThumbnail(file).then((data) {
+        return ObjectDetectionService.instance.predict(data!);
       }),
       }),
       builder: (context, snapshot) {
       builder: (context, snapshot) {
         if (snapshot.hasData) {
         if (snapshot.hasData) {

+ 11 - 0
lib/utils/thumbnail_util.dart

@@ -32,6 +32,17 @@ class FileDownloadItem {
   FileDownloadItem(this.file, this.completer, this.cancelToken, this.counter);
   FileDownloadItem(this.file, this.completer, this.cancelToken, this.counter);
 }
 }
 
 
+Future<Uint8List?> getThumbnail(File file) async {
+  if (file.isRemoteFile) {
+    return getThumbnailFromServer(file);
+  } else {
+    return getThumbnailFromLocal(
+      file,
+      size: thumbnailLargeSize,
+    );
+  }
+}
+
 Future<Uint8List> getThumbnailFromServer(File file) async {
 Future<Uint8List> getThumbnailFromServer(File file) async {
   final cachedThumbnail = cachedThumbnailPath(file);
   final cachedThumbnail = cachedThumbnailPath(file);
   if (await cachedThumbnail.exists()) {
   if (await cachedThumbnail.exists()) {