hande made

app.py

@@ -155,31 +155,23 @@ def add_concept_to_list(selected_concept, user_slider_val, current_list):
         "internal_mag": internal_mag,
     }
     # Add to the beginning of the list
-    updated_list = [new_entry] + current_list
-    return updated_list, gr.update(choices=_build_remove_choices(updated_list))
-
-def remove_concept_from_list(selected_text, current_list):
-    if not selected_text:
-        return current_list, gr.update(choices=_build_remove_choices(current_list))
-    
-    # Remove based on the full formatted text
-    updated_list = [x for x in current_list if f"(+{x['display_mag']:.1f}*) {x['text']}" != selected_text]
-    return updated_list, gr.update(choices=_build_remove_choices(updated_list))
+    current_list = [new_entry]
+    return current_list
 
 def update_dropdown_choices(search_text):
     filtered = filter_concepts(search_text)
     return gr.update(choices=filtered)
 
-with gr.Blocks(theme='allenai/gradio-theme') as demo:
+with gr.Blocks(fill_height=True) as demo:
     # Remove default subspaces
     selected_subspaces = gr.State([])
     
-    with gr.Row():
+    with gr.Row(min_height=700):
         # Left side: bigger chat area
         with gr.Column(scale=7):
             chat_interface = gr.ChatInterface(
                 fn=generate,
-                title="Language Model Concept Steering",
+                title="Chat with a Concept Steering Model",
                 description="Steer responses by selecting concepts on the right →",
                 type="messages",
                 additional_inputs=[selected_subspaces],
@@ -189,7 +181,7 @@ with gr.Blocks(theme='allenai/gradio-theme') as demo:
         # Right side: concept management
         with gr.Column(scale=3):
             gr.Markdown("## Steer Model Responses")
-            gr.Markdown("Search and then add/remove concept.")
+            gr.Markdown("Search and then select a concept to steer the model.")
             # Concept Search and Selection
             with gr.Group():
                 search_box = gr.Textbox(
@@ -198,7 +190,7 @@ with gr.Blocks(theme='allenai/gradio-theme') as demo:
                     lines=2,
                 )
                 concept_dropdown = gr.Dropdown(
-                    label="Select a Concept",
+                    label="Select a concept to steer the model",
                     interactive=True,
                 )
                 concept_magnitude = gr.Slider(
@@ -208,17 +200,6 @@ with gr.Blocks(theme='allenai/gradio-theme') as demo:
                     step=0.1,  # Allow 1 decimal point
                     value=3,
                 )
-                add_button = gr.Button("Add Concept to Steering")
-
-            # Current Steering Concepts
-            gr.Markdown("## Current Steering Concepts")
-            with gr.Group():
-                remove_dropdown = gr.Dropdown(
-                    label="Select a Current Steering Concept to Stop Steering",
-                    choices=[],
-                    multiselect=False,
-                )
-                remove_button = gr.Button("Remove Current Steering Concept", variant="secondary")
 
     # Wire up events
     # When the search box changes, update the concept dropdown choices:
@@ -228,20 +209,16 @@ with gr.Blocks(theme='allenai/gradio-theme') as demo:
         [concept_dropdown]
     )
 
-    # When "Add Concept" is clicked, add the concept + magnitude to the list,
-    # and update the "Remove" dropdown choices.
-    add_button.click(
+    concept_dropdown.select(
         add_concept_to_list,
         [concept_dropdown, concept_magnitude, selected_subspaces],
-        [selected_subspaces, remove_dropdown]
+        [selected_subspaces]
     )
 
-    # When "Remove" is clicked, remove the selected concept from the list,
-    # and update the "Remove" dropdown choices.
-    remove_button.click(
-        remove_concept_from_list,
-        [remove_dropdown, selected_subspaces],
-        [selected_subspaces, remove_dropdown]
+    concept_magnitude.input(
+        add_concept_to_list,
+        [concept_dropdown, concept_magnitude, selected_subspaces],
+        [selected_subspaces]
     )
 
     demo.launch(share=True)