Update app.py

app.py

@@ -325,7 +325,7 @@ Before deciding which model to use, you can expand `Example showcases` to see so
                 with gr.Column(elem_id="aux-btns-popup", visible=True):
                     with gr.Row():
                         stop = gr.Button("Stop", elem_classes=["aux-btn"], interactive=False)
-                        regenerate = gr.Button("Rege", interactive=False, elem_classes=["aux-btn"])
+                        regenerate = gr.Button("Regen", interactive=False, elem_classes=["aux-btn"])
                         clean = gr.Button("Clean", elem_classes=["aux-btn"])
 
                 with gr.Accordion("Context Inspector", elem_id="aux-viewer", open=False):
@@ -468,29 +468,57 @@ Lastly, there is a hidden panel on the top right corner, and it will appear when
             _js="(v)=>{ setStorage('local_data',v) }"
         )
 
+        instruction_txtbox.submit(
+            lambda: [
+                gr.update(visible=False),
+                gr.update(interactive=True)
+            ],
+            None,
+            [example_block, regenerate]
+        )
+        
+        send_event = instruction_txtbox.submit(
+            central.chat_stream,
+            [idx, local_data, instruction_txtbox, chat_state, model_num,
+            global_context, ctx_num_lconv, ctx_sum_prompt,
+            res_temp, res_topp, res_topk, res_rpen, res_mnts, res_beams, res_cache, res_sample, res_eosid, res_padid],
+            [instruction_txtbox, chatbot, context_inspector, local_data],
+        )
+        
+        instruction_txtbox.submit(
+            None, local_data, None, 
+            _js="(v)=>{ setStorage('local_data',v) }"
+        )        
+
         regenerate.click(
             rollback_last,
             [idx, local_data, chat_state],
             [instruction_txtbox, chatbot, local_data, regenerate]
-        ).then(
+        )
+        
+        regen_event = regenerate.click(
             central.chat_stream,
             [idx, local_data, instruction_txtbox, chat_state, model_num,
             global_context, ctx_num_lconv, ctx_sum_prompt,
             res_temp, res_topp, res_topk, res_rpen, res_mnts, res_beams, res_cache, res_sample, res_eosid, res_padid],
             [instruction_txtbox, chatbot, context_inspector, local_data],            
-        ).then(
+        )
+
+        regenerate.click(
             lambda: gr.update(interactive=True),
             None,
             regenerate
-        ).then(
+        )
+        
+        regenerate.click(
             None, local_data, None, 
             _js="(v)=>{ setStorage('local_data',v) }"  
         )
         
-        # stop.click(
-        #     None, None, None,
-        #     cancels=[send_event]
-        # )
+        stop.click(
+            None, None, None,
+            cancels=[send_event, regen_event]
+        )
 
         clean.click(
             reset_chat,