Update app.py

app.py

@@ -108,21 +108,34 @@ body { font-family: 'Open Sans', sans-serif !important; }
 }
 .model-selector {
     display: flex;
-    gap: 1rem;
+    gap: 0.5rem;
     margin-bottom: 1rem;
 }
+.model-button {
+    padding: 0.25rem 0.75rem !important;
+    min-width: 0 !important;
+    height: 32px !important;
+    font-size: 0.875rem !important;
+}
 """
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.Markdown("# Brand Analyzer", elem_classes="text-2xl font-bold mb-2")
     gr.Markdown("*Analyze text using interpretable neural features*", elem_classes="text-gray-600 mb-6")
     
+    current_model = gr.State("Gemini")
     features_state = gr.State([])
-    selected_model = gr.State("Gemini")  # Default to Gemini
     
     with gr.Row(elem_classes="model-selector"):
-        gemini_btn = gr.Button("🧬 Gemini", variant="primary" if selected_model.value == "Gemini" else "secondary")
-        openai_btn = gr.Button("🤖 OpenAI", variant="secondary")
+        with gr.Column(scale=1):
+            model_choice = gr.Radio(
+                choices=["Gemini", "OpenAI"],
+                value="Gemini",
+                label="Select Model",
+                elem_classes="model-buttons",
+                container=False,
+                interactive=True
+            )
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -138,46 +151,27 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             )
         
         with gr.Column(scale=2):
-            @gr.render(inputs=[features_state, selected_model])
-            def render_features(features, current_model):
-                if not features:
-                    return
-                
-                model = Model.GEMMA if current_model == "Gemini" else Model.GPT2
-                
-                for token_group in features:
-                    gr.Markdown(f"### {token_group['token']}")
-                    with gr.Row():
-                        for feature in token_group['features']:
-                            btn = gr.Button(
-                                f"Feature {feature['id']} (Activation: {feature['activation']:.2f})",
-                                elem_classes=["feature-button"]
-                            )
-                            btn.click(
-                                fn=lambda fid=feature['id']: handle_feature_click(fid, model),
-                                outputs=dashboard
-                            )
-            
+            features_list = gr.JSON(label="Features", visible=False)
             dashboard = gr.HTML()
     
-    def update_model(new_model):
-        return new_model
+    def update_and_analyze(text, model):
+        return analyze_text(text, model)
     
-    gemini_btn.click(
-        fn=lambda: update_model("Gemini"),
-        outputs=selected_model,
-        queue=False
+    model_choice.change(
+        fn=lambda x: x,
+        inputs=[model_choice],
+        outputs=[current_model]
     )
     
-    openai_btn.click(
-        fn=lambda: update_model("OpenAI"),
-        outputs=selected_model,
-        queue=False
+    analyze_btn.click(
+        fn=update_and_analyze,
+        inputs=[input_text, current_model],
+        outputs=[features_state, dashboard]
     )
     
-    analyze_btn.click(
-        fn=analyze_text,
-        inputs=[input_text, selected_model],
+    input_text.submit(
+        fn=update_and_analyze,
+        inputs=[input_text, current_model],
         outputs=[features_state, dashboard]
     )