import wx
import os
from pptx import Presentation
from PIL import Image
import io
from pptx.enum.shapes import MSO_SHAPE_TYPE
import comtypes.client
import tempfile

class PPTMergerFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='PPT Merger', size=(400, 300))
        self.init_ui()
        self.Centre()

    def init_ui(self):
        panel = wx.Panel(self)
        vbox = wx.BoxSizer(wx.VERTICAL)

        # 添加選擇文件按鈕
        self.select_btn = wx.Button(panel, label='選擇PPT文件')
        self.select_btn.Bind(wx.EVT_BUTTON, self.on_select)
        vbox.Add(self.select_btn, 0, wx.ALL | wx.CENTER, 20)

        # 添加文件路徑顯示
        self.path_text = wx.TextCtrl(panel, style=wx.TE_READONLY)
        vbox.Add(self.path_text, 0, wx.ALL | wx.EXPAND, 20)

        # 添加開始處理按鈕
        self.process_btn = wx.Button(panel, label='開始處理')
        self.process_btn.Bind(wx.EVT_BUTTON, self.on_process)
        self.process_btn.Disable()
        vbox.Add(self.process_btn, 0, wx.ALL | wx.CENTER, 20)

        # 添加進(jìn)度條
        self.gauge = wx.Gauge(panel, range=100)
        vbox.Add(self.gauge, 0, wx.ALL | wx.EXPAND, 20)

        panel.SetSizer(vbox)

    def on_select(self, event):
        wildcard = "PowerPoint files (*.pptx)|*.pptx"
        with wx.FileDialog(self, "選擇PPT文件", wildcard=wildcard,
                         style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST) as fileDialog:
            
            if fileDialog.ShowModal() == wx.ID_CANCEL:
                return

            self.ppt_path = fileDialog.GetPath()
            self.path_text.SetValue(self.ppt_path)
            self.process_btn.Enable()

    def convert_pptx_to_pdf(self, input_path, output_path):
        """將PPTX轉(zhuǎn)換為PDF"""
        powerpoint = comtypes.client.CreateObject("Powerpoint.Application")
        powerpoint.Visible = True
        slides = powerpoint.Presentations.Open(input_path)
        slides.SaveAs(output_path, 32)  # 32 代表 PDF 格式
        slides.Close()
        powerpoint.Quit()

    def convert_pdf_to_images(self, pdf_path):
        """將PDF轉(zhuǎn)換為圖片列表"""
        try:
            import fitz  # PyMuPDF
        except ImportError:
            wx.MessageBox('請先安裝PyMuPDF庫：pip install PyMuPDF', '錯誤')
            return []

        images = []
        pdf_document = fitz.open(pdf_path)
        
        for page_num in range(pdf_document.page_count):
            page = pdf_document[page_num]
            pix = page.get_pixmap(matrix=fitz.Matrix(300/72, 300/72))  # 300 DPI
            img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
            images.append(img)
            
        pdf_document.close()
        return images

    def on_process(self, event):
        try:
            # 禁用按鈕防止重復(fù)處理
            self.process_btn.Disable()
            self.select_btn.Disable()
            
            # 創(chuàng)建臨時PDF文件
            temp_pdf = tempfile.mktemp(suffix='.pdf')
            
            # 轉(zhuǎn)換PPTX為PDF
            self.gauge.SetValue(25)
            self.convert_pptx_to_pdf(self.ppt_path, temp_pdf)
            
            # 轉(zhuǎn)換PDF為圖片
            self.gauge.SetValue(50)
            images = self.convert_pdf_to_images(temp_pdf)
            
            if not images:
                raise Exception("未能成功轉(zhuǎn)換幻燈片為圖片")
            
            # 計算合并后圖片的尺寸
            self.gauge.SetValue(75)
            total_height = sum(img.size[1] for img in images)
            max_width = max(img.size[0] for img in images)
            
            # 創(chuàng)建新圖片
            merged_image = Image.new('RGB', (max_width, total_height), 'white')
            
            # 垂直拼接圖片
            y_offset = 0
            for img in images:
                merged_image.paste(img, (0, y_offset))
                y_offset += img.size[1]
            
            # 保存合并后的圖片
            save_path = os.path.splitext(self.ppt_path)[0] + '_merged.jpg'
            merged_image.save(save_path, 'JPEG', quality=95)
            
            # 刪除臨時PDF文件
            try:
                os.remove(temp_pdf)
            except:
                pass
            
            self.gauge.SetValue(100)
            
            # 完成處理
            wx.MessageBox(f'處理完成！\n保存路徑：{save_path}', '成功')
            
        except Exception as e:
            wx.MessageBox(f'處理過程中出現(xiàn)錯誤：{str(e)}', '錯誤')
        
        finally:
            # 重置UI狀態(tài)
            self.gauge.SetValue(0)
            self.process_btn.Enable()
            self.select_btn.Enable()

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

def convert_pptx_to_pdf(self, input_path, output_path):
    powerpoint = comtypes.client.CreateObject("Powerpoint.Application")
    powerpoint.Visible = True
    slides = powerpoint.Presentations.Open(input_path)
    slides.SaveAs(output_path, 32)  # 32 代表 PDF 格式
    slides.Close()
    powerpoint.Quit()

def convert_pdf_to_images(self, pdf_path):
    images = []
    pdf_document = fitz.open(pdf_path)
    
    for page_num in range(pdf_document.page_count):
        page = pdf_document[page_num]
        pix = page.get_pixmap(matrix=fitz.Matrix(300/72, 300/72))
        img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
        images.append(img)
        
    pdf_document.close()
    return images

# 計算合并后圖片的尺寸
total_height = sum(img.size[1] for img in images)
max_width = max(img.size[0] for img in images)

# 創(chuàng)建新圖片
merged_image = Image.new('RGB', (max_width, total_height), 'white')

# 垂直拼接圖片
y_offset = 0
for img in images:
    merged_image.paste(img, (0, y_offset))
    y_offset += img.size[1]

pip install wxPython python-pptx Pillow PyMuPDF comtypes

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