Update app.py

app.py

@@ -5,7 +5,11 @@ import json
 import time
 from load_llms import model_choices, llm_intro, load_model
 
+def clear_output(message, chatbot):
+    message.update(value="")
+    chatbot.update(value="")
 
+    
 # Construct chatbot
 def generate_response(model_name, message, chat_history):
     model = load_model(model_name)
@@ -19,7 +23,7 @@ with gr.Blocks(
     "#chatbot { flex-grow: 1 !important; overflow: auto !important; }"
     "#col { height: calc(100vh - 112px - 16px) !important; }"
     "#submit:hover { background-color: green !important; }"
-    "#clear_button { background-color: red !important; }",
+    "#clear:hover { background-color: red !important; }",
     theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Martel Sans")])                   
 ) as demo:
     gr.Markdown("# GPT4All Chatbot")
@@ -46,7 +50,9 @@ with gr.Blocks(
                 submit_button = gr.Button("Submit", scale=1, elem_id="submit")
                 submit_button.click(generate_response, inputs=[model_dropdown, message, chatbot], outputs=[message, chatbot])
 
-            clear = gr.ClearButton([message, chatbot], elem_id="clear_button")
+            # clear = gr.ClearButton([message, chatbot], elem_id="clear")
+            clear_button = gr.Button("Clear", elem_id="clear")
+            clear_button.click(clear_output, inputs=[message, chatbot], outputs=[message, chatbot])
 
 # Launch the Gradio app
 demo.launch()