Update main.py

main.py

@@ -98,8 +98,11 @@ def main():
             for index, row in student_metrics_df.iterrows():
                 tree_diagram = visualization.build_tree_diagram(row)
                 
-                # Use st.expander to wrap the graphviz chart
-                with st.expander(f"Student {index + 1} Decision Tree", expanded=False):
+                # Get the student's name from the DataFrame
+                student_name = row['Student']
+                
+                # Use st.expander to wrap the graphviz chart with the student's name
+                with st.expander(f"{student_name} Decision Tree", expanded=False):
                     st.graphviz_chart(tree_diagram.source)
             
             # Prepare input for the language model