Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import pandas as pd
+from fpdf import FPDF
 
 # Interface utilisateur
 st.set_page_config(
@@ -102,7 +103,9 @@ if sentence:
     html = generate_html(pictogram_ids)
     st.components.v1.html(html, height=800, scrolling=True)
 
-    if st.button("Télécharger en PDF"):
+    # Container to hold the download button
+    download_container = st.container()
+    with download_container:
         pdf_path = generate_pdf(pictogram_ids)
         with open(pdf_path, "rb") as pdf_file:
-            st.download_button(label="Télécharger le PDF", data=pdf_file, file_name="pictograms.pdf", mime="application/pdf")
+            st.download_button(label="Télécharger la traduction en PDF", data=pdf_file, file_name="pictograms.pdf", mime="application/pdf")