Commit e053e21a authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

put localStorage stuff into its own file

parent a32f270a
Loading
Loading
Loading
Loading
+4 −0
Original line number Original line Diff line number Diff line
@@ -87,5 +87,9 @@ module.exports = {
        modalNextImage: "readonly",
        modalNextImage: "readonly",
        // token-counters.js
        // token-counters.js
        setupTokenCounters: "readonly",
        setupTokenCounters: "readonly",
        // localStorage.js
        localSet: "readonly",
        localGet: "readonly",
        localRemove: "readonly"
    }
    }
};
};
+26 −0
Original line number Original line Diff line number Diff line

function localSet(k, v) {
    try {
        localStorage.setItem(k, v);
    } catch (e) {
        console.warn(`Failed to save ${k} to localStorage: ${e}`);
    }
}

function localGet(k, def) {
    try {
        return localStorage.getItem(k);
    } catch (e) {
        console.warn(`Failed to load ${k} from localStorage: ${e}`);
    }

    return def;
}

function localRemove(k) {
    try {
        return localStorage.removeItem(k);
    } catch (e) {
        console.warn(`Failed to remove ${k} from localStorage: ${e}`);
    }
}
+8 −16
Original line number Original line Diff line number Diff line
@@ -152,15 +152,11 @@ function submit() {
    showSubmitButtons('txt2img', false);
    showSubmitButtons('txt2img', false);


    var id = randomId();
    var id = randomId();
    try {
    localSet("txt2img_task_id", id);
        localStorage.setItem("txt2img_task_id", id);
    } catch (e) {
        console.warn(`Failed to save txt2img task id to localStorage: ${e}`);
    }


    requestProgress(id, gradioApp().getElementById('txt2img_gallery_container'), gradioApp().getElementById('txt2img_gallery'), function() {
    requestProgress(id, gradioApp().getElementById('txt2img_gallery_container'), gradioApp().getElementById('txt2img_gallery'), function() {
        showSubmitButtons('txt2img', true);
        showSubmitButtons('txt2img', true);
        localStorage.removeItem("txt2img_task_id");
        localRemove("txt2img_task_id");
        showRestoreProgressButton('txt2img', false);
        showRestoreProgressButton('txt2img', false);
    });
    });


@@ -175,15 +171,11 @@ function submit_img2img() {
    showSubmitButtons('img2img', false);
    showSubmitButtons('img2img', false);


    var id = randomId();
    var id = randomId();
    try {
    localSet("img2img_task_id", id);
        localStorage.setItem("img2img_task_id", id);
    } catch (e) {
        console.warn(`Failed to save img2img task id to localStorage: ${e}`);
    }


    requestProgress(id, gradioApp().getElementById('img2img_gallery_container'), gradioApp().getElementById('img2img_gallery'), function() {
    requestProgress(id, gradioApp().getElementById('img2img_gallery_container'), gradioApp().getElementById('img2img_gallery'), function() {
        showSubmitButtons('img2img', true);
        showSubmitButtons('img2img', true);
        localStorage.removeItem("img2img_task_id");
        localRemove("img2img_task_id");
        showRestoreProgressButton('img2img', false);
        showRestoreProgressButton('img2img', false);
    });
    });


@@ -197,7 +189,7 @@ function submit_img2img() {


function restoreProgressTxt2img() {
function restoreProgressTxt2img() {
    showRestoreProgressButton("txt2img", false);
    showRestoreProgressButton("txt2img", false);
    var id = localStorage.getItem("txt2img_task_id");
    var id = localGet("txt2img_task_id");


    if (id) {
    if (id) {
        requestProgress(id, gradioApp().getElementById('txt2img_gallery_container'), gradioApp().getElementById('txt2img_gallery'), function() {
        requestProgress(id, gradioApp().getElementById('txt2img_gallery_container'), gradioApp().getElementById('txt2img_gallery'), function() {
@@ -211,7 +203,7 @@ function restoreProgressTxt2img() {
function restoreProgressImg2img() {
function restoreProgressImg2img() {
    showRestoreProgressButton("img2img", false);
    showRestoreProgressButton("img2img", false);


    var id = localStorage.getItem("img2img_task_id");
    var id = localGet("img2img_task_id");


    if (id) {
    if (id) {
        requestProgress(id, gradioApp().getElementById('img2img_gallery_container'), gradioApp().getElementById('img2img_gallery'), function() {
        requestProgress(id, gradioApp().getElementById('img2img_gallery_container'), gradioApp().getElementById('img2img_gallery'), function() {
@@ -224,8 +216,8 @@ function restoreProgressImg2img() {




onUiLoaded(function() {
onUiLoaded(function() {
    showRestoreProgressButton('txt2img', localStorage.getItem("txt2img_task_id"));
    showRestoreProgressButton('txt2img', localGet("txt2img_task_id"));
    showRestoreProgressButton('img2img', localStorage.getItem("img2img_task_id"));
    showRestoreProgressButton('img2img', localGet("img2img_task_id"));
});
});