# -*- coding: utf-8 -*-
# 百度智能云 增值税发票OCR API 调用示例
import requests
import base64
import json

# 租户级配置（每个企业独立 API Key）
API_KEY = "your_tenant_api_key"
SECRET_KEY = "your_tenant_secret_key"
TOKEN_URL = "https://aip.baidubce.com/oauth/2.0/token"
OCR_URL = "https://aip.baidubce.com/rest/2.0/ocr/v1/vat_invoice"


def get_access_token():
    """获取百度OCR API访问令牌"""
    params = {
        "grant_type": "client_credentials",
        "client_id": API_KEY,
        "client_secret": SECRET_KEY,
    }
    resp = requests.post(TOKEN_URL, params=params)
    return resp.json()["access_token"]


def recognize_invoice(image_path):
    """识别增值税发票，返回结构化数据"""
    with open(image_path, "rb") as f:
        img_base64 = base64.b64encode(f.read()).decode()

    token = get_access_token()
    response = requests.post(
        OCR_URL,
        params={"access_token": token},
        data={"image": img_base64, "type": "normal"},
    )
    result = response.json()

    # 提取关键字段
    invoice = result.get("words_result", {})
    return {
        "invoice_code": invoice.get("InvoiceCode", ""),
        "invoice_number": invoice.get("InvoiceNum", ""),
        "invoice_date": invoice.get("InvoiceDate", ""),
        "buyer": invoice.get("PurchaserName", ""),
        "seller": invoice.get("SellerName", ""),
        "amount": invoice.get("TotalAmount", ""),
        "tax": invoice.get("TotalTax", ""),
        "total": invoice.get("AmountInFiguers", ""),
    }


if __name__ == "__main__":
    data = recognize_invoice("invoice.jpg")
    print(json.dumps(data, ensure_ascii=False, indent=2))