Update visualization.py

visualization.py

@@ -70,6 +70,20 @@ 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 addressing logistical barriers", 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 download_chart(self, fig, filename):
         buffer = io.BytesIO()
         fig.savefig(buffer, format='png')