update

app.py

@@ -247,26 +247,35 @@ def create_gradio_interface():
     def analyze(text):
         results = analyzer.analyze_content(text)
 
-        output = "Content Analysis Results\n\n"
+        output = "# Content Analysis Results\n\n"
 
-        output += "Category Scores:\n"
+        output += "## Category Scores\n"
         for category, features in results["categories"].items():
             if features:
                 avg_score = np.mean([f["activation_score"] for f in features])
-                output += f"{category.title()}: {avg_score:.2f}\n"
+                output += f"**{category.title()}**: {avg_score:.2f}\n"
 
-        output += "\nFeature Details:\n"
+        output += "\n## Feature Details\n"
         for feature_id, feature in results["features"].items():
-            output += f"\n{feature['name']}:\n"
-            output += f"Score: {feature['activation_score']:.2f}\n"
-            output += f"Interpretation: {feature['interpretation']}\n"
+            output += f"\n### {feature['name']} (Feature {feature_id})\n"
+            output += f"**Score**: {feature['activation_score']:.2f}\n\n"
+            output += f"**Interpretation**: {feature['interpretation']}\n\n"
+            # Add feature explanation from Neuronpedia reference
+            output += f"[View feature details on Neuronpedia](https://neuronpedia.org/gemma-2-2b/20-gemmascope-res-16k/{feature_id})\n\n"
 
         if results["recommendations"]:
-            output += "\nRecommendations:\n"
+            output += "\n## Recommendations\n"
             for rec in results["recommendations"]:
                 output += f"- {rec}\n"
 
-        return output
+        feature_id = max(
+            results["features"].items(), key=lambda x: x[1]["activation_score"]
+        )[0]
+
+        # Build dashboard URL for the highest activating feature
+        dashboard_url = f"https://neuronpedia.org/gemma-2-2b/20-gemmascope-res-16k/{feature_id}?embed=true&embedexplanation=true&embedplots=true&embedtest=true&height=300"
+
+        return output, dashboard_url, feature_id
 
     with gr.Blocks(
         theme=gr.themes.Default(
@@ -282,23 +291,43 @@ def create_gradio_interface():
         )
 
         with gr.Row():
-            input_text = gr.Textbox(
-                lines=5,
-                placeholder="Enter your marketing content here...",
-                label="Marketing Content",
-            )
-            output_text = gr.Textbox(label="Analysis Results")
+            with gr.Column(scale=1):
+                input_text = gr.Textbox(
+                    lines=5,
+                    placeholder="Enter your marketing content here...",
+                    label="Marketing Content",
+                )
+                analyze_btn = gr.Button("Analyze", variant="primary")
+                gr.Examples(
+                    examples=[
+                        "WordLift is an AI-powered SEO tool",
+                        "Our advanced machine learning algorithms optimize your content",
+                        "Simple and effective website optimization",
+                    ],
+                    inputs=input_text,
+                )
 
-        analyze_btn = gr.Button("Analyze", variant="primary")
-        analyze_btn.click(fn=analyze, inputs=input_text, outputs=output_text)
+            with gr.Column(scale=2):
+                output_text = gr.Markdown(label="Analysis Results")
+                with gr.Box():
+                    gr.Markdown("## Feature Dashboard")
+                    feature_id_text = gr.Text(
+                        label="Currently viewing feature", show_label=False
+                    )
+                    dashboard_frame = gr.HTML(label="Feature Dashboard")
+
+        def update_dashboard(text):
+            output, dashboard_url, feature_id = analyze(text)
+            return (
+                output,
+                f"<iframe src='{dashboard_url}' width='100%' height='600px' frameborder='0'></iframe>",
+                f"Currently viewing Feature {feature_id} - Most active feature in your content",
+            )
 
-        gr.Examples(
-            examples=[
-                "WordLift is an AI-powered SEO tool",
-                "Our advanced machine learning algorithms optimize your content",
-                "Simple and effective website optimization",
-            ],
+        analyze_btn.click(
+            fn=update_dashboard,
             inputs=input_text,
+            outputs=[output_text, dashboard_frame, feature_id_text],
         )
 
     return interface