From fd430f281b04ef8ba960131e0c84022d4ad9e645 Mon Sep 17 00:00:00 2001
From: ytqh <yutianqiuhao@gmail.com>
Date: Sat, 26 Apr 2025 17:35:41 +0800
Subject: [PATCH] add html to pdf endpoints

---
 api/controllers/inner_tools/__init__.py    |  2 +-
 api/controllers/inner_tools/html_to_pdf.py | 50 ++++++++++++++++++++++
 2 files changed, 51 insertions(+), 1 deletion(-)
 create mode 100644 api/controllers/inner_tools/html_to_pdf.py

diff --git a/api/controllers/inner_tools/__init__.py b/api/controllers/inner_tools/__init__.py
index 5ecf763ea6..b9d1b8b07b 100644
--- a/api/controllers/inner_tools/__init__.py
+++ b/api/controllers/inner_tools/__init__.py
@@ -4,4 +4,4 @@ from libs.external_api import ExternalApi
 bp = Blueprint("inner_tools", __name__, url_prefix="/inner_tools")
 api = ExternalApi(bp)
 
-from . import answers_summary_analysis, markdown_to_pdf
+from . import answers_summary_analysis, html_to_pdf, markdown_to_pdf
diff --git a/api/controllers/inner_tools/html_to_pdf.py b/api/controllers/inner_tools/html_to_pdf.py
new file mode 100644
index 0000000000..796ac188f2
--- /dev/null
+++ b/api/controllers/inner_tools/html_to_pdf.py
@@ -0,0 +1,50 @@
+import uuid
+
+from controllers.inner_tools import api
+from core.tools.tool_file_manager import ToolFileManager
+from flask import jsonify, request
+from flask_restful import Resource  # type: ignore
+from models.account import Tenant
+from weasyprint import HTML
+
+
+class HtmlToPdfApi(Resource):
+    def post(self):
+        """Generate a PDF from the provided HTML content."""
+        if not request.data:
+            return {"error": "No HTML content provided"}, 400
+
+        html_content = request.data
+
+        # Generate PDF
+        html = HTML(string=html_content)
+        pdf_file = html.write_pdf()
+
+        if pdf_file is None:
+            return {"error": "Failed to generate PDF"}, 500
+
+        # Get the first tenant from database (similar to markdown_to_pdf.py)
+        tenant = Tenant.query.first()
+        if not tenant:
+            return {"error": "No tenant found"}, 400
+
+        tenant_id = tenant.id
+
+        # Generate filename
+        filename = f"html_to_pdf_{uuid.uuid4().hex[:8]}.pdf"
+
+        # Save the file using ToolFileManager
+        tool_file = ToolFileManager.create_file_by_raw(
+            user_id=None,
+            tenant_id=tenant_id,
+            conversation_id=None,
+            file_binary=pdf_file,
+            mimetype='application/pdf',
+        )
+
+        # Return the file info with URL
+        file_url = ToolFileManager.sign_file(tool_file.id, '.pdf')
+        return jsonify({'url': file_url, 'file_id': tool_file.id, 'file_name': filename, 'file_size': tool_file.size})
+
+
+api.add_resource(HtmlToPdfApi, '/html-to-pdf')