import wx
import os
from PIL import Image, ImageDraw, ImageFont, ImageOps

class PokerCardFrame(wx.Frame):
    def __init__(self):
        super().__init__(None, title="撲克牌生成器", size=(800, 1100))  # 調(diào)整窗口大小以更好地預(yù)覽
        panel = wx.Panel(self)
        
        suits = ['?', '?', '?', '?']
        self.suit_choice = wx.Choice(panel, choices=suits)
        self.suit_choice.SetSelection(0)
        
        numbers = ['A', '2', '3', '4', '5', '6', '7', '8', '9', '10', 'J', 'Q', 'K']
        self.number_choice = wx.Choice(panel, choices=numbers)
        self.number_choice.SetSelection(0)
        
        colors = ['黑色', '紅色', '藍(lán)色', '綠色']
        self.color_choice = wx.Choice(panel, choices=colors)
        self.color_choice.SetSelection(0)
        
        bg_btn = wx.Button(panel, label="選擇人物照片")
        generate_btn = wx.Button(panel, label="生成撲克牌")
        save_btn = wx.Button(panel, label="保存圖片")
        
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(wx.StaticText(panel, label="選擇花色:"), 0, wx.ALL, 5)
        sizer.Add(self.suit_choice, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(wx.StaticText(panel, label="選擇數(shù)字:"), 0, wx.ALL, 5)
        sizer.Add(self.number_choice, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(wx.StaticText(panel, label="選擇顏色:"), 0, wx.ALL, 5)
        sizer.Add(self.color_choice, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(bg_btn, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(generate_btn, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(save_btn, 0, wx.ALL|wx.EXPAND, 5)
        
        panel.SetSizer(sizer)
        
        bg_btn.Bind(wx.EVT_BUTTON, self.on_select_background)
        generate_btn.Bind(wx.EVT_BUTTON, self.on_generate)
        save_btn.Bind(wx.EVT_BUTTON, self.on_save)
        
        self.image = None
        self.bg_image = None
        
    def create_card(self, suit, number, color):
        # 真實(shí)撲克牌大?。?50×1050 像素 (300 DPI)
        width, height = 750, 1050
        img = Image.new('RGB', (width, height), 'white')
        draw = ImageDraw.Draw(img)
        
        # 圓角矩形邊框（radius=50 為圓角半徑）
        border_margin = 20
        draw.rounded_rectangle([(border_margin, border_margin), (width - border_margin, height - border_margin)], 
                              radius=50, outline='black', width=5)
        
        if self.bg_image:
            # 裁切為正方形（可選，如果你仍想要正方形背景）
            bg_width, bg_height = self.bg_image.size
            min_size = min(bg_width, bg_height)
            left = (bg_width - min_size) // 2
            top = (bg_height - min_size) // 2
            right = left + min_size
            bottom = top + min_size
            person_img = self.bg_image.crop((left, top, right, bottom)).convert('RGBA')
            
            # 直接調(diào)整為固定大小 400×600
            new_size = (500, 750)
            person_img = person_img.resize(new_size, Image.Resampling.LANCZOS)
            
            # 居中放置
            x = (width - new_size[0]) // 2
            y = (height - new_size[1]) // 2
            img.paste(person_img, (x, y), person_img)
        
        try:
            font = ImageFont.truetype("arial.ttf", 80)
        except:
            font = ImageFont.load_default()
        
        color_map = {'黑色': 'black', '紅色': 'red', '藍(lán)色': 'blue', '綠色': 'green'}
        text_color = color_map[color]
        
        # 上部：數(shù)字在花色上方（左上角）
        draw.text((40, 20), number, fill=text_color, font=font)  # 數(shù)字在上
        draw.text((40, 100), suit, fill=text_color, font=font)   # 花色在下
        
        # 下部：創(chuàng)建臨時(shí)圖片用于旋轉(zhuǎn)，定位到右下角
        temp_img = Image.new('RGBA', (width, height), (255, 255, 255, 0))
        temp_draw = ImageDraw.Draw(temp_img)
        
        # 在臨時(shí)圖片上繪制下部文字（正常順序，靠右）
        temp_draw.text((width- 700, 10), number, fill=text_color, font=font)    # 花色在上
        temp_draw.text((width- 700 , 90), suit, fill=text_color, font=font)  # 數(shù)字在下   
        
        # 旋轉(zhuǎn)180度
        rotated_img = temp_img.rotate(180)
        
        # 將旋轉(zhuǎn)后的文字粘貼到主圖片
        img.paste(rotated_img, (0, 0), rotated_img)
        
        return img
    
    def on_select_background(self, event):
        with wx.FileDialog(self, "選擇人物照片", wildcard="Image files (*.png;*.jpg)|*.png;*.jpg",
                          style=wx.FD_OPEN) as fileDialog:
            if fileDialog.ShowModal() == wx.ID_OK:
                path = fileDialog.GetPath()
                self.bg_image = Image.open(path)
                wx.MessageBox(f"已選擇人物照片: {path}", "成功")
    
    def on_generate(self, event):
        suit = self.suit_choice.GetStringSelection()
        number = self.number_choice.GetStringSelection()
        color = self.color_choice.GetStringSelection()
        
        self.image = self.create_card(suit, number, color)
        
        img = self.image.convert('RGB')
        wx_img = wx.Image(img.size[0], img.size[1])
        wx_img.SetData(img.tobytes())
        
        if hasattr(self, 'bitmap'):
            self.bitmap.Destroy()
        self.bitmap = wx.StaticBitmap(self.GetChildren()[0], -1, wx.Bitmap(wx_img))
        self.Layout()
        
    def on_save(self, event):
        if self.image is None:
            wx.MessageBox("請(qǐng)先生成撲克牌！", "錯(cuò)誤")
            return
            
        with wx.FileDialog(self, "保存撲克牌圖片", wildcard="PNG files (*.png)|*.png",
                          style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT) as fileDialog:
            if fileDialog.ShowModal() == wx.ID_OK:
                path = fileDialog.GetPath()
                self.image.save(path, 'PNG')
                wx.MessageBox(f"圖片已保存到: {path}", "成功")

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