Update app.py

app.py

@@ -133,7 +133,9 @@ async def chat_stream(
         }
     ):
         ppm.append_pong(result)
-        yield "", prompt, ppm.build_uis(), str(res)
+        yield "", prompt, ppm.build_uis(), str(res), gr.update(interactive=False)
+
+        yield "", prompt, ppm.build_uis(), str(res), gr.update(interactive=True)
 
 def channel_num(btn_title):
     choice = 0
@@ -235,7 +237,7 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
 
                 with gr.Column(elem_id="aux-btns-popup", visible=True):
                     with gr.Row():
-                        stop = gr.Button("Stop", elem_classes=["aux-btn"])
+                        # stop = gr.Button("Stop", elem_classes=["aux-btn"])
                         regenerate = gr.Button("Regen", interactive=False, elem_classes=["aux-btn"])
                         clean = gr.Button("Clean", elem_classes=["aux-btn"])
 
@@ -309,7 +311,7 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
         chat_stream,
         [idx, local_data, instruction_txtbox, chat_state,
          global_context, res_temp, res_topk, res_rpen, res_mnts, res_sample, ctx_num_lconv],
-        [instruction_txtbox, context_inspector, chatbot, local_data]
+        [instruction_txtbox, context_inspector, chatbot, local_data, regenerate]
     ).then(
         None, local_data, None, 
         _js="(v)=>{ setStorage('local_data',v) }"