Update app.py

app.py

@@ -223,9 +223,12 @@ chat_css = """
 }
 """
 
-# Function to update button classes
+# === Button state update function ===
 def update_alignment(choice):
-    return [gr.Button.update(variant="secondary", elem_classes=["alignment-btn", "selected"] if choice == c else ["alignment-btn"]) for c in alignment_choices], choice
+    return [
+        gr.Button.update(elem_classes=["alignment-btn", "selected"] if choice == c else ["alignment-btn"])
+        for c in alignment_choices
+    ] + [choice]  # Last item is selected_alignment output
 
 # === GUI ===
 with gr.Blocks(css=chat_css) as chatbot:
@@ -248,7 +251,7 @@ with gr.Blocks(css=chat_css) as chatbot:
                 btn.click(
                     fn=update_alignment,
                     inputs=[gr.State(choice)],
-                    outputs=[buttons, selected_alignment]
+                    outputs=buttons + [selected_alignment]  # ✅ Fixed here
                 )
 
             gr.ChatInterface(