mirrors/pdf.js
1
0
Fork 0
mirror of https://github.com/mozilla/pdf.js.git synced 2025-04-22 16:18:08 +02:00
Code Activity

Load the image-to-text model when opening the pdf viewer in Firefox (bug 1908938)

Browse source
This commit is contained in:
Calixte Denizet 2024-07-19 17:57:53 +02:00
parent 5b0e15ab18
commit b71fa727e1
5 changed files with 62 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/display/editor/stamp.js
View file

@ -430,7 +430,8 @@ class StampEditor extends AnnotationEditor {
return;
}
this.#hasMLBeenQueried = true;
if (!this._uiManager.isMLEnabledFor("altText") || this.hasAltText()) {
const isMLEnabled = await this._uiManager.isMLEnabledFor("altText");
if (!isMLEnabled || this.hasAltText()) {
return;
}
const offscreen = new OffscreenCanvas(width, height);
@ -447,7 +448,7 @@ class StampEditor extends AnnotationEditor {
height
);
const response = await this._uiManager.mlGuess({
service: "image-to-text",
service: "moz-image-to-text",
request: {
data: ctx.getImageData(0, 0, width, height).data,
width,

4
src/display/editor/tools.js
View file

@ -855,8 +855,8 @@ class AnnotationEditorUIManager {
return this.#mlManager?.guess(data) || null;
}
isMLEnabledFor(name) {
return !!this.#mlManager?.isEnabledFor(name);
async isMLEnabledFor(name) {
return !!(await this.#mlManager?.isEnabledFor(name));
}
get useNewAltTextFlow() {