Update pages/reader.py

pages/reader.py

@@ -2,6 +2,12 @@
 import streamlit as st
 import pandas as pd
 import html
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Frame, PageTemplate
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.units import cm
+from reportlab.lib.enums import TA_CENTER, TA_JUSTIFY
+import io
 
 
 #===config===
@@ -27,6 +33,111 @@ df = connect_gsheet()
 def format_html(text):
     return html.escape(text).replace("\n", "<br>")
 
+
+def create_pdf():
+    buffer = io.BytesIO()
+
+    # Document template
+    doc = SimpleDocTemplate(
+        buffer,
+        pagesize=A4,
+        rightMargin=2*cm, leftMargin=2*cm,
+        topMargin=2*cm, bottomMargin=2*cm
+    )
+
+    # Create styles
+    styles = getSampleStyleSheet()
+    styles.add(ParagraphStyle(
+        name="ArticleTitle",
+        alignment=TA_CENTER,
+        fontSize=28,
+        leading=32,
+        spaceAfter=6,
+        textColor="#222",
+        fontName="Times-Bold"
+    ))
+    styles.add(ParagraphStyle(
+        name="Author",
+        alignment=TA_CENTER,
+        fontSize=14,
+        leading=18,
+        textColor="#666",
+        italic=True,
+        spaceAfter=12
+    ))
+    styles.add(ParagraphStyle(
+        name="SectionTitle",
+        alignment=TA_CENTER,
+        fontSize=18,
+        leading=22,
+        spaceBefore=16,
+        spaceAfter=8,
+        textColor="#004488",
+        fontName="Times-Bold"
+    ))
+    styles.add(ParagraphStyle(
+        name="Meta",
+        alignment=TA_CENTER,
+        fontSize=fontsize,
+        leading=18,
+        italic=True,
+        textColor="#333",
+        spaceAfter=10
+    ))
+    styles.add(ParagraphStyle(
+        name="Content",
+        alignment=TA_JUSTIFY,
+        fontSize=fontsize,
+        leading=18,
+        textColor="#333"
+    ))
+
+    # Elements list
+    elements = []
+
+    # Title + Authors
+    elements.append(Paragraph(title, styles["ArticleTitle"]))
+    elements.append(Paragraph(f"By {author}", styles["Author"]))
+    elements.append(Paragraph(affiliation, styles["Author"]))
+
+    # Abstracts
+    elements.append(Paragraph("Abstract", styles["SectionTitle"]))
+    elements.append(Paragraph(abstracten, styles["Meta"]))
+
+    elements.append(Paragraph("Abstrak", styles["SectionTitle"]))
+    elements.append(Paragraph(abstractid, styles["Meta"]))
+    elements.append(Paragraph(f"Keywords: {keywords}", styles["Author"]))
+
+    # Sections
+    sections = [
+        ("Introduction", intro),
+        ("Method", method),
+        ("Result", result),
+        ("Discussion", discussion),
+        ("Result & Discussion", rnd),
+        ("Conclusion", conclusion),
+    ]
+
+    for sec_title, sec_content in sections:
+        elements.append(Paragraph(sec_title, styles["SectionTitle"]))
+        elements.append(Paragraph(sec_content, styles["Content"]))
+
+    # Multi-column layout
+    frame_width = (A4[0] - 4*cm) / no_columns
+    frames = [
+        Frame(2*cm + i*frame_width, 2*cm, frame_width, A4[1]-4*cm, leftPadding=0, rightPadding=0)
+        for i in range(no_columns)
+    ]
+    template = PageTemplate(frames=frames)
+    doc.addPageTemplates([template])
+
+    # Build PDF
+    doc.build(elements)
+    buffer.seek(0)
+    return buffer
+
+
+
 st.title('TXTperpus: Reader')
 
 paramx = st.query_params["art"]
@@ -48,6 +159,14 @@ if selection == "Baca Artikel":
     
         fontsize = st.number_input("Ukuran kata", min_value=5, value=16, step=1)
         no_columns = st.number_input("Jumlah kolom", min_value=1, max_value=4, value=1, step=1)
+
+        pdf_buffer = create_pdf()
+        st.download_button(
+            label="📄 Download as PDF",
+            data=pdf_buffer,
+            file_name="article.pdf",
+            mime="application/pdf"
+        )
     
     if not filtered_df.empty:
         filtered_df = filtered_df.replace("NaN", "Tidak tersedia")