画像キャプチャをPythonで解決する方法

pip install capsolver

import capsolver
import base64
import requests
from io import BytesIO

# -------------------これらの値を変更してください-------------------
# CapSolverダッシュボードから取得したAPIキー
CAPSOLVER_API_KEY = "Your_API_KEY"
# ----------------------------------------------------------------

def setup_capsolver():
    """CapSolver APIキーを設定します。"""
    capsolver.api_key = CAPSOLVER_API_KEY

def get_base64_from_image(source, is_url=False):
    """
    画像ファイルまたはURLをBase64エンコードされた文字列に変換します。
    :param source: 画像のパス（ローカルファイルの場合）またはURL。
    :param is_url: ソースがURLの場合にTrue。
    :return: Base64エンコードされた文字列、または失敗時はNone。
    """
    try:
        if is_url:
            response = requests.get(source)
            response.raise_for_status()  # 不正なステータスコードに対して例外を発生させます
            image_bytes = BytesIO(response.content)
        else:
            with open(source, 'rb') as image_file:
                image_bytes = BytesIO(image_file.read())
        
        base64_string = base64.b64encode(image_bytes.read()).decode('utf-8')
        return base64_string
    except requests.exceptions.RequestException as e:
        print(f"URLから画像を取得できませんでした: {e}")
        return None
    except FileNotFoundError:
        print(f"エラー: {source}にファイルが見つかりません")
        return None
    except Exception as e:
        print(f"画像をBase64に変換中にエラーが発生しました: {e}")
        return None

def solve_image_captcha(base64_image, module_name="common"):
    """
    CapSolverを使用して画像キャプチャを解決します。
    :param base64_image: Base64エンコードされた画像文字列。
    :param module_name: 認識モジュール、'common'は汎用的なモジュールです。
    :return: 解決オブジェクト、または失敗時はNone。
    """
    print("CapSolverに画像キャプチャタスクを提出しています...")
    try:
        solution = capsolver.solve({
            "type": "ImageToTextTask",
            "module": module_name,
            "body": base64_image
        })
        return solution
    except Exception as e:
        print(f"キャプチャを解決中にエラーが発生しました: {e}")
        return None

def main():
    """プロセスを実行するメイン関数です。"""
    setup_capsolver()

    # --- 例1: ローカル画像ファイルを使用する ---
    print("--- 例1: ローカル画像ファイルを使用する ---")
    local_image_path = "path/to/your/captcha.jpg"  # お使いの画像パスに置き換えてください
    base64_from_file = get_base64_from_image(local_image_path)
    
    if base64_from_file:
        solution = solve_image_captcha(base64_from_file)
        if solution:
            print("解決テキスト:", solution.get("text"))

    print("\n" + "="*30 + "\n")

    # --- 例2: 画像URLを使用する ---
    print("--- 例2: 画像URLを使用する ---")
    image_url = "https://i.postimg.cc/B6hK2V19/captcha-example.png" # 例の画像URL
    base64_from_url = get_base64_from_image(image_url, is_url=True)

    if base64_from_url:
        solution = solve_image_captcha(base64_from_url)
        if solution:
            print("解決テキスト:", solution.get("text"))


if __name__ == "__main__":
    main()