Update app.py

app.py

@@ -285,7 +285,7 @@ def gen_art(editor, cover_art_image, gen_cover_art_prompt):
   else:
     global pipe
     pipe = pipe.to("cuda")
-    
+
     return [
         pipe(gen_cover_art_prompt).images[0],
         gen_cover_art_prompt
@@ -374,12 +374,24 @@ Answer to questions based on the written stories so far as below
   ]
 
 def chat(editor, chat_txt, chatbot, ppm):
-  return chat_gen(editor, chat_txt, chatbot, ppm, regen=False)
+  return chat_gen(editor, chat_txt, chatbot, ppm, regen=False) + [
+      gr.update(interactive=True),
+      gr.update(interactive=True)
+  ]
 
 def regen_chat(editor, chat_txt, chatbot, ppm):
   return chat_gen(editor, chat_txt, chatbot, ppm, regen=True)
 
 
+def clear_chat():
+  return [
+      "",
+      [],
+      get_new_ppm_for_chat(),
+      gr.update(interactive=False),
+      gr.update(interactive=False)
+  ]
+
 def get_new_ppm_for_range():
   ppm = LLaMA2ChatPPManager()
   ppm.ctx = """\
@@ -516,8 +528,8 @@ with gr.Blocks(css=STYLES) as demo:
           chat_txt = gr.Textbox(placeholder="enter question", elem_classes=['no-label'])
 
           with gr.Row():
-            clear_btn = gr.Button("clear", elem_classes=['control-label-font', 'control-button'])
-            regen_btn = gr.Button("regenerate", elem_classes=['control-label-font', 'control-button'])
+            clear_btn = gr.Button("clear", interactive=False, elem_classes=['control-label-font', 'control-button'])
+            regen_btn = gr.Button("regenerate", interactive=False, elem_classes=['control-label-font', 'control-button'])
 
       with gr.Tab("Exporting"):
         with gr.Column(elem_classes=['group-border']):
@@ -623,7 +635,7 @@ with gr.Blocks(css=STYLES) as demo:
   chat_txt.submit(
       fn=chat,
       inputs=[editor, chat_txt, chatbot, chat_history],
-      outputs=[chat_txt, chatbot, chat_history]
+      outputs=[chat_txt, chatbot, chat_history, clear_btn, regen_btn]
   )
 
   regen_btn.click(
@@ -631,5 +643,12 @@ with gr.Blocks(css=STYLES) as demo:
       inputs=[editor, chat_txt, chatbot, chat_history],
       outputs=[chat_txt, chatbot, chat_history]
   )
+
+  clear_btn.click(
+      fn=clear_chat,
+      inputs=None,
+      outputs=[chat_txt, chatbot, chat_history, clear_btn, regen_btn],
+      show_progress='minimal'
+  )
     
 demo.launch()