# 基礎(chǔ)GUI庫(kù)
pip install wxpython
 
# 圖像處理庫(kù)
pip install opencv-python numpy Pillow
 
# AI摳圖庫(kù)
pip install rembg

def check_dependencies():
    """檢查所有依賴庫(kù)"""
    dependencies = {
        'wxpython': 'wx',
        'opencv-python': 'cv2', 
        'numpy': 'numpy',
        'Pillow': 'PIL',
        'rembg': 'rembg'
    }
    
    missing = []
    for package, module in dependencies.items():
        try:
            __import__(module)
            print(f"? {package} - 已安裝")
        except ImportError:
            missing.append(package)
            print(f"? {package} - 未安裝")
    
    return len(missing) == 0

class PhotoMattingFrame(wx.Frame):
    def __init__(self):
        super().__init__(None, title="證件照摳圖程序", size=(800, 600))
        self.original_image = None      # 原始圖片
        self.processed_image = None     # 處理后圖片
        self.current_bg_color = (255, 255, 255)  # 當(dāng)前背景色
        
        self.create_widgets()   # 創(chuàng)建界面
        self.create_menu()      # 創(chuàng)建菜單

def create_widgets(self):
    panel = wx.Panel(self)
    main_sizer = wx.BoxSizer(wx.VERTICAL)  # 主垂直布局
    
    # 標(biāo)題區(qū)域
    title = wx.StaticText(panel, label="證件照摳圖程序")
    main_sizer.Add(title, 0, wx.ALL | wx.CENTER, 10)
    
    # 圖片顯示區(qū)域 - 水平布局
    image_sizer = wx.BoxSizer(wx.HORIZONTAL)
    # ... 原圖和處理后圖片并排顯示
    
    # 控制按鈕區(qū)域 - 水平布局
    control_sizer = wx.BoxSizer(wx.HORIZONTAL)
    # ... 各種控制按鈕

# 使用StaticBox創(chuàng)建分組
original_box = wx.StaticBox(image_panel, label="原圖")
original_sizer = wx.StaticBoxSizer(original_box, wx.VERTICAL)
 
# StaticBitmap用于顯示圖片
self.original_bitmap = wx.StaticBitmap(image_panel, size=(300, 300))
self.original_bitmap.SetBackgroundColour(wx.Colour(240, 240, 240))

# 顏色選擇器
self.color_picker = wx.ColourPickerCtrl(control_panel, 
                                       colour=wx.Colour(255, 255, 255))
 
# 預(yù)設(shè)顏色按鈕
self.white_btn = wx.Button(control_panel, label="白色", size=(60, 30))
self.blue_btn = wx.Button(control_panel, label="藍(lán)色", size=(60, 30))
self.red_btn = wx.Button(control_panel, label="紅色", size=(60, 30))
 
# 設(shè)置按鈕背景色
self.blue_btn.SetBackgroundColour(wx.Colour(0, 100, 255))
self.blue_btn.SetForegroundColour(wx.Colour(255, 255, 255))

def on_open_image(self, event):
    """打開圖片文件"""
    with wx.FileDialog(self, "選擇證件照",
                      wildcard="圖片文件 (*.jpg;*.jpeg;*.png;*.bmp)|*.jpg;*.jpeg;*.png;*.bmp",
                      style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST) as fileDialog:
        
        if fileDialog.ShowModal() == wx.ID_CANCEL:
            return
            
        pathname = fileDialog.GetPath()
        self.original_image = cv2.imread(pathname)
        self.display_original_image()

def display_original_image(self):
    """顯示原圖"""
    if self.original_image is not None:
        # 調(diào)整大小適應(yīng)顯示區(qū)域
        display_image = self.resize_image_for_display(self.original_image, 300, 300)
        
        # OpenCV(BGR) -> wxPython(RGB)
        height, width = display_image.shape[:2]
        display_image_rgb = cv2.cvtColor(display_image, cv2.COLOR_BGR2RGB)
        wx_image = wx.Image(width, height, display_image_rgb)
        bitmap = wx.Bitmap(wx_image)
        
        self.original_bitmap.SetBitmap(bitmap)

def on_process_image(self, event):
    """處理圖片 - 摳圖并替換背景"""
    try:
        self.status_text.SetLabel("正在處理圖片，請(qǐng)稍候...")
        wx.GetApp().Yield()  # 更新UI
        
        # OpenCV -> PIL格式轉(zhuǎn)換
        image_rgb = cv2.cvtColor(self.original_image, cv2.COLOR_BGR2RGB)
        pil_image = Image.fromarray(image_rgb)
        
        # 使用rembg進(jìn)行背景移除
        output = remove(pil_image)
        
        # 創(chuàng)建新背景
        new_bg = Image.new('RGB', output.size, self.current_bg_color)
        
        # 將摳圖結(jié)果粘貼到新背景上
        new_bg.paste(output, (0, 0), output)
        
        # 轉(zhuǎn)換回OpenCV格式
        result_array = np.array(new_bg)
        self.processed_image = cv2.cvtColor(result_array, cv2.COLOR_RGB2BGR)
        
        self.display_processed_image()
        self.status_text.SetLabel("圖片處理完成")
        
    except Exception as e:
        wx.MessageBox(f"圖片處理失敗: {str(e)}", "錯(cuò)誤", wx.OK | wx.ICON_ERROR)

def on_color_changed(self, event):
    """顏色選擇器改變事件"""
    color = self.color_picker.GetColour()
    self.current_bg_color = (color.red, color.green, color.blue)
    
    # 如果已經(jīng)有處理后的圖片，重新處理
    if self.processed_image is not None:
        self.on_process_image(event)
 
def set_background_color(self, color):
    """設(shè)置預(yù)設(shè)背景顏色"""
    self.current_bg_color = color
    self.color_picker.SetColour(wx.Colour(*color))
    
    # 實(shí)時(shí)更新處理后的圖片
    if self.processed_image is not None:
        self.on_process_image(None)

def resize_image_for_display(self, image, max_width, max_height):
    """調(diào)整圖片大小以適應(yīng)顯示區(qū)域"""
    height, width = image.shape[:2]
    
    # 計(jì)算縮放比例，保持寬高比
    scale_w = max_width / width
    scale_h = max_height / height
    scale = min(scale_w, scale_h)
    
    if scale < 1:
        new_width = int(width * scale)
        new_height = int(height * scale)
        return cv2.resize(image, (new_width, new_height), interpolation=cv2.INTER_AREA)
    else:
        return image

# 初始狀態(tài)：只有打開按鈕可用
self.process_btn.Enable(False)
self.save_btn.Enable(False)
 
# 加載圖片后：?jiǎn)⒂锰幚戆粹o
self.process_btn.Enable(True)
 
# 處理完成后：?jiǎn)⒂帽４姘粹o
self.save_btn.Enable(True)

try:
    self.original_image = cv2.imread(pathname)
    if self.original_image is None:
        raise ValueError("無(wú)法讀取圖片文件")
    # ... 處理邏輯
except Exception as e:
    wx.MessageBox(f"加載圖片失敗: {str(e)}", "錯(cuò)誤", wx.OK | wx.ICON_ERROR)

# 安裝PyInstaller
pip install pyinstaller
 
# 打包成單個(gè)exe文件
pyinstaller --onefile --windowed --name="證件照摳圖工具" photo_matting.py
 
# 添加自定義圖標(biāo)
pyinstaller --onefile --windowed --icon=app.ico --name="證件照摳圖工具" photo_matting.py

# 預(yù)加載模型，避免每次處理都重新加載
from rembg import new_session
 
class PhotoMattingFrame(wx.Frame):
    def __init__(self):
        super().__init__(...)
        self.rembg_session = new_session('u2net')  # 預(yù)加載模型
        
    def on_process_image(self, event):
        # 使用預(yù)加載的模型
        output = remove(pil_image, session=self.rembg_session)

# wxPython安裝失敗
pip install -U -f https://extras.wxpython.org/wxPython4/extras/linux/gtk3/ubuntu-20.04 wxpython
 
# rembg安裝慢
pip install rembg -i https://pypi.tuna.tsinghua.edu.cn/simple/

# 添加詳細(xì)的錯(cuò)誤日志
import logging
logging.basicConfig(level=logging.DEBUG)
 
try:
    # 主要邏輯
    pass
except Exception as e:
    logging.error(f"處理失敗: {e}", exc_info=True)

def cleanup_images(self):
    """清理內(nèi)存中的圖片數(shù)據(jù)"""
    if hasattr(self, 'original_image') and self.original_image is not None:
        del self.original_image
    if hasattr(self, 'processed_image') and self.processed_image is not None:
        del self.processed_image
    import gc
    gc.collect()