Update pages/reader.py

pages/reader.py

@@ -2,11 +2,7 @@
 import streamlit as st
 import pandas as pd
 import html
-from reportlab.platypus import (SimpleDocTemplate, Paragraph, Spacer, Frame, PageTemplate, NextPageTemplate, PageBreak)
-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
+from xhtml2pdf import pisa
 import io
 
 
@@ -33,91 +29,13 @@ df = connect_gsheet()
 def format_html(text):
     return html.escape(text).replace("\n", "<br>")
 
-def create_pdf():
-    buffer = io.BytesIO()
-
-    doc = SimpleDocTemplate(
-        buffer, pagesize=A4,
-        rightMargin=2*cm, leftMargin=2*cm,
-        topMargin=2*cm, bottomMargin=2*cm
-    )
-
-    # 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"
-    ))
-
-    # Page templates
-    frame_full = Frame(2*cm, 2*cm, A4[0]-4*cm, A4[1]-4*cm, id="normal")
-    frame_col1 = Frame(2*cm, 2*cm, (A4[0]-4*cm)/2 - 0.5*cm, A4[1]-4*cm, id="col1")
-    frame_col2 = Frame(2*cm + (A4[0]-4*cm)/2 + 0.5*cm, 2*cm,
-                       (A4[0]-4*cm)/2 - 0.5*cm, A4[1]-4*cm, id="col2")
-
-    template_single = PageTemplate(id="OneCol", frames=[frame_full])
-    template_double = PageTemplate(id="TwoCol", frames=[frame_col1, frame_col2])
-
-    doc.addPageTemplates([template_double, template_single])
-
-    # Elements
-    elements = []
-
-    # --- Cover (single column) ---
-    elements.append(Paragraph(title, styles["ArticleTitle"]))
-    elements.append(Paragraph(f"By {author}", styles["Author"]))
-    elements.append(Paragraph(affiliation, styles["Author"]))
-    elements.append(Spacer(1, 12))
-
-    elements.append(Paragraph("Abstract", styles["SectionTitle"]))
-    elements.append(Paragraph(abstracten, styles["Meta"]))
-    elements.append(Spacer(1, 12))
-
-    elements.append(Paragraph("Abstrak", styles["SectionTitle"]))
-    elements.append(Paragraph(abstractid, styles["Meta"]))
-    elements.append(Paragraph(f"Keywords: {keywords}", styles["Author"]))
-
-    # Switch to 2-column template from next page
-    elements.append(NextPageTemplate("TwoCol"))
-    elements.append(PageBreak())
-
-    # --- Main content (2 columns) ---
-    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"]))
-
-    doc.build(elements)
-    buffer.seek(0)
-    return buffer
+def html_to_pdf(html_content: str) -> bytes:
+    result = io.BytesIO()
+    pisa_status = pisa.CreatePDF(io.StringIO(html_content), dest=result)
+    if pisa_status.err:
+        return None
+    result.seek(0)
+    return result.read()
 
 
 
@@ -160,13 +78,7 @@ 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 dalam PDF",
-            data=pdf_buffer,
-            file_name="article.pdf",
-            mime="application/pdf"
-        )
+        
     
     if not filtered_df.empty:
         
@@ -245,6 +157,14 @@ if selection == "Baca Artikel":
         """
     
         st.markdown(html_output, unsafe_allow_html=True)
+
+        pdf_bytes = html_to_pdf(html_output)
+        st.download_button(
+            "📄 Download PDF",
+            data=pdf_bytes,
+            file_name="article.pdf",
+            mime="application/pdf"
+        )
     
     else:
         st.warning("No article found with the specified ID.")