Update app.py

app.py

@@ -272,6 +272,12 @@ def change_prompt(state, prompt):
     if opt.verbose:
         log_state(state)
 
+    if state.is_running:
+        model.update_single_layer(
+            idx=state.current_palette,
+            prompt=prompt,
+        )
+
     return state
 
 
@@ -280,6 +286,12 @@ def change_neg_prompt(state, neg_prompt):
     if opt.verbose:
         log_state(state)
 
+    if state.is_running:
+        model.update_single_layer(
+            idx=state.current_palette,
+            neg_prompt=neg_prompt,
+        )
+
     return state
 
 
@@ -440,9 +452,9 @@ def show_element():
 
 
 def draw(state, drawpad):
-    # if not state.is_running:
-    #     print('[WARNING] Streaming is currently off, update ignored.')
-    #     return
+    if not state.is_running:
+        print('[WARNING] Streaming is currently off, update ignored.')
+        return
 
     # ZeroGPU hack.
     # conn = Client(opt.address, authkey=opt.authkey)