Update app.py

app.py

@@ -1,9 +1,17 @@
 import gradio as gr
 import base64
-from weasyprint import HTML, CSS
 import json
-from datetime import datetime
 import tempfile
+import os
+import sys
+
+# Try to import WeasyPrint with error handling
+try:
+    from weasyprint import HTML, CSS
+    WEASYPRINT_AVAILABLE = True
+except ImportError as e:
+    print(f"WeasyPrint import error: {e}")
+    WEASYPRINT_AVAILABLE = False
 
 # Default CSS for professional PDFs
 DEFAULT_CSS = """
@@ -21,26 +29,35 @@ th { background-color: #f4f4f4; }
 
 def html_to_pdf_ui(html_content, custom_css, use_default_css):
     """Convert HTML to PDF for UI"""
+    if not WEASYPRINT_AVAILABLE:
+        return None, "❌ WeasyPrint is not available. Please check installation."
+    
     try:
         if not html_content:
             return None, "Please provide HTML content"
         
         # Prepare CSS
-        css_content = ""
+        stylesheets = []
         if use_default_css:
-            css_content = DEFAULT_CSS
-        if custom_css:
-            css_content += "\n" + custom_css
+            stylesheets.append(CSS(string=DEFAULT_CSS))
+        if custom_css and custom_css.strip():
+            stylesheets.append(CSS(string=custom_css))
         
-        # Create HTML object
+        # Create HTML object and generate PDF
         html_obj = HTML(string=html_content)
         
-        # Generate PDF with or without CSS
-        if css_content:
-            css_obj = CSS(string=css_content)
-            pdf_bytes = html_obj.write_pdf(stylesheets=[css_obj])
-        else:
-            pdf_bytes = html_obj.write_pdf()
+        # Generate PDF - handle different WeasyPrint versions
+        try:
+            if stylesheets:
+                pdf_bytes = html_obj.write_pdf(stylesheets=stylesheets)
+            else:
+                pdf_bytes = html_obj.write_pdf()
+        except TypeError:
+            # Fallback for older WeasyPrint versions
+            if stylesheets:
+                pdf_bytes = html_obj.write_pdf(stylesheets=stylesheets)
+            else:
+                pdf_bytes = html_obj.write_pdf()
         
         # Save to temp file
         with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as f:
@@ -52,6 +69,9 @@ def html_to_pdf_ui(html_content, custom_css, use_default_css):
 
 def html_to_pdf_api(json_input):
     """Convert HTML to PDF for API - returns base64"""
+    if not WEASYPRINT_AVAILABLE:
+        return {"error": "WeasyPrint is not available", "status": "error"}
+    
     try:
         data = json.loads(json_input) if isinstance(json_input, str) else json_input
         
@@ -62,16 +82,14 @@ def html_to_pdf_api(json_input):
         if not html:
             return {"error": "No HTML content", "status": "error"}
         
+        # Prepare stylesheets
+        stylesheets = [CSS(string=DEFAULT_CSS)]
+        if css and css.strip():
+            stylesheets.append(CSS(string=css))
+        
         # Generate PDF
         html_obj = HTML(string=html)
-        
-        # Prepare CSS
-        full_css = DEFAULT_CSS
-        if css:
-            full_css += "\n" + css
-        
-        css_obj = CSS(string=full_css)
-        pdf_bytes = html_obj.write_pdf(stylesheets=[css_obj])
+        pdf_bytes = html_obj.write_pdf(stylesheets=stylesheets)
         
         return {
             "status": "success",
@@ -87,6 +105,10 @@ def html_to_pdf_api(json_input):
 with gr.Blocks(title="HTML to PDF Converter") as app:
     gr.Markdown("# 📄 HTML to PDF Converter\nConvert HTML to PDF with custom CSS. Perfect for n8n workflows!")
     
+    # Show WeasyPrint status
+    if not WEASYPRINT_AVAILABLE:
+        gr.Markdown("⚠️ **WeasyPrint is not available. Please check the installation.**")
+    
     with gr.Tab("✏️ Convert"):
         with gr.Row():
             with gr.Column():