Update app.py

app.py

@@ -525,7 +525,7 @@ def download_chart(fig, filename):
     # Set the file pointer to the beginning
     buffer.seek(0)
     # Add a download button to Streamlit
-    st.download_button(label="Download Chart", data=buffer, file_name=filename, mime='image/png', icon="📊")
+    st.download_button(label="Download Chart", data=buffer, file_name=filename, mime='image/png', icon="📊", use_container_width=True)
 
 def download_llm_output(content, filename):
     # Create a buffer to hold the text data
@@ -533,7 +533,7 @@ def download_llm_output(content, filename):
     buffer.write(content.encode('utf-8'))
     buffer.seek(0)
     # Add a download button to Streamlit
-    st.download_button(label="Download LLM Output", data=buffer, file_name=filename, mime='text/plain')
+    st.download_button(label="Download LLM Output", data=buffer, file_name=filename, mime='text/plain', icon="✏️", use_container_width=True)
 
 def prepare_llm_input(student_metrics_df):
     # Convert the student metrics DataFrame to a string