prasule/lib/views/settings/tessdata_list.dart
2023-12-25 19:03:52 +01:00

160 lines
6.3 KiB
Dart

import 'dart:async';
import 'dart:io';
import 'package:flutter/material.dart';
import 'package:flutter_tesseract_ocr/flutter_tesseract_ocr.dart';
import 'package:prasule/main.dart';
import 'package:prasule/network/tessdata.dart';
import 'package:prasule/pw/platformbutton.dart';
import 'package:prasule/pw/platformdialog.dart';
import 'package:flutter_gen/gen_l10n/app_localizations.dart';
class TessdataListView extends StatefulWidget {
const TessdataListView({super.key});
@override
State<TessdataListView> createState() => _TessdataListViewState();
}
class _TessdataListViewState extends State<TessdataListView> {
final _tessdata = [
{"eng": true}
];
@override
void didChangeDependencies() {
super.didChangeDependencies();
loadAllTessdata();
}
@override
Widget build(BuildContext context) {
return Scaffold(
appBar: AppBar(title: Text(AppLocalizations.of(context).ocrData)),
body: Center(
child: SizedBox(
width: MediaQuery.of(context).size.width * 0.9,
height: MediaQuery.of(context).size.height,
child: (_tessdata.length == 1)
? const Column(
mainAxisAlignment: MainAxisAlignment.center,
children: [
SizedBox(
width: 80,
height: 80,
child: CircularProgressIndicator(),
),
],
)
: ListView.builder(
itemBuilder: (context, i) => ListTile(
title: Text(_tessdata[i].keys.first),
trailing: TextButton(
child: Text(_tessdata[i][_tessdata[i].keys.first]!
? AppLocalizations.of(context).downloaded
: AppLocalizations.of(context).download),
onPressed: () async {
var lang = _tessdata[i].keys.first;
if (_tessdata[i][lang]!) {
// deleting data
showDialog(
context: context,
builder: (context) => PlatformDialog(
title: AppLocalizations.of(context).sureDialog,
content: Text(AppLocalizations.of(context)
.deleteOcr(lang)),
actions: [
PlatformButton(
text: AppLocalizations.of(context).yes,
onPressed: () async {
await TessdataApi.deleteData(lang);
_tessdata[i][lang] = true;
if (mounted) Navigator.of(context).pop();
},
),
PlatformButton(
text: AppLocalizations.of(context).no,
onPressed: () {
Navigator.of(context).pop();
},
),
],
),
);
setState(() {});
return;
}
// TODO: handle wifi errors
//* downloading traineddata
var progressStream = StreamController<double>();
showDialog(
context: context,
builder: (c) => PlatformDialog(
title: AppLocalizations.of(context)
.langDownloadDialog(lang),
content: StreamBuilder(
builder: (context, snapshot) {
if (snapshot.connectionState ==
ConnectionState.waiting) {
return const CircularProgressIndicator();
}
if (snapshot.hasError) {
return const Text("Error");
}
return Text(AppLocalizations.of(context)
.langDownloadProgress(snapshot.data!));
},
stream: progressStream.stream,
),
),
);
await TessdataApi.downloadData(lang, callback: (a, b) {
if (progressStream.isClosed) return;
var p = a / b * 1000;
progressStream.add(p.roundToDouble() / 10);
if (p / 10 >= 100) {
logger.i("Done");
Navigator.of(context, rootNavigator: true)
.pop("dialog");
_tessdata[i][lang] = true;
progressStream.close();
}
setState(() {});
});
},
),
),
itemCount: _tessdata.length,
),
),
),
);
}
/// Used to find which `.traineddata` is already downloaded and which not
/// so we can show it to the user
void loadAllTessdata() async {
var tessDir = Directory(await FlutterTesseractOcr.getTessdataPath());
var d = await TessdataApi.getAvailableData();
var dataStatus = <Map<String, bool>>[];
for (var data in d) {
var e = <String, bool>{};
e[data] = false;
dataStatus.add(e);
}
var appDir = tessDir.listSync();
for (var file in appDir) {
if (file is! File ||
!file.path.endsWith("traineddata") ||
file.path.endsWith("eng.traineddata")) continue;
logger.i(file.path);
var filename = file.path.split("/").last;
dataStatus[dataStatus.indexWhere((element) =>
element.keys.first == filename.replaceAll(".traineddata", ""))]
[filename.replaceAll(".traineddata", "")] = true;
}
_tessdata.addAll(dataStatus);
setState(() {});
}
}