Spaces:

awacke1
/

Pillow-PyMuPDF-ReportLab

Sleeping

App Files Files Community

awacke1 commited on Mar 31

Commit

6d17d44

verified ·

1 Parent(s): 01087ae

Update app.py

Browse files

Files changed (1) hide show

app.py +35 -16

app.py CHANGED Viewed

@@ -1,7 +1,8 @@
 import streamlit as st
 import base64
-from weasyprint import HTML
 import io
 # Define the 12-point ML outline with emojis
 ml_outline = [
@@ -19,34 +20,36 @@ ml_outline = [
     "💻 12. ML Code Generation with Streamlit/Gradio/HTML5+JS"
 ]
-def create_pdf_with_weasyprint(outline_items):
     # Create HTML content with two columns
     html_content = """
     <html>
     <head>
         <style>
             body {
                 font-family: Arial, sans-serif;
                 margin: 20px;
             }
             .container {
                 display: flex;
                 width: 100%;
-                height: 500px;
             }
             .column {
                 width: 50%;
-                padding: 10px;
             }
             h2 {
                 color: #2c3e50;
             }
             ul {
                 list-style-type: none;
                 padding-left: 0;
             }
             li {
-                margin: 8px 0;
             }
         </style>
     </head>
@@ -81,16 +84,23 @@ def create_pdf_with_weasyprint(outline_items):
     </html>
     """
-    # Convert HTML to PDF
-    buffer = io.BytesIO()
-    HTML(string=html_content).write_pdf(buffer)
-    pdf_bytes = buffer.getvalue()
-    buffer.close()
     return pdf_bytes
-def get_binary_file_downloader_html(bin_file, file_label='File'):
-    bin_str = base64.b64encode(bin_file).decode()
     href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{file_label}">Download {file_label}</a>'
     return href
@@ -105,9 +115,9 @@ with col1:
     st.markdown(outline_text)
     md_file = "ml_outline.md"
-    with open(md_file, "w") as f:
         f.write(outline_text)
-    st.markdown(get_binary_file_downloader_html(md_file.encode(), "ml_outline.md"), unsafe_allow_html=True)
 with col2:
     st.header("📑 PDF Preview")
@@ -115,12 +125,13 @@ with col2:
     if st.button("Generate PDF"):
         with st.spinner("Generating PDF..."):
             try:
-                pdf_bytes = create_pdf_with_weasyprint(ml_outline)
                 # Save to file for download
                 with open("ml_outline.pdf", "wb") as f:
                     f.write(pdf_bytes)
                 st.download_button(
                     label="Download PDF",
                     data=pdf_bytes,
@@ -128,9 +139,17 @@ with col2:
                     mime="application/pdf"
                 )
                 base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
-                pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="400" type="application/pdf"></iframe>'
                 st.markdown(pdf_display, unsafe_allow_html=True)
             except Exception as e:
                 st.error(f"Error generating PDF: {str(e)}")

 import streamlit as st
 import base64
+import pdfkit
 import io
+import os
 # Define the 12-point ML outline with emojis
 ml_outline = [
     "💻 12. ML Code Generation with Streamlit/Gradio/HTML5+JS"
 ]
+def create_pdf_with_pdfkit(outline_items):
     # Create HTML content with two columns
     html_content = """
     <html>
     <head>
+        <meta charset="UTF-8">
         <style>
             body {
                 font-family: Arial, sans-serif;
+                font-size: 12pt;
                 margin: 20px;
             }
             .container {
                 display: flex;
                 width: 100%;
             }
             .column {
                 width: 50%;
+                padding: 20px;
             }
             h2 {
                 color: #2c3e50;
+                font-size: 16pt;
             }
             ul {
                 list-style-type: none;
                 padding-left: 0;
             }
             li {
+                margin: 10px 0;
             }
         </style>
     </head>
     </html>
     """
+    # Convert HTML to PDF with pdfkit
+    options = {
+        'page-size': 'A4',
+        'orientation': 'Landscape',
+        'encoding': 'UTF-8',
+        'margin-top': '0.5in',
+        'margin-right': '0.5in',
+        'margin-bottom': '0.5in',
+        'margin-left': '0.5in',
+    }
+    # Create PDF in memory
+    pdf_bytes = pdfkit.from_string(html_content, False, options=options)
     return pdf_bytes
+def get_binary_file_downloader_html(bin_data, file_label='File'):
+    bin_str = base64.b64encode(bin_data).decode()
     href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{file_label}">Download {file_label}</a>'
     return href
     st.markdown(outline_text)
     md_file = "ml_outline.md"
+    with open(md_file, "w", encoding='utf-8') as f:
         f.write(outline_text)
+    st.markdown(get_binary_file_downloader_html(outline_text.encode('utf-8'), "ml_outline.md"), unsafe_allow_html=True)
 with col2:
     st.header("📑 PDF Preview")
     if st.button("Generate PDF"):
         with st.spinner("Generating PDF..."):
             try:
+                pdf_bytes = create_pdf_with_pdfkit(ml_outline)
                 # Save to file for download
                 with open("ml_outline.pdf", "wb") as f:
                     f.write(pdf_bytes)
+                # Download button
                 st.download_button(
                     label="Download PDF",
                     data=pdf_bytes,
                     mime="application/pdf"
                 )
+                # Fix PDF viewer
                 base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+                pdf_display = f'''
+                    <embed
+                        src="data:application/pdf;base64,{base64_pdf}"
+                        width="100%"
+                        height="400px"
+                        type="application/pdf">
+                '''
                 st.markdown(pdf_display, unsafe_allow_html=True)
             except Exception as e:
                 st.error(f"Error generating PDF: {str(e)}")