import requests
import json
import time
import os
API_URL = "http://127.0.0.1:8008"

# 테스트할 이미지 번호 (1~7)
img_num = 5  # 필요시 1~7로 변경
img_path = os.path.join(os.path.dirname(__file__), "modules","img",f"{img_num}.jpg")

# 저장할 결과 이미지 경로
output_image_path = os.path.join(os.path.dirname(img_path), f"result_{img_num}.png")

# # /translate_me 테스트
# translate_payload = {
#     "local_image_path": img_path,
#     "unwanted_words": {"테스트": "치환값", "삭제": "이미지삭제"},
#     "method": "lama",
#     "index": 0,
#     "delay": 1.0,
#     "output_image_path": output_image_path,
#     "toggle_states": {"ocr": True, "watermark_text": "테스트워터마크", "remove_background_white": True}
# }

# print("/translate_me 요청 시작...")
# start_time = time.time()
# resp = requests.post(f"{API_URL}/translate_me", json=translate_payload, timeout=40)
# elapsed = time.time() - start_time
# print(f"/translate_me 결과 (소요 시간: {elapsed:.2f}초):")
# print(json.dumps(resp.json(), ensure_ascii=False, indent=2))

# /remove_bg_me 테스트
remove_bg_payload = {
    "local_image_path": img_path,
    "index": 0,
    "output_image_path": output_image_path,
    "toggle_states": {"remove_background_white": True}
}

print("\n/remove_bg_me 요청 시작...")
start_time = time.time()
resp2 = requests.post(f"{API_URL}/remove_bg_me", json=remove_bg_payload, timeout=40)
elapsed2 = time.time() - start_time
print(f"/remove_bg_me 결과 (소요 시간: {elapsed2:.2f}초):")
print(json.dumps(resp2.json(), ensure_ascii=False, indent=2))

# 이미지 번호만 바꿔가며 테스트 가능