back to basics

app.py

@@ -382,7 +382,7 @@ def process_input(selected_sae, feature_number, weight_type, use_token_centroid,
             
             if top_500:
                 # Generate the top 500 list 
-                result = ", ".join([f"{token!r}: {value:.4f}" for token, value in closest_tokens_with_values])
+                result = ", ".join([f"'{token}': {value:.4f}" for token, value in closest_tokens_with_values])
                 logger.info("Returning top 500 list")
                 return result, None
             else:
@@ -487,14 +487,14 @@ def gradio_interface():
             inputs=inputs,
             outputs=[output_text, output_image],
             show_progress="full"
-        )
+        ).then(lambda: (gr.update(visible=False), gr.update(value="")), None, [generate_top_500_btn, output_500_text])
 
         generate_top_500_btn.click(
             generate_top_500,
             inputs=inputs,
             outputs=[output_500_text],
             show_progress="full"
-        )
+        ).then(lambda: gr.update(visible=True), None, [output_500_text])
         
         trim_btn.click(trim_tree, inputs=[trim_slider, tree_data_state], outputs=[output_image])
 
@@ -504,7 +504,7 @@ def gradio_interface():
         feature_number.change(update_neuronpedia, inputs=[selected_sae, feature_number], outputs=[neuronpedia_html])
 
         output_text.change(
-            lambda text: (gr.update(visible=True), gr.update(visible=True)) if "100 tokens" in text else (gr.update(visible(False)), gr.update(visible(False))),
+            lambda text: (gr.update(visible=True), gr.update(visible=True)) if "100 tokens" in text else (gr.update(visible=False), gr.update(visible=False)),
             inputs=[output_text],
             outputs=[generate_top_500_btn, output_500_text]
         )
@@ -514,7 +514,6 @@ def gradio_interface():
 
 
 
-
 if __name__ == "__main__":
     try:
         logger.info("Starting application initialization...")