Bladeren bron

fix: 改进 paddleocr_wrapper.py 的 markdown 文件查找逻辑

- 添加多种可能的路径查找
- 支持递归查找所有 .md 文件
- 添加调试信息输出
- 改进错误提示
何文松 1 dag geleden
bovenliggende
commit
54fbc7fa71
1 gewijzigde bestanden met toevoegingen van 20 en 4 verwijderingen
  1. 20 4
      paddleocr_wrapper.py

+ 20 - 4
paddleocr_wrapper.py

@@ -66,14 +66,30 @@ def main():
             if hasattr(item, 'save'):
                 item_save_path = output_dir / image_basename
                 item.save(str(item_save_path))
+                print(f"结果已保存到: {item_save_path}", file=sys.stderr)
                 
-                # 查找生成的 markdown 文件
-                markdown_file = item_save_path / f"{image_basename}.md"
-                if markdown_file.exists():
-                    print(f"SUCCESS: {markdown_file}")
+                # 查找生成的 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:
+                    if markdown_file.exists():
+                        print(f"SUCCESS: {markdown_file}")
+                        sys.exit(0)
+                
+                # 如果没找到,尝试递归查找所有 .md 文件
+                import glob
+                md_files = list(item_save_path.rglob("*.md"))
+                if md_files:
+                    print(f"SUCCESS: {md_files[0]}")
                     sys.exit(0)
         
         print("错误: 未找到 markdown 输出文件", file=sys.stderr)
+        print(f"输出目录: {output_dir}", file=sys.stderr)
+        print(f"保存路径: {item_save_path if 'item_save_path' in locals() else 'N/A'}", file=sys.stderr)
         sys.exit(1)
         
     except ImportError as e: