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)