Spaces:

awacke1
/

Pillow-PyMuPDF-ReportLab

Sleeping

App Files Files Community

awacke1 commited on Mar 31

Commit

2a632c8

verified ·

1 Parent(s): 6d4ddf4

Update app.py

Browse files

Files changed (1) hide show

app.py +114 -81

app.py CHANGED Viewed

@@ -1,6 +1,15 @@
 import streamlit as st
 import base64
-import pdfkit
 import io
 import os
@@ -20,84 +29,88 @@ 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>
-    <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 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()
@@ -120,26 +133,47 @@ with col1:
     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,
                     file_name="ml_outline.pdf",
                     mime="application/pdf"
                 )
-                # Fix PDF viewer
                 base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
                 pdf_display = f'''
                     <embed
@@ -149,7 +183,6 @@ with col2:
                         type="application/pdf">
                 '''
                 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
+from reportlab.lib.pagesizes import A4
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.lib import colors
+import pikepdf
+import fpdf
+import fitz  # pymupdf
+import cv2
+from PIL import Image
+import imutils.video
 import io
 import os
     "💻 12. ML Code Generation with Streamlit/Gradio/HTML5+JS"
 ]
+# Demo functions for PDF libraries
+def demo_pikepdf():
+    pdf = pikepdf.Pdf.new()
+    pdf.pages.append(pikepdf.Page(pikepdf.Dictionary()))
+    buffer = io.BytesIO()
+    pdf.save(buffer)
+    buffer.seek(0)
+    return buffer.getvalue()
+def demo_fpdf():
+    pdf = fpdf.FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.cell(200, 10, txt="FPDF Demo", ln=True)
+    buffer = io.BytesIO()
+    pdf.output(buffer)
+    buffer.seek(0)
+    return buffer.getvalue()
+def demo_pymupdf():
+    doc = fitz.open()
+    page = doc.new_page()
+    page.insert_text((100, 100), "PyMuPDF Demo")
+    buffer = io.BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return buffer.getvalue()
+# Demo function for image capture (using OpenCV as representative)
+def demo_image_capture():
+    try:
+        cap = cv2.VideoCapture(0)
+        ret, frame = cap.read()
+        if ret:
+            rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            img = Image.fromarray(rgb_frame)
+            buffer = io.BytesIO()
+            img.save(buffer, format="JPEG")
+            buffer.seek(0)
+            cap.release()
+            return buffer.getvalue()
+        cap.release()
+    except:
+        return None
+    return None
+# Main PDF creation using ReportLab
+def create_main_pdf(outline_items):
+    buffer = io.BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=(A4[1], A4[0]))  # Landscape
+    styles = getSampleStyleSheet()
+    story = []
+    # Title style
+    title_style = styles['Heading1']
+    title_style.textColor = colors.darkblue
+    # Normal style
+    normal_style = styles['Normal']
+    normal_style.fontSize = 10
+    normal_style.leading = 14
+    # Page 1: Items 1-6
+    story.append(Paragraph("Cutting-Edge ML Areas (1-6)", title_style))
+    story.append(Spacer(1, 12))
+    for item in outline_items[:6]:
+        story.append(Paragraph(item, normal_style))
+        story.append(Spacer(1, 6))
+    # Page break
+    story.append(Spacer(1, 500))  # Force new page
+    # Page 2: Items 7-12
+    story.append(Paragraph("Cutting-Edge ML Areas (7-12)", title_style))
+    story.append(Spacer(1, 12))
+    for item in outline_items[6:]:
+        story.append(Paragraph(item, normal_style))
+        story.append(Spacer(1, 6))
+    doc.build(story)
+    buffer.seek(0)
+    return buffer.getvalue()
 def get_binary_file_downloader_html(bin_data, file_label='File'):
     bin_str = base64.b64encode(bin_data).decode()
     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 & Demos")
+    # Library Demos
+    st.subheader("Library Demos")
+    if st.button("Run PDF Demos"):
+        with st.spinner("Running demos..."):
+            # pikepdf demo
+            pike_pdf = demo_pikepdf()
+            st.download_button("Download pikepdf Demo", pike_pdf, "pikepdf_demo.pdf")
+            # fpdf demo
+            fpdf_pdf = demo_fpdf()
+            st.download_button("Download fpdf Demo", fpdf_pdf, "fpdf_demo.pdf")
+            # pymupdf demo
+            pymupdf_pdf = demo_pymupdf()
+            st.download_button("Download pymupdf Demo", pymupdf_pdf, "pymupdf_demo.pdf")
+            # Image capture demo
+            img_data = demo_image_capture()
+            if img_data:
+                st.image(img_data, caption="OpenCV Image Capture Demo")
+            else:
+                st.warning("Image capture demo failed - camera not detected")
+    # Main PDF Generation
+    if st.button("Generate Main PDF"):
         with st.spinner("Generating PDF..."):
             try:
+                pdf_bytes = create_main_pdf(ml_outline)
                 with open("ml_outline.pdf", "wb") as f:
                     f.write(pdf_bytes)
                 st.download_button(
+                    label="Download Main PDF",
                     data=pdf_bytes,
                     file_name="ml_outline.pdf",
                     mime="application/pdf"
                 )
                 base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
                 pdf_display = f'''
                     <embed
                         type="application/pdf">
                 '''
                 st.markdown(pdf_display, unsafe_allow_html=True)
             except Exception as e:
                 st.error(f"Error generating PDF: {str(e)}")