فهرست منبع

fix: 使用正确的 save_to_markdown 方法保存结果

- 根据官方文档,使用 save_to_markdown 而不是 save
- 简化文件查找逻辑,直接递归查找 .md 文件
- 移除不必要的路径检查
何文松 1 روز پیش
والد
کامیت
c662419077
1فایلهای تغییر یافته به همراه11 افزوده شده و 23 حذف شده
  1. 11 23
      paddleocr_wrapper.py

+ 11 - 23
paddleocr_wrapper.py

@@ -67,35 +67,23 @@ def main():
         
         for idx, item in enumerate(result):
             print(f"处理结果项 {idx}: {type(item)}", file=sys.stderr)
-            print(f"  - 有 save 方法: {hasattr(item, 'save')}", file=sys.stderr)
+            print(f"  - 有 save_to_markdown 方法: {hasattr(item, 'save_to_markdown')}", file=sys.stderr)
             
-            if hasattr(item, 'save'):
-                item_save_path = output_dir / image_basename
-                print(f"  - 保存到: {item_save_path}", file=sys.stderr)
-                item.save(str(item_save_path))
+            if hasattr(item, 'save_to_markdown'):
+                print(f"  - 保存到: {output_dir}", file=sys.stderr)
+                # 使用官方 API 保存 markdown
+                item.save_to_markdown(save_path=str(output_dir))
                 
                 # 列出保存后的文件
-                if item_save_path.exists():
-                    files = list(item_save_path.iterdir())
+                if output_dir.exists():
+                    files = list(output_dir.iterdir())
                     print(f"  - 保存后的文件列表: {[f.name for f in files]}", file=sys.stderr)
                 
-                # 查找生成的 markdown 文件(多种可能的路径)
-                possible_paths = [
-                    item_save_path / f"{image_basename}.md",
-                    output_dir / f"{image_basename}.md",
-                    item_save_path / "result.md",
-                ]
-                
-                for markdown_file in possible_paths:
-                    print(f"  - 检查路径: {markdown_file} -> {markdown_file.exists()}", file=sys.stderr)
-                    if markdown_file.exists():
-                        print(f"SUCCESS: {markdown_file}")
-                        sys.exit(0)
-                
-                # 如果没找到,尝试递归查找所有 .md 文件
+                # 递归查找所有 .md 文件
                 print(f"  - 递归查找 .md 文件...", file=sys.stderr)
-                md_files = list(item_save_path.rglob("*.md")) if item_save_path.exists() else []
-                print(f"  - 找到 {len(md_files)} 个 .md 文件", file=sys.stderr)
+                md_files = list(output_dir.rglob("*.md"))
+                print(f"  - 找到 {len(md_files)} 个 .md 文件: {[f.name for f in md_files]}", file=sys.stderr)
+                
                 if md_files:
                     print(f"SUCCESS: {md_files[0]}")
                     sys.exit(0)