Update app.py

app.py

@@ -225,7 +225,9 @@ def gen_txt(editor):
       gr.update(visible=False),
       gr.update(visible=False),
       gr.update(visible=False),
-      gr.update(interactive=True)
+      gr.update(interactive=True),
+      gr.update(interactive=True),
+      gr.update(interactive=True),
   ]
 
 def gen_txt_with_prompt(editor, prompt):
@@ -236,7 +238,9 @@ def gen_txt_with_prompt(editor, prompt):
       gr.update(visible=False),
       gr.update(visible=False),
       gr.update(visible=False),
-      gr.update(interactive=True)
+      gr.update(interactive=True),
+      gr.update(interactive=True),
+      gr.update(interactive=True),
   ]
 
 def chat_gen(editor, chat_txt, chatbot, ppm, regen=False):
@@ -319,7 +323,10 @@ def gen_alt(editor, num_enabled_alts, alt_btn1, alt_btn2, alt_btn3):
       gr.update(value=gen_txt if num_enabled_alts == 1 else alt_btn2),
       gr.update(visible=True if num_enabled_alts >=2 else False),
       gr.update(value=gen_txt if num_enabled_alts == 2 else alt_btn3),
-      " "
+      " ",
+      gr.update(interactive=True),
+      gr.update(interactive=True),
+      gr.update(interactive=True),
   ]
 
 def fill_with_gen(alt_txt, editor):
@@ -395,10 +402,10 @@ with gr.Blocks(css=STYLES) as demo:
               replace_sel_btn = gr.Button("replace selection", elem_classes=['control-label-font', 'control-button'])
               replace_type = gr.Dropdown(choices=['word', 'sentense', 'phrase', 'paragraph'], value='sentense', interactive=True, elem_classes=['no-label'])
 
-          with gr.Row():
-            with gr.Column(elem_classes=['no-gap']):
-              rewrite_sel_btn = gr.Button("rewrite selection", elem_classes=['control-label-font', 'control-button'])
-              rewrite_prompt = gr.Textbox(placeholder="Rewrite the text: ", elem_classes=['no-label'])
+          # with gr.Row():
+          #   with gr.Column(elem_classes=['no-gap']):
+          #     rewrite_sel_btn = gr.Button("rewrite selection", elem_classes=['control-label-font', 'control-button'])
+          #     rewrite_prompt = gr.Textbox(placeholder="Rewrite the text: ", elem_classes=['no-label'])
 
       with gr.Tab("Chatting"):
         chatbot = gr.Chatbot([], elem_classes=['no-label-chatbot'])
@@ -423,23 +430,33 @@ with gr.Blocks(css=STYLES) as demo:
   )
 
   gen_btn.click(
-      lambda: (gr.update(interactive=False), gr.update(interactive=False)),
+      lambda: (
+          gr.update(interactive=False),
+          gr.update(interactive=False),
+          gr.update(interactive=False),
+          gr.update(interactive=False),
+      ),
       inputs=None,
-      outputs=[gen_btn, gen_alt_btn]
+      outputs=[gen_btn, gen_alt_btn, gen_with_prompt_btn, replace_sel_btn]
   ).then(
       fn=gen_txt,
       inputs=[editor],
-      outputs=[editor, num_enabled_alts, gen_alt_btn, first_alt, second_alt, third_alt, gen_btn]
+      outputs=[editor, num_enabled_alts, gen_alt_btn, first_alt, second_alt, third_alt, gen_btn, gen_with_prompt_btn, replace_sel_btn]
   )
 
   gen_alt_btn.click(
-      lambda: gr.update(interactive=False),
+      lambda: (
+          gr.update(interactive=False),
+          gr.update(interactive=False),
+          gr.update(interactive=False),
+          gr.update(interactive=False),
+      ),
       inputs=None,
-      outputs=[gen_alt_btn]
+      outputs=[gen_btn, gen_alt_btn, gen_with_prompt_btn, replace_sel_btn]
   ).then(
       fn=gen_alt,
       inputs=[editor, num_enabled_alts, alt_btn1, alt_btn2, alt_btn3],
-      outputs=[num_enabled_alts, gen_alt_btn, first_alt, alt_btn1, second_alt, alt_btn2, third_alt, alt_btn3, progress_bar],
+      outputs=[num_enabled_alts, gen_alt_btn, first_alt, alt_btn1, second_alt, alt_btn2, third_alt, alt_btn3, progress_bar, gen_btn, gen_with_prompt_btn, replace_sel_btn],
   )
 
   alt_btn1.click(
@@ -467,7 +484,7 @@ with gr.Blocks(css=STYLES) as demo:
   replace_sel_btn.click(
       fn=replace_sel,
       inputs=[editor, replace_type, selected_text, sel_index_from, sel_index_to],
-      outputs=[editor, selected_text, sel_index_from, sel_index_to, progress_bar],
+      outputs=[editor, selected_text, sel_index_from, sel_index_to],
       show_progress='minimal'
   )