Spaces:

awacke1
/

Pillow-PyMuPDF-ReportLab

Sleeping

App Files Files Community

awacke1 commited on Mar 31

Commit

3edd9d1

verified ·

1 Parent(s): 0aea23e

Update app.py

Browse files

Files changed (1) hide show

app.py +79 -56

app.py CHANGED Viewed

@@ -1,8 +1,7 @@
 import streamlit as st
 import base64
-from fpdf import FPDF
-import pikepdf
-import fitz  # pymupdf
 # Define the 12-point ML outline with emojis
 ml_outline = [
@@ -20,50 +19,79 @@ ml_outline = [
     "💻 12. ML Code Generation with Streamlit/Gradio/HTML5+JS"
 ]
-def create_pdf_with_library(library_name, outline_items):
-    if library_name == "fpdf":
-        pdf = FPDF(orientation='L', unit='mm', format='A4')
-        pdf.add_page()
-        # Add Unicode font support
-        pdf.add_font('DejaVu', '', 'DejaVuSans.ttf', uni=True)
-        pdf.set_font('DejaVu', '', 10)
-        # Left page (1-6)
-        pdf.set_xy(10, 10)
-        pdf.multi_cell(140, 5, "Cutting-Edge ML Areas (1-6)\n\n" + "\n".join(outline_items[:6]))
-        # Right page (7-12)
-        pdf.set_xy(150, 10)
-        pdf.multi_cell(140, 5, "Cutting-Edge ML Areas (7-12)\n\n" + "\n".join(outline_items[6:]))
-        filename = "ml_outline_fpdf.pdf"
-        pdf.output(filename)
-        return filename
-    elif library_name == "pikepdf":
-        pdf = pikepdf.Pdf.new()
-        page = pikepdf.Page(pdf.make_stream(b""))
-        pdf.pages.append(page)
-        filename = "ml_outline_pikepdf.pdf"
-        pdf.save(filename)
-        return filename
-    elif library_name == "pymupdf":
-        doc = fitz.open()
-        page = doc.new_page(width=842, height=595)  # A4 landscape
-        page.insert_text((20, 20), "Cutting-Edge ML Areas (1-6)\n\n" + "\n".join(outline_items[:6]),
-                        fontsize=10)
-        page.insert_text((422, 20), "Cutting-Edge ML Areas (7-12)\n\n" + "\n".join(outline_items[6:]),
-                        fontsize=10)
-        filename = "ml_outline_pymupdf.pdf"
-        doc.save(filename)
-        return filename
 def get_binary_file_downloader_html(bin_file, file_label='File'):
-    with open(bin_file, 'rb') as f:
-        data = f.read()
-    bin_str = base64.b64encode(data).decode()
-    href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{file_label}">Download {file_label aprovada}</a>'
     return href
 # Streamlit UI
@@ -79,32 +107,30 @@ with col1:
     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, "ml_outline.md"), unsafe_allow_html=True)
 with col2:
     st.header("📑 PDF Preview")
-    pdf_library = st.selectbox("Choose PDF Library", ["fpdf", "pikepdf", "pymupdf"])
     if st.button("Generate PDF"):
         with st.spinner("Generating PDF..."):
             try:
-                pdf_file = create_pdf_with_library(pdf_library, ml_outline)
-                with open(pdf_file, "rb") as f:
-                    pdf_bytes = f.read()
                 st.download_button(
                     label="Download PDF",
                     data=pdf_bytes,
-                    file_name=pdf_file,
                     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 FileNotFoundError:
-                st.error("Please download the DejaVuSans.ttf font and place it in the working directory for FPDF Unicode support")
             except Exception as e:
                 st.error(f"Error generating PDF: {str(e)}")
@@ -114,8 +140,5 @@ st.markdown("""
         background-color: #4CAF50;
         color: white;
     }
-    .stSelectbox {
-        max-width: 300px;
-    }
 </style>
 """, unsafe_allow_html=True)

 import streamlit as st
 import base64
+from weasyprint import HTML
+import io
 # Define the 12-point ML outline with emojis
 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>
+    <body>
+        <div class="container">
+            <div class="column">
+                <h2>Cutting-Edge ML Areas (1-6)</h2>
+                <ul>
+    """
+    # Add items 1-6
+    for item in outline_items[:6]:
+        html_content += f"<li>{item}</li>"
+    html_content += """
+                </ul>
+            </div>
+            <div class="column">
+                <h2>Cutting-Edge ML Areas (7-12)</h2>
+                <ul>
+    """
+    # Add items 7-12
+    for item in outline_items[6:]:
+        html_content += f"<li>{item}</li>"
+    html_content += """
+                </ul>
+            </div>
+        </div>
+    </body>
+    </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
 # Streamlit UI
     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")
     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,
+                    file_name="ml_outline.pdf",
                     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)}")
         background-color: #4CAF50;
         color: white;
     }
 </style>
 """, unsafe_allow_html=True)