Update app.py

app.py

@@ -133,10 +133,6 @@ STYLES = """
   text-align: justify !important;
 }
 
-.wrap {
-  display: contents !important;
-}
-
 .white-text {
   color: #000 !important;
 }
@@ -228,7 +224,8 @@ def gen_txt(editor):
       gr.update(interactive=True),
       gr.update(visible=False),
       gr.update(visible=False),
-      gr.update(visible=False)
+      gr.update(visible=False),
+      gr.update(interactive=True)
   ]
 
 def gen_txt_with_prompt(editor, prompt):
@@ -238,7 +235,8 @@ def gen_txt_with_prompt(editor, prompt):
       gr.update(interactive=True),
       gr.update(visible=False),
       gr.update(visible=False),
-      gr.update(visible=False)
+      gr.update(visible=False),
+      gr.update(interactive=True)
   ]
 
 def chat_gen(editor, chat_txt, chatbot, ppm, regen=False):
@@ -321,6 +319,7 @@ 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),
+      " "
   ]
 
 def fill_with_gen(alt_txt, editor):
@@ -349,6 +348,8 @@ with gr.Blocks(css=STYLES) as demo:
       "be changed over time, but [meta-llama/Llama-2-70b-chat-hf](https://huggingface.co/meta-llama/Llama-2-70b-chat-hf) is selected for now.",
       elem_classes=['center', 'small-big'])
 
+  progress_bar = gr.Textbox(elem_classes=['no-label'])
+
   with gr.Row():
     with gr.Column(scale=2):
       editor = gr.Textbox(lines=32, max_lines=32, elem_classes=['no-label', 'small-big-textarea'])
@@ -367,15 +368,15 @@ with gr.Blocks(css=STYLES) as demo:
           with gr.Column():
             with gr.Row(visible=False) as first_alt:
               gr.Markdown("↳", scale=1, elem_classes=['wrap'])
-              alt_btn1 = gr.Button("Alternative 1", elem_classes=['alt-button'], scale=5)
+              alt_btn1 = gr.Button("Alternative 1", elem_classes=['alt-button'], scale=8)
 
             with gr.Row(visible=False) as second_alt:
               gr.Markdown("↳", scale=1, elem_classes=['wrap'])
-              alt_btn2 = gr.Button("Alternative 2", elem_classes=['alt-button'], scale=5)
+              alt_btn2 = gr.Button("Alternative 2", elem_classes=['alt-button'], scale=8)
 
             with gr.Row(visible=False) as third_alt:
               gr.Markdown("↳", scale=1, elem_classes=['wrap'])
-              alt_btn3 = gr.Button("Alternative 3", elem_classes=['alt-button'], scale=5)
+              alt_btn3 = gr.Button("Alternative 3", elem_classes=['alt-button'], scale=8)
 
         with gr.Row(elem_classes=['group-border']):
           with gr.Column():
@@ -422,15 +423,23 @@ with gr.Blocks(css=STYLES) as demo:
   )
 
   gen_btn.click(
+      lambda: (gr.update(interactive=False), gr.update(interactive=False)),
+      inputs=None,
+      outputs=[gen_btn, gen_alt_btn]
+  ).then(
       fn=gen_txt,
       inputs=[editor],
-      outputs=[editor, num_enabled_alts, gen_alt_btn, first_alt, second_alt, third_alt]
+      outputs=[editor, num_enabled_alts, gen_alt_btn, first_alt, second_alt, third_alt, gen_btn]
   )
 
   gen_alt_btn.click(
+      lambda: gr.update(interactive=False),
+      inputs=None,
+      outputs=[gen_alt_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],
+      outputs=[num_enabled_alts, gen_alt_btn, first_alt, alt_btn1, second_alt, alt_btn2, third_alt, alt_btn3, progress_bar],
   )
 
   alt_btn1.click(
@@ -458,7 +467,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],
+      outputs=[editor, selected_text, sel_index_from, sel_index_to, progress_bar],
       show_progress='minimal'
   )