Update main.py

main.py

@@ -89,10 +89,18 @@ def main():
             st.subheader("Student Evaluations")
             st.write(student_metrics_df[['Student', 'Evaluation']])
             
+            # # Build and display decision tree diagrams for each student
+            # for index, row in student_metrics_df.iterrows():
+            #     tree_diagram = visualization.build_tree_diagram(row)
+            #     st.graphviz_chart(tree_diagram.source)
+
             # Build and display decision tree diagrams for each student
             for index, row in student_metrics_df.iterrows():
                 tree_diagram = visualization.build_tree_diagram(row)
-                st.graphviz_chart(tree_diagram.source)
+                
+                # Use st.expander to wrap the graphviz chart
+                with st.expander(f"Student {index + 1} Decision Tree", expanded=False):
+                    st.graphviz_chart(tree_diagram.source)
             
             # Prepare input for the language model
             llm_input = ai_analysis.prepare_llm_input(student_metrics_df)