import wx
import os
import random
import threading
from pptx import Presentation
from pptx.util import Inches
from pptx.enum.action import PP_ACTION
from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, CompositeAudioClip
from PIL import Image
import cv2
import numpy as np
import time
from pathlib import Path

def __init__(self, parent, title):
    super(PhotoManager, self).__init__(parent, title=title, size=(800, 600))
    self.photo_list = []
    self.video_list = []
    self.InitUI()
    self.Centre()
    self.Show()

def InitUI(self):
    panel = wx.Panel(self)
    main_sizer = wx.BoxSizer(wx.HORIZONTAL)
    # 左側(cè)面板：照片管理
    left_panel = wx.Panel(panel)
    left_sizer = wx.BoxSizer(wx.VERTICAL)
    # ... 添加控件，如 ListBox、Button、Gauge 等
    # 右側(cè)面板：視頻合成
    right_panel = wx.Panel(panel)
    right_sizer = wx.BoxSizer(wx.VERTICAL)
    # ... 添加控件
    main_sizer.Add(left_panel, 1, wx.EXPAND | wx.ALL, 10)
    main_sizer.Add(right_panel, 1, wx.EXPAND | wx.ALL, 10)
    panel.SetSizer(main_sizer)
    self.statusbar = self.CreateStatusBar()
    self.statusbar.SetStatusText('準備就緒')
    self.audio_path = None

def load_photos(self, folder_path):
    photo_extensions = ['.jpg', '.jpeg', '.png', '.bmp', '.gif']
    self.photo_list = []
    self.photo_listbox.Clear()
    try:
        files = os.listdir(folder_path)
        for file in files:
            if any(file.lower().endswith(ext) for ext in photo_extensions):
                full_path = self.sanitize_path(os.path.join(folder_path, file))
                self.photo_list.append(full_path)
                self.photo_listbox.Append(file)
        self.statusbar.SetStatusText(f'已加載 {len(self.photo_list)} 張照片')
    except Exception as e:
        wx.MessageBox(f"加載照片時出錯: {str(e)}", "錯誤", wx.OK | wx.ICON_ERROR)

def generate_presentation_and_video(self, save_path):
    try:
        ppt_path = os.path.join(save_path, "照片展示.pptx")
        prs = Presentation()
        video_path = os.path.join(save_path, "照片展示.mp4")
        first_image = Image.open(self.photo_list[0])
        width, height = first_image.size
        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
        video = cv2.VideoWriter(video_path, fourcc, 1, (width, height))
        for i, photo_path in enumerate(self.photo_list):
            slide = prs.slides.add_slide(prs.slide_layouts[5])
            pic = slide.shapes.add_picture(photo_path, Inches(0.5), Inches(0.5), width=Inches(9), height=Inches(5))
            pil_img = Image.open(photo_path)
            img = cv2.cvtColor(np.array(pil_img), cv2.COLOR_RGB2BGR)
            img = cv2.resize(img, (width, height))
            for _ in range(3):  # 每張圖持續(xù)3秒
                video.write(img)
            progress = int((i + 1) / len(self.photo_list) * 90)
            self.update_progress(progress)
        prs.save(ppt_path)
        video.release()
    except Exception as e:
        import traceback
        error_msg = f"生成時出錯: {str(e)}\n{traceback.format_exc()}"
        print(error_msg)
        wx.CallAfter(wx.MessageBox, error_msg, "錯誤", wx.OK | wx.ICON_ERROR)

def merge_videos(self, output_path):
    video_clips = []
    try:
        for i, video_path in enumerate(self.video_list):
            clip = VideoFileClip(video_path)
            video_clips.append(clip)
        final_clip = concatenate_videoclips(video_clips)
        if self.audio_path:
            if not os.path.exists(self.audio_path):
                raise FileNotFoundError(f"音頻文件不存在: {self.audio_path}")
            audio_clip = AudioFileClip(self.audio_path)
            if audio_clip.duration < final_clip.duration:
                audio_clip = audio_clip.loop(duration=final_clip.duration)
            else:
                audio_clip = audio_clip.subclip(0, final_clip.duration)
            audio_clip = audio_clip.volumex(0.5)
            final_clip = final_clip.set_audio(audio_clip)
        final_clip.write_videofile(output_path, codec='libx264')
        final_clip.close()
        for clip in video_clips:
            clip.close()
    except Exception as e:
        import traceback
        error_msg = f"合成視頻時出錯: {str(e)}\n{traceback.format_exc()}"
        print(error_msg)
        wx.CallAfter(wx.MessageBox, error_msg, "錯誤", wx.OK | wx.ICON_ERROR)