Just fucking around, still

app.py

@@ -24,84 +24,33 @@ def create_concept(name, pos_examples, neg_examples):
     except Exception as e:
         return f"❌ Error: {e}"
 
-def build_generation_ui():
-    """Returns prompt input, list of sliders, and output box."""
+def update_sliders():
     s = get_steerer()
     concepts = s.get_concept_names()
-    
-    prompt = gr.Textbox(label="Prompt", lines=2, placeholder="Enter your prompt...")
-    output = gr.Textbox(label="Output", lines=8, interactive=False)
-    
-    if not concepts:
-        sliders = [gr.Markdown("ℹ️ No concepts registered. Go to 'Create Concepts' tab.")]
-        return prompt, sliders, output
-    
-    sliders = []
-    for concept in concepts:
-        slider = gr.Slider(-5.0, 5.0, value=0.0, step=0.1, label=concept)
-        sliders.append(slider)
-    
-    return prompt, sliders, output
+    MAX = 10
+    updates = []
+    for i in range(MAX):
+        if i < len(concepts):
+            updates.append(gr.update(visible=True, label=concepts[i]))
+        else:
+            updates.append(gr.update(visible=False))
+    return updates
 
-def generate_text(prompt, *slider_values):
+def generate_text(prompt, *slider_vals):
     s = get_steerer()
     concepts = s.get_concept_names()
     steering = {}
-    for name, val in zip(concepts, slider_values):
-        if abs(val) > 1e-6:  # skip zero
+    for name, val in zip(concepts, slider_vals[:len(concepts)]):
+        if abs(val) > 1e-6:
             steering[name] = float(val)
     try:
         return s.generate(prompt, steering_config=steering, max_new_tokens=150)
     except Exception as e:
-        return f"❌ Generation failed: {e}"
+        return f"❌ Error: {e}"
 
-# Main interface
+# Build UI with fixed 10 sliders (hidden by default)
 with gr.Blocks() as demo:
-    gr.Markdown("# 🧠 LLM Concept Steering — Working Edition")
-    
-    with gr.Tab("Create Concepts"):
-        with gr.Row():
-            with gr.Column():
-                name_in = gr.Textbox(label="Concept Name")
-                pos_in = gr.Textbox(label="Positive Prompts (one per line)", lines=4)
-                neg_in = gr.Textbox(label="Negative Prompts (one per line)", lines=4)
-                create_btn = gr.Button("Register Concept")
-            with gr.Column():
-                status_out = gr.Textbox(label="Status", interactive=False)
-        create_btn.click(
-            create_concept,
-            inputs=[name_in, pos_in, neg_in],
-            outputs=status_out
-        )
-    
-    with gr.Tab("Generate"):
-        # We will dynamically replace this row
-        with gr.Row() as dynamic_row:
-            pass
-        
-        gen_btn = gr.Button("Generate")
-        output_box = gr.Textbox(label="Output", lines=8, interactive=False)
-
-        # Function to refresh the Generate tab UI
-        def refresh_ui():
-            prompt, sliders, _ = build_generation_ui()
-            return [prompt] + sliders
-
-        # Load initial UI
-        demo.load(
-            refresh_ui,
-            inputs=None,
-            outputs=None,
-            # We need to define the exact output components.
-            # So we pre-define a fixed max number of sliders (e.g., 10)
-        )
-
-# 🚨 CRITICAL: Gradio requires fixed output structure.
-# So we pre-define up to 10 sliders (more than enough for a demo).
-MAX_CONCEPTS = 10
-
-with gr.Blocks() as demo_fixed:
-    gr.Markdown("# 🧠 LLM Concept Steering — Working Edition")
+    gr.Markdown("# 🧠 LLM Concept Steering — Working Version")
     
     with gr.Tab("Create Concepts"):
         with gr.Row():
@@ -116,55 +65,28 @@ with gr.Blocks() as demo_fixed:
             create_concept,
             inputs=[name_in, pos_in, neg_in],
             outputs=status_out
-        ).then(  # After registering, refresh the other tab
-            None, None, None, 
-            _js="() => { document.querySelectorAll('.tabs button')[1].click(); }"  # Switch to Generate tab
         )
     
     with gr.Tab("Generate"):
-        prompt_input = gr.Textbox(label="Prompt", lines=2, placeholder="Enter your prompt...")
+        prompt_in = gr.Textbox(label="Prompt", lines=2, placeholder="e.g., Tell me a story.")
         
-        # Pre-define 10 optional sliders
+        # Pre-create 10 sliders (will be shown/hidden dynamically)
         sliders = []
-        for i in range(MAX_CONCEPTS):
+        for i in range(10):
             slider = gr.Slider(-5.0, 5.0, value=0.0, step=0.1, label=f"Concept {i+1}", visible=False)
             sliders.append(slider)
         
         gen_btn = gr.Button("Generate")
-        output_box = gr.Textbox(label="Output", lines=8, interactive=False)
-
-        def update_sliders():
-            s = get_steerer()
-            concepts = s.get_concept_names()
-            updates = []
-            for i in range(MAX_CONCEPTS):
-                if i < len(concepts):
-                    updates.append(gr.Slider(visible=True, label=concepts[i]))
-                else:
-                    updates.append(gr.Slider(visible=False))
-            return updates
-
-        def gen_wrapper(prompt, *slider_vals):
-            s = get_steerer()
-            concepts = s.get_concept_names()
-            steering = {
-                name: float(val) 
-                for name, val in zip(concepts, slider_vals[:len(concepts)])
-                if abs(val) > 1e-6
-            }
-            return s.generate(prompt, steering_config=steering, max_new_tokens=150)
-
-        # Update sliders on tab load
-        demo_fixed.load(update_sliders, None, sliders)
+        output_out = gr.Textbox(label="Output", lines=8, interactive=False)
         
-        # Connect generate
         gen_btn.click(
-            gen_wrapper,
-            inputs=[prompt_input] + sliders,
-            outputs=output_box
+            generate_text,
+            inputs=[prompt_in] + sliders,
+            outputs=output_out
         )
-
-demo = demo_fixed
+    
+    # Update sliders when the app loads or when tab is viewed
+    demo.load(update_sliders, inputs=None, outputs=sliders)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", share=False)