.

test/api.py

@@ -0,0 +1,62 @@
+import requests
+import time
+import base64
+
+def test_translate_image_api(image_url, use_celery=False):
+    url = "http://cckb9998.synology.me:8001/translate_image/"
+    payload = {
+        "image_url": image_url,
+        "use_celery": use_celery
+    }
+    try:
+        response = requests.post(url, json=payload)
+        if response.status_code == 200:
+            print(f"Request successful: {response.json()}")
+            return response.json()
+        else:
+            print(f"Error: {response.status_code}, {response.text}")
+    except requests.exceptions.RequestException as e:
+        print(f"Request failed: {e}")
+
+def check_task_status(task_id):
+    url = f"http://cckb9998.synology.me:8001/task_status/{task_id}"
+    try:
+        response = requests.get(url)
+        if response.status_code == 200:
+            result = response.json()
+            print(f"Task status: {result}")
+            if result.get("status") == "Success" and "result" in result:
+                # 번역된 이미지를 base64로 받아 로컬에 저장
+                save_image(result["result"], f"translated_image_{task_id}.png")
+        else:
+            print(f"Error: {response.status_code}, {response.text}")
+    except requests.exceptions.RequestException as e:
+        print(f"Request failed: {e}")
+
+def save_image(image_base64, file_name):
+    try:
+        # base64 디코딩하여 이미지 저장
+        image_data = base64.b64decode(image_base64)
+        with open(file_name, "wb") as file:
+            file.write(image_data)
+        print(f"Image saved successfully as {file_name}")
+    except Exception as e:
+        print(f"Failed to save image: {e}")
+
+if __name__ == "__main__":
+
+    # 테스트할 이미지 URL
+    image_url = "https://file.percenty.co.kr/public/652bed8e865b1f32ea62bf1f/products/66f7489b86608659ad011faf/0fbf1050-0a49-4c37-83b5-6279b66d6e48.jpg"
+
+    # API 호출
+    result = test_translate_image_api(image_url, use_celery=True)
+
+    # 작업 ID가 있는 경우 상태를 조회
+    if result and "task_id" in result:
+        task_id = result["task_id"]
+
+        # 상태 확인을 위한 대기
+        time.sleep(5)  # Celery 작업이 끝나기를 잠시 대기
+
+        # 작업 상태 확인
+        check_task_status(task_id)