Update main.py

main.py

@@ -93,21 +93,6 @@ def main():
             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 = visualization.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)