102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
import requests
|
|
import json
|
|
import base64
|
|
import os
|
|
import time
|
|
|
|
API_URL = "http://192.168.0.150:7000/translate_image"
|
|
# API_URL = "http://127.0.0.1:7000/translate_image"
|
|
|
|
# 이미지 파일을 base64로 변환하는 함수
|
|
def image_to_base64(image_path):
|
|
"""이미지 파일을 base64 문자열로 변환"""
|
|
if not os.path.exists(image_path):
|
|
raise FileNotFoundError(f"이미지 파일을 찾을 수 없습니다: {image_path}")
|
|
|
|
with open(image_path, "rb") as image_file:
|
|
encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
|
|
return encoded_string
|
|
|
|
# base64 데이터를 이미지 파일로 저장하는 함수
|
|
def base64_to_image(base64_data, output_path):
|
|
"""base64 문자열을 이미지 파일로 저장"""
|
|
try:
|
|
image_data = base64.b64decode(base64_data)
|
|
with open(output_path, "wb") as image_file:
|
|
image_file.write(image_data)
|
|
return True
|
|
except Exception as e:
|
|
print(f"이미지 저장 중 오류: {e}")
|
|
return False
|
|
|
|
# 이미지 파일 경로
|
|
image_path = "d:/py/IT_Server/modules/img/6.jpg"
|
|
|
|
print("=== 이미지 번역 API 테스트 시작 ===")
|
|
start_total_time = time.time()
|
|
|
|
# 이미지를 base64로 변환
|
|
print("\n1. 이미지 base64 변환 중...")
|
|
start_encode_time = time.time()
|
|
try:
|
|
image_base64 = image_to_base64(image_path)
|
|
encode_time = time.time() - start_encode_time
|
|
print(f" ✓ 이미지 파일 '{image_path}' 를 base64로 변환 완료")
|
|
print(f" ✓ Base64 길이: {len(image_base64):,} 문자")
|
|
print(f" ✓ 인코딩 시간: {encode_time:.3f}초")
|
|
except FileNotFoundError as e:
|
|
print(f" ✗ 오류: {e}")
|
|
exit(1)
|
|
|
|
payload = {
|
|
"image_data": image_base64, # 필드명을 image_data로 수정
|
|
"file_prefix": "test",
|
|
"toggle_states": {"ocr": True},
|
|
"unwanted_texts": {"크리스탈": "크리미"},
|
|
"watermark_text": "테스트 워터마크",
|
|
"watermark_opacity": 0.5,
|
|
"watermark_font_size": 32
|
|
}
|
|
|
|
headers = {"Content-Type": "application/json"}
|
|
|
|
print(f"\n2. API 요청 전송 중... (URL: {API_URL})")
|
|
start_api_time = time.time()
|
|
response = requests.post(API_URL, data=json.dumps(payload), headers=headers)
|
|
api_time = time.time() - start_api_time
|
|
|
|
total_time = time.time() - start_total_time
|
|
|
|
print(f"\n=== 처리 결과 ===")
|
|
print(f"상태 코드: {response.status_code}")
|
|
print(f"API 처리 시간: {api_time:.3f}초")
|
|
print(f"전체 처리 시간: {total_time:.3f}초")
|
|
|
|
try:
|
|
response_data = response.json()
|
|
print(f"\n응답 내용:")
|
|
print(json.dumps(response_data, indent=2, ensure_ascii=False))
|
|
|
|
# 성공적으로 처리된 경우 추가 정보 출력
|
|
if response.status_code == 200:
|
|
print(f"\n=== 성능 요약 ===")
|
|
print(f"• Base64 인코딩: {encode_time:.3f}초")
|
|
print(f"• API 서버 처리: {api_time:.3f}초")
|
|
print(f"• 전체 소요 시간: {total_time:.3f}초")
|
|
if api_time > 10:
|
|
print("⚠️ API 처리 시간이 10초를 초과했습니다.")
|
|
elif api_time > 5:
|
|
print("⚠️ API 처리 시간이 5초를 초과했습니다.")
|
|
else:
|
|
print("✓ 처리 시간이 양호합니다.")
|
|
|
|
# 결과 이미지를 파일로 저장
|
|
output_path = "d:/py/IT_Server/modules/translated_result.png"
|
|
if base64_to_image(response_data["result"], output_path):
|
|
print(f"처리된 이미지가 저장되었습니다: {output_path}")
|
|
else:
|
|
print("이미지 저장에 실패했습니다.")
|
|
|
|
except json.JSONDecodeError:
|
|
print(f"\nJSON 응답이 아닙니다:")
|
|
print(response.text) |