Fixed issue with clear button

app.py

@@ -184,9 +184,9 @@ def respond(message, chat_history, temperature=0.9, top_p=0.6, max_token_count=5
 
     stream, prompt_and_time, suggestions = get_answer(message, formatted_history, temperature, top_p, max_token_count)
 
-    suggestion1_visible = bool(suggestions[0])
-    suggestion2_visible = bool(suggestions[1])
-    suggestion3_visible = bool(suggestions[2])
+    suggestion1_update = gr.update(visible=bool(suggestions[0]), value=suggestions[0])
+    suggestion2_update = gr.update(visible=bool(suggestions[1]), value=suggestions[1])
+    suggestion3_update = gr.update(visible=bool(suggestions[2]), value=suggestions[2])
 
     for idx, text_token in enumerate(stream):
         if idx == 0 and text_token.startswith(" "):
@@ -195,13 +195,11 @@ def respond(message, chat_history, temperature=0.9, top_p=0.6, max_token_count=5
         chat_history[-1][1] += text_token
         yield (
             "", chat_history, prompt_and_time, 
-            suggestions[0], suggestion1_visible, 
-            suggestions[1], suggestion2_visible, 
-            suggestions[2], suggestion3_visible
+            suggestion1_update, suggestion2_update, suggestion3_update
         )
     
 def clear_chat_history():
-    return '', []
+    return '', [], '', gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
 def main():
     with gr.Blocks(css="""
@@ -210,15 +208,11 @@ def main():
         }
     """) as demo:
         chatbot = gr.Chatbot([[None, initial_message]], height=600)
-        
-        suggestion1_visible = gr.State(value=True)
-        suggestion2_visible = gr.State(value=True)
-        suggestion3_visible = gr.State(value=True)
 
         with gr.Row():
-            suggestion1 = gr.Button("How to edit RFID crew badge?", elem_classes="suggestion-button", visible=suggestion1_visible)
-            suggestion2 = gr.Button("What types of RFID tags exist?", elem_classes="suggestion-button", visible=suggestion2_visible)
-            suggestion3 = gr.Button("Is it possible to delete a facility?", elem_classes="suggestion-button", visible=suggestion3_visible)
+            suggestion1 = gr.Button("How to edit RFID crew badge?", elem_classes="suggestion-button", visible=True)
+            suggestion2 = gr.Button("What types of RFID tags exist?", elem_classes="suggestion-button", visible=True)
+            suggestion3 = gr.Button("Is it possible to delete a facility?", elem_classes="suggestion-button", visible=True)
 
         msg = gr.Textbox(label="Question")
         
@@ -236,9 +230,7 @@ def main():
         ]
         outputs = [
             msg, chatbot, prompt_and_time,
-            suggestion1, suggestion1_visible,
-            suggestion2, suggestion2_visible,
-            suggestion3, suggestion3_visible
+            suggestion1, suggestion2, suggestion3
         ]
         
         btn.click(respond, inputs=inputs, outputs=outputs)
@@ -261,7 +253,10 @@ def main():
 
         msg.submit(respond, inputs=inputs, outputs=outputs)
     
-        clear.click(clear_chat_history, outputs=[msg, chatbot, prompt_and_time])
+        clear.click(clear_chat_history, outputs=[
+            msg, chatbot, prompt_and_time,
+            suggestion1, suggestion2, suggestion3
+        ])
     
     demo.queue().launch()