Update app.py

app.py

@@ -470,11 +470,10 @@ Lastly, there is a hidden panel on the top right corner, and it will appear when
 
         instruction_txtbox.submit(
             lambda: [
-                gr.update(visible=False),
-                gr.update(interactive=True)
+                gr.update(visible=False)
             ],
             None,
-            [example_block, regenerate]
+            [example_block]
         )
         
         send_event = instruction_txtbox.submit(
@@ -488,7 +487,14 @@ Lastly, there is a hidden panel on the top right corner, and it will appear when
         instruction_txtbox.submit(
             None, local_data, None, 
             _js="(v)=>{ setStorage('local_data',v) }"
-        )        
+        ).then(
+            lambda: [
+                gr.update(interactive=True),
+                gr.update(interactive=True),    
+            ],
+            None,
+            [regenerate, stop]
+        )
 
         regenerate.click(
             rollback_last,
@@ -513,6 +519,13 @@ Lastly, there is a hidden panel on the top right corner, and it will appear when
         regenerate.click(
             None, local_data, None, 
             _js="(v)=>{ setStorage('local_data',v) }"  
+        ).then(
+            lambda: [
+                gr.update(interactive=True),
+                gr.update(interactive=True),    
+            ],
+            None,
+            [regenerate, stop]
         )
         
         stop.click(