Update main.py

main.py

@@ -88,14 +88,27 @@ def main():
             )
             st.subheader("Student Evaluations")
             st.write(student_metrics_df[['Student', 'Evaluation']])
-
-            st.table(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)
-
+                
+                # Render and prepare the diagram for download
+                filename = f"{row['Student']}_decision_tree"
+                filepath = download_tree_diagram(tree_diagram, filename)
+                
+                # Provide a download button
+                with open(filepath, "rb") as file:
+                    st.download_button(
+                        label=f"Download {row['Student']}'s Diagram",
+                        data=file.read(),
+                        file_name=f"{filename}.png",
+                        mime="image/png",
+                        icon="↹", 
+                        use_container_width=True
+                    )
+            
             # Prepare input for the language model
             llm_input = ai_analysis.prepare_llm_input(student_metrics_df)