update app.py

app.py

@@ -160,18 +160,20 @@ start_work = """async() => {
         window['dalle_mini_iframe'] = null;
         for (var i = 0; i < tabitems.length; i++) {   
             if (tabitems_title[i].innerText.indexOf('SD') >= 0) {
-                tabitems[i].childNodes[0].children[0].style.display='none';
-                for (var j = 0; j < tabitems[i].childNodes[0].children[1].children.length; j++) {
-                    if (j != 1) {
-                        tabitems[i].childNodes[0].children[1].children[j].style.display='none';
+                if (tabitems[i].childNodes[0]) { 
+                    tabitems[i].childNodes[0].children[0].style.display='none';
+                    for (var j = 0; j < tabitems[i].childNodes[0].children[1].children.length; j++) {
+                        if (j != 1) {
+                            tabitems[i].childNodes[0].children[1].children[j].style.display='none';
+                        }
                     }
-                }
-                if (tabitems_title[i].innerText.indexOf('SD 1') >= 0) {
-                    for (var j = 0; j < 4; j++) {
-                        tabitems[i].childNodes[0].children[1].children[3].children[1].children[j].children[2].removeAttribute("disabled");
+                    if (tabitems_title[i].innerText.indexOf('SD 1') >= 0) {
+                        for (var j = 0; j < 4; j++) {
+                            tabitems[i].childNodes[0].children[1].children[3].children[1].children[j].children[2].removeAttribute("disabled");
+                        }
+                    } else if (tabitems_title[i].innerText.indexOf('SD 2') >= 0) {
+                        tabitems[i].children[0].children[1].children[3].children[0].click();
                     }
-                } else if (tabitems_title[i].innerText.indexOf('SD 2') >= 0) {
-                    tabitems[i].children[0].children[1].children[3].children[0].click();
                 }
             } else if (tabitems_title[i].innerText.indexOf('ernie') >= 0) {
                 tabitems[i].children[0].children[0].style.display='none';