Spaces:

codelion
/

markdownPDF

Running

App Files Files Community

codelion commited on Feb 20

Commit

81b49c9

verified ·

1 Parent(s): 3463898

Update app.py

Browse files

Files changed (1) hide show

app.py +34 -18

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 # app.py
 import gradio as gr
 import markdown
-import pdfkit
 import tempfile
 import os
 from pathlib import Path
 def convert_markdown_to_pdf(markdown_file):
@@ -19,18 +19,38 @@ def convert_markdown_to_pdf(markdown_file):
     # Convert markdown to HTML
     html_content = markdown.markdown(markdown_content)
-    # Wrap HTML content with proper HTML structure
     full_html = f"""
     <!DOCTYPE html>
     <html>
     <head>
         <meta charset="UTF-8">
         <style>
-            body {{ font-family: Arial, sans-serif; line-height: 1.6; max-width: 800px; margin: 0 auto; padding: 20px; }}
-            h1 {{ color: #2c3e50; }}
-            h2 {{ color: #34495e; }}
-            code {{ background-color: #f7f7f7; padding: 2px 5px; border-radius: 3px; }}
-            pre {{ background-color: #f7f7f7; padding: 15px; border-radius: 5px; overflow-x: auto; }}
         </style>
     </head>
     <body>
@@ -39,22 +59,18 @@ def convert_markdown_to_pdf(markdown_file):
     </html>
     """
-    # Create temporary files for HTML and PDF
-    with tempfile.NamedTemporaryFile(delete=False, suffix='.html') as html_file:
-        html_file.write(full_html.encode('utf-8'))
-        html_path = html_file.name
-    pdf_path = html_path.replace('.html', '.pdf')
-    # Convert HTML to PDF using pdfkit
     try:
-        pdfkit.from_file(html_path, pdf_path)
-        os.unlink(html_path)  # Clean up temporary HTML file
         return full_html, pdf_path
     except Exception as e:
         print(f"Error converting to PDF: {e}")
-        if os.path.exists(html_path):
-            os.unlink(html_path)
         return None, None
 def process_file(file):

 # app.py
 import gradio as gr
 import markdown
 import tempfile
 import os
+from weasyprint import HTML
 from pathlib import Path
 def convert_markdown_to_pdf(markdown_file):
     # Convert markdown to HTML
     html_content = markdown.markdown(markdown_content)
+    # Wrap HTML content with proper HTML structure and CSS
     full_html = f"""
     <!DOCTYPE html>
     <html>
     <head>
         <meta charset="UTF-8">
         <style>
+            @page {{
+                margin: 2.5cm;
+            }}
+            body {{
+                font-family: Arial, sans-serif;
+                line-height: 1.6;
+                max-width: 800px;
+                margin: 0 auto;
+                padding: 20px;
+            }}
+            h1 {{ color: #2c3e50; margin-top: 1em; }}
+            h2 {{ color: #34495e; margin-top: 0.8em; }}
+            code {{
+                background-color: #f7f7f7;
+                padding: 2px 5px;
+                border-radius: 3px;
+                font-family: monospace;
+            }}
+            pre {{
+                background-color: #f7f7f7;
+                padding: 15px;
+                border-radius: 5px;
+                white-space: pre-wrap;
+                font-family: monospace;
+            }}
         </style>
     </head>
     <body>
     </html>
     """
+    # Create temporary file for PDF
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as pdf_file:
+        pdf_path = pdf_file.name
+    # Convert HTML to PDF using WeasyPrint
     try:
+        HTML(string=full_html).write_pdf(pdf_path)
         return full_html, pdf_path
     except Exception as e:
         print(f"Error converting to PDF: {e}")
+        if os.path.exists(pdf_path):
+            os.unlink(pdf_path)
         return None, None
 def process_file(file):