# 創(chuàng)建虛擬環(huán)境
python -m venv ocr_env
source ocr_env/bin/activate  # Linux/Mac
ocr_env\Scripts\activate      # Windows

???????# 安裝依賴
pip install PyQt5==5.15.4 wechat-ocr==0.2.1

class OCRService:
    def __init__(self, base_dir=None):
        self.ocr_manager = None  # 單例模式管理
    
    def process_ocr(self, img_path: str):
        """核心識別邏輯"""
        if not self.ocr_manager:
            self.initialize_ocr_manager()
        
        try:
            self.ocr_manager.DoOCRTask(img_path)
            # 異步等待結果回調(diào)
            while self.ocr_manager.m_task_id.qsize() != OCR_MAX_TASK_ID:
                time.sleep(0.1)
        except Exception as e:
            logger.error(f"OCR失敗: {str(e)}")

class DropArea(QLabel):
    def dropEvent(self, event):
        """處理拖放事件"""
        urls = event.mimeData().urls()
        if urls and urls[0].toLocalFile().endswith(IMAGE_EXTENSIONS):
            self.window().handle_dropped_image(urls[0].toLocalFile())

def copy_text(self):
    """帶格式處理的復制功能"""
    text = self.text_edit.toPlainText()
    if not text.strip():
        self.show_error("無內(nèi)容可復制")
        return
    
    clipboard = QApplication.clipboard()
    clipboard.setText(text)
    # 添加動畫反饋
    self.copy_btn.setText("? 復制成功")
    QTimer.singleShot(1000, lambda: self.copy_btn.setText("?? 一鍵復制"))

import sys
import os
import json
import time
import threading
import gc
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, 
                             QLabel, QTextEdit, QPushButton, QFileDialog, QMenuBar, QMenu, 
                             QStatusBar, QMessageBox)
from PyQt5.QtCore import Qt, QMimeData, QSize, pyqtSignal
from PyQt5.QtGui import QPixmap, QDragEnterEvent, QDropEvent, QIcon, QPalette, QColor, QImage
from wechat_ocr.ocr_manager import OcrManager, OCR_MAX_TASK_ID

# 配置日志
import logging
log_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "WechatOCR/app.log")
logging.basicConfig(level=logging.DEBUG, filename=log_file, filemode='a',
                    format='%(asctime)s - %(levelname)s - %(message)s')

# OCR相關配置
weiXin_ocr_dir = "WechatOCR"
weiXin_dir = "WechatOCR/wx"

class EmojiLabel(QLabel):
    """支持emoji的標簽"""
    def __init__(self, text="", parent=None):
        super().__init__(parent)
        self.setText(text)

def get_base_dir():
    return os.path.dirname(os.path.abspath(__file__))

def get_json_directory():
    exe_dir = get_base_dir()
    json_dir = os.path.join(exe_dir, "WechatOCR/json")
    os.makedirs(json_dir, exist_ok=True)
    return json_dir

def ocr_result_callback(img_path: str, results: dict):
    save_dir = get_json_directory()
    try:
        base_name = os.path.basename(img_path) + ".json"
        full_save_path = os.path.join(save_dir, base_name)

        if os.path.exists(full_save_path):
            timestamp = int(time.time() * 1000)
            backup_file = f"{base_name}_{timestamp}.json"
            backup_save_path = os.path.join(save_dir, backup_file)
            os.rename(full_save_path, backup_save_path)

        with open(full_save_path, 'w', encoding='utf-8') as f:
            f.write(json.dumps(results, ensure_ascii=False, indent=2))
    except Exception as e:
        print(f"Permission denied for directory {save_dir}: {e}")

class OCRService:
    def __init__(self, base_dir=None):
        self.base_dir = base_dir
        self.ocr_manager = None
        if self.base_dir is None:
            self.base_dir = get_base_dir()

    def initialize_ocr_manager(self):
        """初始化 OcrManager，只創(chuàng)建一次"""
        if self.ocr_manager is None:
            print("初始化 OCR 管理器")
            wei_xin = os.path.join(self.base_dir, weiXin_dir)
            ocr_dir = os.path.join(self.base_dir, weiXin_ocr_dir)
            self.ocr_manager = OcrManager(wei_xin)
            self.ocr_manager.SetExePath(ocr_dir)
            self.ocr_manager.SetUsrLibDir(wei_xin)
            self.ocr_manager.SetOcrResultCallback(ocr_result_callback)
            self.ocr_manager.StartWeChatOCR()

    def process_ocr(self, img_path: str):
        """進行 OCR 任務"""
        if self.ocr_manager is None:
            self.initialize_ocr_manager()

        try:
            self.ocr_manager.DoOCRTask(img_path)
            time.sleep(1)  # 等待OCR任務處理

            timeout = 30
            start_time = time.time()

            while self.ocr_manager.m_task_id.qsize() != OCR_MAX_TASK_ID:
                if time.time() - start_time > timeout:
                    print("OCR任務超時!")
                    break
                time.sleep(0.1)

        except Exception as e:
            print(f"OCR請求失敗: {e}")
        finally:
            gc.collect()

    def reset_ocr_manager(self):
        """如果需要，可以重置 OCR 管理器"""
        if self.ocr_manager is not None:
            print("重置 OCR 管理器")
            self.ocr_manager.KillWeChatOCR()
            self.ocr_manager = None  # 清理 OCR 管理器
            gc.collect()

def read_json(file_path):
    """讀取 JSON 文件并返回數(shù)據(jù)"""
    try:
        with open(file_path, 'r', encoding='utf-8') as f:
            data = json.load(f)
        return data
    except FileNotFoundError:
        print(f"File not found: {file_path}")
        return None
    except json.JSONDecodeError as e:
        print(f"Error decoding JSON: {e}")
        return None

def process_ocr_result(data):
    """處理 OCR JSON 數(shù)據(jù)"""
    if not data:
        print("No data to process.")
        return []

    ocr_results = data.get("ocrResult", [])
    result_list = []
    for index, result in enumerate(ocr_results, start=1):
        text = result.get("text", "No text found")
        result_list.append(f"{text}")
    return result_list

def delete_file(file_path):
    """刪除文件的方法"""
    try:
        if os.path.exists(file_path):
            os.remove(file_path)
            print(f"成功刪除文件: {file_path}")
        else:
            print(f"文件不存在: {file_path}")
    except Exception as e:
        print(f"刪除文件失敗: {e}")

class OcrThread(threading.Thread):
    def __init__(self, window, image_path, ocr_service):
        super().__init__()
        self.window = window
        self.image_path = image_path
        self.ocr_service = ocr_service

    def run(self):
        try:
            print("開始識別")
            print(f"self.image_path： {self.image_path}")
            self.ocr_service.process_ocr(self.image_path)
            time.sleep(3)
            file_name = os.path.basename(self.image_path)
            json_dir = get_json_directory()
            file_path = os.path.join(json_dir, file_name + '.json')

            result_list = process_ocr_result(read_json(file_path))
            self.window.update_text_display_signal.emit(result_list, file_path)
            self.window.upload_status_signal.emit("上傳成功! ??")

        except Exception as e:
            self.window.error_signal.emit(f"識別失敗: {str(e)}")

class DropArea(QLabel):
    """自定義拖放區(qū)域"""
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setAlignment(Qt.AlignCenter)
        self.setStyleSheet("""
            QLabel {
                border: 3px dashed #aaa;
                border-radius: 10px;
                padding: 20px;
                font-size: 16px;
                color: #666;
            }
            QLabel:hover {
                border-color: #4CAF50;
                background-color: rgba(76, 175, 80, 0.1);
            }
        """)
        self.setText("拖拽圖片到這里\n或者\n粘貼圖片 (Ctrl+V)")
        self.setAcceptDrops(True)

    def dragEnterEvent(self, event: QDragEnterEvent):
        if event.mimeData().hasUrls():
            event.acceptProposedAction()

    def dropEvent(self, event: QDropEvent):
        urls = event.mimeData().urls()
        if urls:
            file_path = urls[0].toLocalFile()
            if file_path.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp')):
                # 獲取主窗口引用
                main_window = self.window()
                if isinstance(main_window, MainWindow):
                    main_window.handle_dropped_image(file_path)

class MainWindow(QMainWindow):
    # 定義信號
    update_text_display_signal = pyqtSignal(list, str)
    upload_status_signal = pyqtSignal(str)
    error_signal = pyqtSignal(str)
    
    def __init__(self):
        super().__init__()
        
        # 初始化窗口置頂狀態(tài)
        self.always_on_top = False
        
        # 全局OCR服務實例
        self.ocr_service = OCRService()
        
        self.setWindowTitle("WechatOCR ")
        self.setWindowIcon(QIcon.fromTheme("accessories-text-editor"))
        self.resize(1000, 600)
        
        # 設置現(xiàn)代UI風格和配色方案
        self.set_modern_style()
        
        # 創(chuàng)建主窗口部件
        self.create_widgets()
        
        # 創(chuàng)建菜單欄
        self.create_menubar()
        
        # 創(chuàng)建狀態(tài)欄
        self.statusBar().showMessage("準備就緒 ??")
        
        # 連接信號槽
        self.connect_signals()
    
    def set_modern_style(self):
        """設置現(xiàn)代UI風格和配色方案"""
        palette = QPalette()
        palette.setColor(QPalette.Window, QColor(240, 240, 240))
        palette.setColor(QPalette.WindowText, QColor(50, 50, 50))
        palette.setColor(QPalette.Base, QColor(255, 255, 255))
        palette.setColor(QPalette.AlternateBase, QColor(240, 240, 240))
        palette.setColor(QPalette.ToolTipBase, QColor(255, 255, 255))
        palette.setColor(QPalette.ToolTipText, QColor(50, 50, 50))
        palette.setColor(QPalette.Text, QColor(50, 50, 50))
        palette.setColor(QPalette.Button, QColor(240, 240, 240))
        palette.setColor(QPalette.ButtonText, QColor(50, 50, 50))
        palette.setColor(QPalette.BrightText, QColor(255, 0, 0))
        palette.setColor(QPalette.Highlight, QColor(76, 175, 80))
        palette.setColor(QPalette.HighlightedText, QColor(255, 255, 255))
        QApplication.setPalette(palette)
        
        self.setStyleSheet("""
            QMainWindow {
                background-color: #f0f0f0;
            }
            QTextEdit {
                border: 1px solid #ccc;
                border-radius: 5px;
                padding: 10px;
                font-size: 14px;
                selection-background-color: #4CAF50;
            }
            QPushButton {
                background-color: #4CAF50;
                color: white;
                border: none;
                padding: 8px 16px;
                text-align: center;
                text-decoration: none;
                font-size: 14px;
                margin: 4px 2px;
                border-radius: 4px;
            }
            QPushButton:hover {
                background-color: #45a049;
            }
            QPushButton:pressed {
                background-color: #3e8e41;
            }
        """)
    
    def create_widgets(self):
        """創(chuàng)建主窗口部件"""
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        
        # 主布局
        main_layout = QHBoxLayout(central_widget)
        main_layout.setContentsMargins(20, 20, 20, 20)
        main_layout.setSpacing(20)
        
        # 左側面板 - 圖片上傳和顯示
        left_panel = QWidget()
        left_layout = QVBoxLayout(left_panel)
        left_layout.setContentsMargins(0, 0, 0, 0)
        left_layout.setSpacing(20)
        
        # 拖放區(qū)域
        self.drop_area = DropArea()
        self.drop_area.setMinimumSize(400, 200)
        left_layout.addWidget(self.drop_area)
        
        # 圖片顯示區(qū)域
        self.image_label = QLabel()
        self.image_label.setAlignment(Qt.AlignCenter)
        self.image_label.setStyleSheet("""
            QLabel {
                border: 1px solid #ddd;
                border-radius: 5px;
                background-color: white;
            }
        """)
        self.image_label.setMinimumSize(400, 300)
        left_layout.addWidget(self.image_label)
        
        # 上傳按鈕
        self.upload_btn = QPushButton("?? 選擇圖片 ")
        self.upload_btn.setIconSize(QSize(20, 20))
        left_layout.addWidget(self.upload_btn)
        
        # 右側面板 - 文本顯示
        right_panel = QWidget()
        right_layout = QVBoxLayout(right_panel)
        right_layout.setContentsMargins(0, 0, 0, 0)
        right_layout.setSpacing(10)
        
        # 文本顯示區(qū)域
        self.text_edit = QTextEdit()
        self.text_edit.setReadOnly(True)
        self.text_edit.setStyleSheet("""
            QTextEdit {
                font-size: 14px;
                line-height: 1.5;
            }
        """)
        right_layout.addWidget(self.text_edit, 1)
        
        # 添加復制按鈕
        self.copy_btn = QPushButton("?? 一鍵復制 ")
        self.copy_btn.setIconSize(QSize(20, 20))
        self.copy_btn.setStyleSheet("""
            QPushButton {
                background-color: #2196F3;
                color: white;
                border: none;
                padding: 8px 16px;
                text-align: center;
                font-size: 14px;
                border-radius: 4px;
            }
            QPushButton:hover {
                background-color: #0b7dda;
            }
            QPushButton:pressed {
                background-color: #0a68b4;
            }
        """)
        right_layout.addWidget(self.copy_btn)
        
        # 添加到主布局
        main_layout.addWidget(left_panel, 1)
        main_layout.addWidget(right_panel, 1)
    
    def create_menubar(self):
        """創(chuàng)建菜單欄"""
        menubar = self.menuBar()
        
        # 文件菜單
        file_menu = menubar.addMenu("??? 文件 ")
        
        open_action = file_menu.addAction("??? 打開圖片 ")
        open_action.setShortcut("Ctrl+O")
        open_action.triggered.connect(self.open_image)
        
        always_on_top_action = file_menu.addAction("?? 窗口置頂 ")
        always_on_top_action.setCheckable(True)
        always_on_top_action.setChecked(self.always_on_top)
        always_on_top_action.triggered.connect(self.toggle_always_on_top)
        
        exit_action = file_menu.addAction("?? 退出 ")
        exit_action.setShortcut("Ctrl+Q")
        exit_action.triggered.connect(self.close)
        
        # 編輯菜單
        edit_menu = menubar.addMenu("?? 編輯 ")
        
        copy_action = edit_menu.addAction("?? 復制 ")
        copy_action.setShortcut("Ctrl+C")
        copy_action.triggered.connect(self.copy_text)
        
        clear_action = edit_menu.addAction("??? 清空 ")
        clear_action.triggered.connect(self.clear_text)
        
        # 幫助菜單
        help_menu = menubar.addMenu("? 幫助 ")
        
        about_action = help_menu.addAction("?? 關于 ")
        about_action.triggered.connect(self.show_about)
    
    def connect_signals(self):
        """連接信號槽"""
        self.upload_btn.clicked.connect(self.open_image)
        self.copy_btn.clicked.connect(self.copy_text)
        
        # 連接自定義信號
        self.update_text_display_signal.connect(self.update_text_display)
        self.upload_status_signal.connect(self.update_status)
        self.error_signal.connect(self.show_error)
    
    def open_image(self):
        """打開圖片文件"""
        file_path, _ = QFileDialog.getOpenFileName(
            self, "選擇圖片", "", 
            "圖片文件 (*.png *.jpg *.jpeg *.bmp);;所有文件 (*.*)"
        )
        
        if file_path:
            self.handle_dropped_image(file_path)
    
    def handle_dropped_image(self, file_path):
        """處理拖放或選擇的圖片"""
        pixmap = QPixmap(file_path)
        if pixmap.isNull():
            self.show_error("無法加載圖片，請檢查文件格式")
            return
        
        # 縮放圖片以適應顯示區(qū)域
        scaled_pixmap = pixmap.scaled(
            self.image_label.width(), self.image_label.height(),
            Qt.KeepAspectRatio, Qt.SmoothTransformation
        )
        self.image_label.setPixmap(scaled_pixmap)
        self.update_status("正在識別，請稍等... ?")
        
        # 啟動OCR線程
        ocr_thread = OcrThread(self, file_path, self.ocr_service)
        ocr_thread.start()
    
    def update_text_display(self, result_list, file_path):
        """更新文本顯示"""
        if not result_list:
            self.text_edit.setPlainText("沒有識別到文本內(nèi)容")
        else:
            self.text_edit.setPlainText("\n".join(result_list))
        
        delete_file(file_path)
        self.update_status("? 識別完成! ")
    
    def update_status(self, message):
        """更新狀態(tài)欄"""
        self.statusBar().showMessage(message)
    
    def show_error(self, message):
        """顯示錯誤信息"""
        QMessageBox.critical(self, "錯誤", message)
        self.update_status("操作失敗 ?")
    
    def copy_text(self):
        """復制文本到剪貼板"""
        clipboard = QApplication.clipboard()
        clipboard.setText(self.text_edit.toPlainText())
        self.update_status("文本已復制到剪貼板 ??")
    
    def clear_text(self):
        """清空文本"""
        self.text_edit.clear()
        self.image_label.clear()
        self.update_status("?? 已清空內(nèi)容 ")
    
    def toggle_always_on_top(self):
        """切換窗口置頂狀態(tài)"""
        self.always_on_top = not self.always_on_top
        self.setWindowFlag(Qt.WindowStaysOnTopHint, self.always_on_top)
        self.show()
        
        if self.always_on_top:
            self.update_status("?? 窗口已置頂 ")
        else:
            self.update_status("取消窗口置頂")
    
    def show_about(self):
        """顯示關于對話框"""
        about_text = """
        <h2>WechatOCR ???????</h2>
        <p>版本: 1.0.0</p>
        <p>使用微信OCR引擎實現(xiàn)的圖片文字識別工具</p>
        <p>功能:</p>
        <ul>
            <li>支持拖放圖片識別</li>
            <li>支持粘貼圖片識別 (Ctrl+V)</li>
            <li>支持窗口置頂</li>
            <li>簡潔現(xiàn)代的UI界面</li>
            <li>一鍵復制識別結果</li>
        </ul>
        <p>? 2025 創(chuàng)客白澤-WechatOCR 項目</p>
        """
        QMessageBox.about(self, "關于 WechatOCR", about_text)
    
    def keyPressEvent(self, event):
        """處理鍵盤事件"""
        # 處理粘貼圖片 (Ctrl+V)
        if event.modifiers() == Qt.ControlModifier and event.key() == Qt.Key_V:
            clipboard = QApplication.clipboard()
            mime_data = clipboard.mimeData()
            
            if mime_data.hasImage():
                # 從剪貼板獲取圖片
                image = clipboard.image()
                if not image.isNull():
                    # 保存臨時圖片文件
                    temp_path = os.path.join(get_json_directory(), "clipboard_temp.png")
                    image.save(temp_path)
                    self.handle_dropped_image(temp_path)
                    return
            
            elif mime_data.hasUrls():
                # 處理文件路徑
                urls = mime_data.urls()
                if urls:
                    file_path = urls[0].toLocalFile()
                    if file_path.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp')):
                        self.handle_dropped_image(file_path)
                        return
        
        super().keyPressEvent(event)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    
    # 設置應用程序字體
    font = app.font()
    font.setPointSize(12)
    app.setFont(font)
    
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

   # 添加銳化處理
   kernel = np.array([[0, -1, 0], [-1, 5, -1], [0, -1, 0]])
   cv2.filter2D(img, -1, kernel)