Markdown 导出器

import osimport tempfileimport uuidfrom docx import Documentfrom docx.shared import Ptfrom docx.enum.text import WD_PARAGRAPH_ALIGNMENTimport markdownfrom bs4 import BeautifulSoupdef convert_markdown_to_docx(markdown_text, document_name=None):    """    Convert markdown text to a Word document and return the file path    """    if not document_name:        document_name = f"exported_document_{uuid.uuid4().hex[:8]}"        # Convert markdown to HTML    html = markdown.markdown(markdown_text)    soup = BeautifulSoup(html, 'html.parser')        # Create a new Word document    doc = Document()        # Process HTML elements and add to document    for element in soup.find_all(['h1', 'h2', 'h3', 'h4', 'p', 'ul', 'ol']):        if element.name == 'h1':            heading = doc.add_heading(element.text.strip(), level=1)            heading.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER        elif element.name == 'h2':            doc.add_heading(element.text.strip(), level=2)        elif element.name == 'h3':            doc.add_heading(element.text.strip(), level=3)        elif element.name == 'h4':            doc.add_heading(element.text.strip(), level=4)        elif element.name == 'p':            paragraph = doc.add_paragraph(element.text.strip())        elif element.name in ('ul', 'ol'):            for li in element.find_all('li'):                doc.add_paragraph(li.text.strip(), style='ListBullet')        # Create temp directory if it doesn't exist    temp_dir = tempfile.gettempdir()    if not os.path.exists(temp_dir):        os.makedirs(temp_dir)        # Save the document    file_path = os.path.join(temp_dir, f"{document_name}.docx")    doc.save(file_path)        return file_path

import osimport tempfileimport uuidimport markdownfrom weasyprint import HTML, CSSfrom weasyprint.text.fonts import FontConfigurationdef convert_markdown_to_pdf(markdown_text, document_name=None):    """    Convert markdown text to a PDF document and return the file path    """    if not document_name:        document_name = f"exported_document_{uuid.uuid4().hex[:8]}"        # Convert markdown to HTML    html_content = markdown.markdown(markdown_text)        # Add basic styling    styled_html = f"""    <!DOCTYPE html>    <html>    <head>        <meta charset="UTF-8">        <title>{document_name}</title>        <style>            body {{ font-family: Arial, sans-serif; margin: 40px; line-height: 1.6; }}            h1 {{ text-align: center; color: #333; }}            h2, h3, h4 {{ color: #444; margin-top: 20px; }}            p {{ margin-bottom: 15px; }}            ul, ol {{ margin-left: 20px; }}        </style>    </head>    <body>        {html_content}    </body>    </html>    """        # Create temp directory if it doesn't exist    temp_dir = tempfile.gettempdir()    if not os.path.exists(temp_dir):        os.makedirs(temp_dir)        # Output file path    file_path = os.path.join(temp_dir, f"{document_name}.pdf")        # Configure fonts    font_config = FontConfiguration()        # Render PDF    HTML(string=styled_html).write_pdf(        file_path,        stylesheets=[],        font_config=font_config    )        return file_path

import osimport base64from collections.abc import Generatorfrom typing import Anyfrom dify_plugin import Toolfrom dify_plugin.entities.tool import ToolInvokeMessagefrom utils.docx_utils import convert_markdown_to_docxclass WordExportTool(Tool):    def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:        # Extract parameters        markdown_content = tool_parameters.get("markdown_content", "")        document_name = tool_parameters.get("document_name", "exported_document")                if not markdown_content:            yield self.create_text_message("Error: No content provided for export.")            return                try:            # Convert markdown to Word            file_path = convert_markdown_to_docx(markdown_content, document_name)                        # Read the file as binary            with open(file_path, 'rb') as file:                file_content = file.read()                        # Encode as base64            file_base64 = base64.b64encode(file_content).decode('utf-8')                        # Return success message and file            yield self.create_text_message(                f"Document exported successfully as Word (.docx) format."            )                        yield self.create_file_message(                file_name=f"{document_name}.docx",                 file_content=file_base64,                mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document"            )                    except Exception as e:            yield self.create_text_message(f"Error exporting to Word: {str(e)}")            return

import osimport base64from collections.abc import Generatorfrom typing import Anyfrom dify_plugin import Toolfrom dify_plugin.entities.tool import ToolInvokeMessagefrom utils.pdf_utils import convert_markdown_to_pdfclass PDFExportTool(Tool):    def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:        # Extract parameters        markdown_content = tool_parameters.get("markdown_content", "")        document_name = tool_parameters.get("document_name", "exported_document")                if not markdown_content:            yield self.create_text_message("Error: No content provided for export.")            return                try:            # Convert markdown to PDF            file_path = convert_markdown_to_pdf(markdown_content, document_name)                        # Read the file as binary            with open(file_path, 'rb') as file:                file_content = file.read()                        # Encode as base64            file_base64 = base64.b64encode(file_content).decode('utf-8')                        # Return success message and file            yield self.create_text_message(                f"Document exported successfully as PDF format."            )                        yield self.create_file_message(                file_name=f"{document_name}.pdf",                 file_content=file_base64,                mime_type="application/pdf"            )                    except Exception as e:            yield self.create_text_message(f"Error exporting to PDF: {str(e)}")            return

from dify_plugin import PluginRunnerfrom tools.word_export import WordExportToolfrom tools.pdf_export import PDFExportToolplugin = PluginRunner(    tools=[        WordExportTool(),        PDFExportTool(),    ],    providers=[]  # No credential providers needed)