Update app.py

app.py

@@ -96,9 +96,32 @@ css="""
 """
 
 js="""
-document.addEventListener("load",function(e){
-    document.querySelector("div#prompt input").setAttribute(maxlength,"38")
-});
+function custom(){
+    window.Wait = function (Test, Success, Fail = function () { }, timeout = Number.MAX_SAFE_INTEGER) {
+    	let seconds = 0;
+    	function Internal() {
+    		if (!Test()) {
+    		if (seconds >= timeout) {
+    				Fail();
+    				return;
+    			}
+    			setTimeout(function () {
+    				seconds += 0.01;
+    				Internal(...arguments);
+    			}, 300);
+    			return;
+    	}
+    		Success();
+    	}
+    	Internal();
+    };
+    
+    Wait(function(){
+        return document.querySelector("div#prompt input")
+    },function(){
+        document.querySelector("div#prompt input").setAttribute(maxlength,"38");
+    },function(){});
+}
 """
 
 if torch.cuda.is_available():
@@ -106,9 +129,6 @@ if torch.cuda.is_available():
 else:
     power_device = "CPU"
 
-def limiter(txt):
-    return txt[:38]
-
 with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""