Update app.py

app.py

@@ -40,6 +40,7 @@ def respond(
 
     return response
 
+
 # CSS for styling the interface
 css = """
 body {
@@ -60,7 +61,8 @@ body {
 
 # Define the button click handler to hide the dropdown
 def on_button_click(model_name, message, history, system_message, dropdown_state):
-    return respond(model_name, message, history, system_message), gr.update(visible=False)  # Hide dropdown after click
+    # Close the dropdown by setting the state to False
+    return respond(model_name, message, history, system_message), gr.update(visible=False)
 
 # Define the Gradio interface with buttons and model selection
 demo = gr.Interface(
@@ -71,11 +73,11 @@ demo = gr.Interface(
         gr.Button("Chatgpt"),
         gr.Button("Llama"),
         gr.Button("Claude"),
-        gr.State()  # To track the state of dropdown visibility
+        gr.State(value=False),  # Add state to track visibility of the dropdown
     ],
     outputs=[
         "text",  # For the model's response
-        gr.Textbox(visible=False)  # Dropdown (this textbox will be hidden after the button is clicked)
+        gr.State(value=False),  # This should be the state output, which tracks visibility
     ],
     css=css,  # Pass the custom CSS here
 )