Update app.py

app.py

@@ -21,32 +21,29 @@ def get_features(text: str) -> Dict:
     except Exception as e:
         return None
 
-def process_features(text: str) -> Tuple[List[gr.Tab], str]:
+def process_features(text: str) -> Tuple[Dict[str, list], str]:
+    """Returns a dictionary of token: [features] and dashboard HTML"""
     if not text:
-        return [], ""
+        return {}, ""
         
     features_data = get_features(text)
     if not features_data:
-        return [], ""
+        return {}, ""
     
-    tabs = []
+    token_features = {}
     for result in features_data['results']:
         if result['token'] == '<bos>':
             continue
             
         token = result['token']
-        features = result['top_features']
-        
-        with gr.Tab(token):
-            feature_list = []
-            for feature in features[:3]:
-                gr.Button(
-                    f"Feature {feature['feature_index']} (Activation: {feature['activation_value']:.2f})",
-                    variant="secondary"
-                )
-        tabs.append(gr.Tab)
+        features = []
+        for feature in result['top_features'][:3]:
+            features.append(
+                f"Feature {feature['feature_index']} (Activation: {feature['activation_value']:.2f})"
+            )
+        token_features[token] = features
 
-    # Create initial dashboard for first feature
+    # Create initial dashboard
     if features_data['results']:
         first_feature = features_data['results'][0]['top_features'][0]
         dashboard_html = create_dashboard({
@@ -56,7 +53,7 @@ def process_features(text: str) -> Tuple[List[gr.Tab], str]:
     else:
         dashboard_html = ""
     
-    return tabs, dashboard_html
+    return token_features, dashboard_html
 
 def create_dashboard(feature: Dict) -> str:
     if not feature:
@@ -118,16 +115,16 @@ def create_interface():
                 )
             
             with gr.Column(scale=2):
-                token_tabs = gr.Tabs()
+                features_text = gr.JSON()
                 dashboard = gr.HTML()
         
         analyze_btn.click(
             fn=process_features,
             inputs=[input_text],
-            outputs=[token_tabs, dashboard]
+            outputs=[features_text, dashboard]
         )
     
     return interface
 
 if __name__ == "__main__":
-    create_interface().launch()
+    create_interface().launch(share=True)