Update visualization.py

visualization.py

@@ -73,21 +73,7 @@ class Visualization:
         st.pyplot(fig)
         return fig
 
-    # def build_tree_diagram(self, row):
-    #     dot = Digraph()
-    #     dot.node("Q1", "Has the student attended ≥ 90% of interventions?")
-    #     dot.node("Q2", "Has the student been engaged ≥ 80% of intervention time?")
-    #     dot.node("A1", "Address Attendance", shape="box")
-    #     dot.node("A2", "Address Engagement", shape="box")
-    #     dot.node("A3", "Consider barriers, fidelity, and progress monitoring", shape="box")
-    #     if row["Attended ≥ 90%"] == "No":
-    #         dot.edge("Q1", "A1", label="No")
-    #     else:
-    #         dot.edge("Q1", "Q2", label="Yes")
-    #         dot.edge("Q2", "A2" if row["Engagement ≥ 80%"] == "No" else "A3", label="Yes")
-    #     return dot
-
-    def build_tree_diagram(self, row):
+    def build_tree_diagram(row):
         dot = Digraph()
     
         # Add the student's name as the first node
@@ -119,6 +105,11 @@ class Visualization:
                 dot.edge("Q2", "A3", label="Yes")
     
         return dot
+    
+    def download_tree_diagram(dot, filename):
+        # Render the diagram to a file
+        filepath = dot.render(filename, format='png', cleanup=True)
+        return filepath
         
     def download_chart(self, fig, filename):
         buffer = io.BytesIO()