Contents Moderationで画像認識 - Alibaba Cloud

#coding=utf-8

from aliyunsdkcore import client
from aliyunsdkgreen.request.v20180509 import ImageSyncScanRequest
from aliyunsdkgreen.request.extension import HttpContentHelper
import os
import sys
import json
import uuid

# Alibaba Cloudの認証情報
accessKeyId = os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID']
accessKeySecret = os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET']

# クライアントの初期化
clt = client.AcsClient(accessKeyId, accessKeySecret, "ap-southeast-1")

# ファイルからURLを読み込む関数
def load_urls_from_file(file_path):
    with open(file_path, 'r') as file:
        return [line.strip() for line in file.readlines()]

# 複数のファイルを処理するメイン処理
def process_images(urls):
    for url in urls:
        request = ImageSyncScanRequest.ImageSyncScanRequest()
        request.set_accept_format('JSON')
        
        # 1つのタスクを作成
        task = {"dataId": str(uuid.uuid1()),
                "url": url
               }
        print(f"Processing task for URL: {url}")
        
        # リクエストの内容を設定
        request.set_content(HttpContentHelper.toValue({"tasks": [task],
                                                       "scenes": ["ocr"]
                                                      }))
        # API呼び出し
        response = clt.do_action_with_exception(request)
        result = json.loads(response)
        
        # 結果の処理
        if 200 == result["code"]:
            taskResults = result["data"]
            for taskResult in taskResults:
                if (200 == taskResult["code"]):
                    sceneResults = taskResult["results"]
                    print(sceneResults)
                else:
                    print(f"Task failed: {taskResult['code']}")
        else:
            print(f"Request failed: {result['code']}")

if __name__ == "__main__":
    # コマンドライン引数からテキストファイルパスを取得
    if len(sys.argv) < 2:
        print("Usage: python script.py <file_path>")
        sys.exit(1)

    file_path = sys.argv[1]

    # ファイルからURLを読み込み、処理を開始
    urls = load_urls_from_file(file_path)
    process_images(urls)

[{'label': 'ocr', 'ocrData': ['デモ申込書,フリガナ,氏名,生年月日,電話番号,携帯電話,フリガナ ,123-0987,住所,信考'], 'ocrLocations': [{'h': 20.0, 'text': 'デモ申込書', 'w': 96.0, 'x': 234.0, 'y': 72.0}, {'h': 12.0, 'text': 'フリガナ', 'w': 47.0, 'x': 66.0, 'y': 137.0}, {'h': 13.0, 'text': '氏名', 'w': 44.0, 'x': 66.0, 'y': 167.0}, {'h': 12.0, 'text': '生年月日', 'w': 46.0, 'x': 66.0, 'y': 204.0}, {'h': 13.0, 'text': '電話番号', 'w': 48.0, 'x': 65.0, 'y': 273.0}, {'h': 13.0, 'text': '携帯電話', 'w': 51.0, 'x': 297.0, 'y': 273.0}, {'h': 11.0, 'text': 'フリガナ ', 'w': 47.0, 'x': 66.0, 'y': 302.0}, {'h': 11.0, 'text': '123-0987', 'w': 61.0, 'x': 130.0, 'y': 330.0}, {'h': 12.0, 'text': '住所', 'w': 42.0, 'x': 67.0, 'y': 351.0}, {'h': 12.0, 'text': '信考', 'w': 43.0, 'x': 64.0, 'y': 465.0}], 'rate': 99.91, 'scene': 'ocr', 'suggestion': 'review'}]