import subprocess
import os

# 定义图像路径
image_dir = "./test_img/69.jpg"

# 定义模型路径
det_model_v2 = "./inference_model/det_v2"
det_model_v4 = "./inference_model/det_v4"
rec_model_v2 = "./inference_model/rec_v2"
rec_model_v4 = "./inference_model/rec_v4"
cls_model = "./inference_model/ch_ppstructure_mobile_v2.0_SLANet_infer"

# 定义图像保存路径
save_dir_v2 = "./inference_results/v2"
save_dir_v4 = "./inference_results/v4"

# 检查路径是否存在
if not os.path.exists(image_dir):
    raise FileNotFoundError(f"Image directory not found: {image_dir}")
if not os.path.exists(det_model_v2) or not os.path.exists(rec_model_v2):
    raise FileNotFoundError("det_v2 or rec_v2 model directory not found")
if not os.path.exists(det_model_v4) or not os.path.exists(rec_model_v4):
    raise FileNotFoundError("det_v4 or rec_v4 model directory not found")

# 调用 v2 模型
command_v2 = [
    "python",
    "tools/infer/predict_system.py",
    "--image_dir", image_dir,
    "--det_model_dir", det_model_v2,
    "--rec_model_dir", rec_model_v2,
    "--cls_model_dir", cls_model,
    "--draw_img_save_dir", save_dir_v2,
    # "--convert_to_gray",
    # "--enable_preprocess",
]

s|u|b|p|r|o|c|e|s|s.run(command_v2, check=True)
print(f"[INFO] Model v2 results saved in: {save_dir_v2}")

# 调用 v4 模型
command_v4 = [
    "python",
    "tools/infer/predict_system.py",
    "--image_dir", image_dir,
    "--det_model_dir", det_model_v4,
    "--rec_model_dir", rec_model_v4,
    "--cls_model_dir", cls_model,
    "--draw_img_save_dir", save_dir_v4,
    # "--convert_to_gray",
    # "--enable_preprocess",
]
s|u|b|p|r|o|c|e|s|s.run(command_v4, check=True)
print(f"[INFO] Model v4 results saved in: {save_dir_v4}")