import wx
import markdown2
from docx import Document
from docx.shared import Pt, RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH
import re

class MainFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='Markdown to DOCX Converter', size=(800, 600))
        self.init_ui()
        
    def init_ui(self):
        # 創(chuàng)建主面板
        panel = wx.Panel(self)
        
        # 創(chuàng)建垂直布局
        vbox = wx.BoxSizer(wx.VERTICAL)
        
        # 添加說明文字
        help_text = wx.StaticText(panel, label="請輸入或粘貼Markdown格式的文本:")
        vbox.Add(help_text, 0, wx.ALL, 5)
        
        # 創(chuàng)建多行文本輸入框
        self.text_ctrl = wx.TextCtrl(panel, style=wx.TE_MULTILINE)
        vbox.Add(self.text_ctrl, 1, wx.EXPAND | wx.ALL, 5)
        
        # 創(chuàng)建水平布局用于按鈕
        hbox = wx.BoxSizer(wx.HORIZONTAL)
        
        # 添加轉(zhuǎn)換按鈕
        convert_btn = wx.Button(panel, label='轉(zhuǎn)換為DOCX')
        convert_btn.Bind(wx.EVT_BUTTON, self.on_convert)
        hbox.Add(convert_btn, 0, wx.ALL, 5)
        
        # 添加清除按鈕
        clear_btn = wx.Button(panel, label='清除內(nèi)容')
        clear_btn.Bind(wx.EVT_BUTTON, self.on_clear)
        hbox.Add(clear_btn, 0, wx.ALL, 5)
        
        vbox.Add(hbox, 0, wx.ALIGN_RIGHT)
        
        panel.SetSizer(vbox)
        
    def on_convert(self, event):
        # 獲取輸入的Markdown文本
        markdown_text = self.text_ctrl.GetValue()
        if not markdown_text.strip():
            wx.MessageBox('請輸入Markdown文本內(nèi)容', '提示', wx.OK | wx.ICON_INFORMATION)
            return
            
        # 創(chuàng)建保存文件對話框
        with wx.FileDialog(self, "保存DOCX文件", wildcard="Word files (*.docx)|*.docx",
                         style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT) as fileDialog:
            
            if fileDialog.ShowModal() == wx.ID_CANCEL:
                return
                
            # 保存文件
            pathname = fileDialog.GetPath()
            try:
                self.convert_to_docx(markdown_text, pathname)
                wx.MessageBox('轉(zhuǎn)換成功！', '提示', wx.OK | wx.ICON_INFORMATION)
            except Exception as e:
                wx.MessageBox(f'轉(zhuǎn)換失?。簕str(e)}', '錯誤', wx.OK | wx.ICON_ERROR)
    
    def on_clear(self, event):
        self.text_ctrl.SetValue('')
        
    def convert_to_docx(self, markdown_text, output_path):
        # 將Markdown轉(zhuǎn)換為HTML
        html = markdown2.markdown(markdown_text, extras=['tables', 'fenced-code-blocks'])
        
        # 創(chuàng)建新的Word文檔
        doc = Document()
        
        # 解析HTML并添加到文檔
        # 移除HTML標(biāo)簽后按段落分割
        paragraphs = re.split(r'\n+', re.sub(r'<[^>]+>', '', html))
        
        for para_text in paragraphs:
            if para_text.strip():
                # 檢查是否是標(biāo)題（以#開頭）
                header_match = re.match(r'^(#{1,6})\s+(.+)$', para_text)
                if header_match:
                    level = len(header_match.group(1))
                    text = header_match.group(2)
                    p = doc.add_heading(text, level=level)
                else:
                    p = doc.add_paragraph(para_text)
                    
                # 設(shè)置段落格式
                p.alignment = WD_ALIGN_PARAGRAPH.LEFT
                
                # 設(shè)置字體
                run = p.runs[0] if p.runs else p.add_run()
                font = run.font
                font.name = '微軟雅黑'
                font.size = Pt(11)
                
        # 保存文檔
        doc.save(output_path)

if __name__ == '__main__':
    app = wx.App()
    frame = MainFrame()
    frame.Show()
    app.MainLoop()

def init_ui(self):
    panel = wx.Panel(self)
    vbox = wx.BoxSizer(wx.VERTICAL)
    
    help_text = wx.StaticText(panel, label="請輸入或粘貼Markdown格式的文本:")
    vbox.Add(help_text, 0, wx.ALL, 5)
    
    self.text_ctrl = wx.TextCtrl(panel, style=wx.TE_MULTILINE)
    vbox.Add(self.text_ctrl, 1, wx.EXPAND | wx.ALL, 5)

def convert_to_docx(self, markdown_text, output_path):
    # 轉(zhuǎn)換為HTML
    html = markdown2.markdown(markdown_text, extras=['tables', 'fenced-code-blocks'])
    
    # 創(chuàng)建Word文檔
    doc = Document()
    
    # 解析并添加內(nèi)容
    paragraphs = re.split(r'\n+', re.sub(r'<[^>]+>', '', html))
    
    for para_text in paragraphs:
        if para_text.strip():
            # 處理標(biāo)題和段落
            header_match = re.match(r'^(#{1,6})\s+(.+)$', para_text)
            if header_match:
                level = len(header_match.group(1))
                text = header_match.group(2)
                p = doc.add_heading(text, level=level)
            else:
                p = doc.add_paragraph(para_text)

# 設(shè)置段落格式
p.alignment = WD_ALIGN_PARAGRAPH.LEFT

# 設(shè)置字體
run = p.runs[0] if p.runs else p.add_run()
font = run.font
font.name = '微軟雅黑'
font.size = Pt(11)

pip install wxPython markdown2 python-docx