Update app.py

app.py

@@ -69,6 +69,26 @@ def generate_html(ids):
     html_content += '</body></html>'
     return html_content
 
+
+def generate_pdf(ids):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    
+    for picto_id, lemma in ids:
+        if picto_id != 0:  # ignore invalid IDs
+            img_url = f"https://static.arasaac.org/pictograms/{picto_id}/{picto_id}_500.png"
+            pdf.image(img_url, x=None, y=None, w=50, h=50)
+            pdf.ln(55)
+            pdf.set_font("Arial", size=12)
+            pdf.cell(200, 10, txt=lemma, ln=True, align='C')
+    
+    pdf_path = "pictograms.pdf"
+    pdf.output(pdf_path)
+    return pdf_path
+
+
+
 st.title("Traduction d'une phrase en pictogrammes ARASAAC")
 sentence = st.text_input("Entrez une phrase en français:")
 if sentence:
@@ -80,4 +100,9 @@ if sentence:
     pictogram_ids = [get_id_picto_from_predicted_lemma(lexicon, lemma) for lemma in sentence_to_map]
 
     html = generate_html(pictogram_ids)
-    st.components.v1.html(html, height=800, scrolling=True)
+    st.components.v1.html(html, height=800, scrolling=True)
+
+    if st.button("Télécharger en PDF"):
+        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")