0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Skip model downloads over mobile data

Browse source
This commit is contained in:
vishnukvmd 2024-03-15 16:36:45 +05:30
parent 22685ae911
commit 7941d4bcb6
2 changed files with 14 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
mobile/lib/services/machine_learning/semantic_search/frameworks/ml_framework.dart
View file

@ -6,7 +6,6 @@ import "package:logging/logging.dart";
import "package:photos/core/errors.dart";
import "package:photos/core/event_bus.dart";
import "package:photos/core/network/network.dart";
import "package:photos/events/event.dart";
import "package:photos/services/remote_assets_service.dart";
@ -103,40 +102,35 @@ abstract class MLFramework {
return;
}
_initState = InitializationState.initializingImageModel;
final imageModel =
await RemoteAssetsService.instance.getAsset(getImageModelRemotePath());
final imageModel = await _getModel(getImageModelRemotePath());
await loadImageModel(imageModel.path);
_initState = InitializationState.initializedImageModel;
}
Future<void> _initTextModel() async {
_initState = InitializationState.initializingTextModel;
final textModel =
await RemoteAssetsService.instance.getAsset(getTextModelRemotePath());
final textModel = await _getModel(getTextModelRemotePath());
await loadTextModel(textModel.path);
_initState = InitializationState.initializedTextModel;
}
Future<void> _downloadFile(
String url,
String savePath, {
Future<File> _getModel(
String url, {
int trialCount = 1,
}) async {
if (await RemoteAssetsService.instance.hasAsset(url)) {
return RemoteAssetsService.instance.getAsset(url);
}
if (!await _canDownload()) {
_initState = InitializationState.waitingForNetwork;
throw WiFiUnavailableError();
}
_logger.info("Downloading " + url);
final existingFile = File(savePath);
if (await existingFile.exists()) {
await existingFile.delete();
}
try {
await NetworkClient.instance.getDio().download(url, savePath);
return RemoteAssetsService.instance.getAsset(url);
} catch (e, s) {
_logger.severe(e, s);
if (trialCount < kMaximumRetrials) {
return _downloadFile(url, savePath, trialCount: trialCount + 1);
return _getModel(url, trialCount: trialCount + 1);
} else {
rethrow;
}

5
mobile/lib/services/remote_assets_service.dart
View file

@ -26,6 +26,11 @@ class RemoteAssetsService {
}
}
Future<bool> hasAsset(String remotePath) async {
final path = await _getLocalPath(remotePath);
return File(path).exists();
}
Future<String> _getLocalPath(String remotePath) async {
return (await getApplicationSupportDirectory()).path +
"/assets/" +