back to last promising version with treebuild crashlog, revising whole tree gen thing with Claude 7

app.py

@@ -535,10 +535,10 @@ def gradio_interface():
                 tree_generator = generate_definition_tree(base_prompt, feature_vector, model, tokenizer, config)
                 
                 log_output = []
+                tree_data = None
                 for item in tree_generator:
                     if isinstance(item, str):
                         log_output.append(item)
-                        yield item, None
                     else:
                         tree_data = item
                 
@@ -547,10 +547,10 @@ def gradio_interface():
                     max_weight, min_weight = find_max_min_cumulative_weight(tree_data)
                     tree_image = create_tree_diagram(tree_data, config, max_weight, min_weight)
                     logger.info("Tree image generated successfully")
-                    yield "\n".join(log_output), tree_image
+                    return "\n".join(log_output), tree_image
                 else:
                     logger.error("Failed to generate tree data")
-                    yield "Error: Failed to generate tree data.", None
+                    return "Error: Failed to generate tree data.", None
 
             return "Mode not recognized or not implemented in this step.", None
 
@@ -627,7 +627,7 @@ def gradio_interface():
             inputs=[selected_sae, feature_number, weight_type, use_token_centroid, scaling_factor, use_pca, pca_weight, num_exp, denom_exp, mode],
             outputs=[output_stream, output_image],
             show_progress="full"
-        ).then(lambda: gr.update(visible=False, value=""), None, [output_500_text])
+        )
 
         generate_top_500_btn.click(
             generate_top_500,