Update main.py

main.py

@@ -166,7 +166,7 @@ def main():
             
             # Build and display decision tree diagrams for each student
             for index, row in student_metrics_df.iterrows():
-                tree_diagram = data_processor.build_tree_diagram(row)
+                tree_diagram = visualization.build_tree_diagram(row)
                 st.graphviz_chart(tree_diagram.source)
 
             # Prepare input for the language model