#!/usr/bin/env python3
"""
macOS OCR Helper - Python Skill Implementation
直接调用 macOS Vision 框架进行 OCR，不依赖 MCP 服务器
"""
import sys
import os
import json
# 添加项目根目录到路径
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..")))
from src.ocr import recognize_text, recognize_text_with_layout
def main():
if len(sys.argv) < 3:
print(json.dumps({"error": "Usage: python main.py <tool> <image_path>"}))
sys.exit(1)
tool = sys.argv[1]
image_path = sys.argv[2]
try:
if not os.path.exists(image_path):
print(json.dumps({"error": f"File not found: {image_path}"}))
sys.exit(1)
if tool == "read_image_text":
result = recognize_text(image_path)
print(result)
elif tool == "read_image_layout":
result = recognize_text_with_layout(image_path)
print(json.dumps(result, ensure_ascii=False))
else:
print(json.dumps({"error": f"Unknown tool: {tool}"}))
sys.exit(1)
except Exception as e:
print(json.dumps({"error": str(e)}))
sys.exit(1)
if __name__ == "__main__":
main()
