Update main.py

main.py

@@ -94,20 +94,20 @@ def main():
                 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)
+                # 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
-                #     )
+                # 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)