Conversation content extraction

#!/usr/bin/env python
# coding=utf-8

import os
import json
import datetime
from aliyunsdkcore.client import AcsClient
from aliyunsdkcore.request import CommonRequest
from aliyunsdkcore.auth.credentials import AccessKeyCredential


def create_common_request(domain, version, protocolType, method, uri):
    request = CommonRequest()
    request.set_accept_format('json')
    request.set_domain(domain)
    request.set_version(version)
    request.set_protocol_type(protocolType)
    request.set_method(method)
    request.set_uri_pattern(uri)
    request.add_header('Content-Type', 'application/json')
    return request


def init_parameters():
    root = dict()
    root['AppKey'] = 'Enter the AppKey that you created in the Tingwu console'

    # Basic request parameters
    input = dict()
    input['SourceLanguage'] = 'cn'
    input['TaskKey'] = 'task' + datetime.datetime.now().strftime('%Y%m%d%H%M%S')
    input['FileUrl'] = 'Enter the URL of the audio file to test'
    root['Input'] = input

    # AI-related parameters. Set them as needed.
    parameters = dict()
    # Conversation content extraction
    parameters['ContentExtractionEnabled'] = True
    content_extraction = {
        "SceneIntroduction": "Offline car outlet sales scenario",
        "ExtractionContents": [
            {
                "Title": "Customer price objections",
                "Content": "Customer mentions car model prices, promotional offers, or concerns about future price drops",
                "Identity": "Customer"
            },
            {
                "Title": "Competitor model feedback",
                "Content": "Summarize all competitor brands or models that the customer explicitly mentioned, and summarize the customer's thoughts on them",
                "Identity": "Customer"
            },
            {
                "Title": "Sales script for guiding purchase",
                "Content": "Extract the sales script used to guide the customer to purchase a car from the conversation",
                "Identity": "Sales"
            }
        ]
    }
    parameters['ContentExtraction'] = content_extraction
    root['Parameters'] = parameters
    return root


body = init_parameters()
print(body)

# TODO: Set your AccessKeyId and AccessKeySecret as environment variables.
credentials = AccessKeyCredential(os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID'], os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET'])
client = AcsClient(region_id='cn-beijing', credential=credentials)

request = create_common_request('tingwu.cn-beijing.aliyuncs.com', '2023-09-30', 'https', 'PUT', '/openapi/tingwu/v2/tasks')
request.add_query_param('type', 'offline')

request.set_content(json.dumps(body).encode('utf-8'))
response = client.do_action_with_exception(request)
print("response: \n" + json.dumps(json.loads(response), indent=4, ensure_ascii=False))

package com.alibaba.tingwu.client.demo.aitest;

import com.alibaba.fastjson.JSONObject;
import com.aliyuncs.CommonRequest;
import com.aliyuncs.CommonResponse;
import com.aliyuncs.DefaultAcsClient;
import com.aliyuncs.IAcsClient;
import com.aliyuncs.exceptions.ClientException;
import com.aliyuncs.http.FormatType;
import com.aliyuncs.http.MethodType;
import com.aliyuncs.http.ProtocolType;
import com.aliyuncs.profile.DefaultProfile;
import org.junit.Test;

/**
 * @author tingwu2023
 */
public class ContentExtractionTest {

    @Test
    public void testContentExtraction() throws ClientException {
        CommonRequest request = createCommonRequest("tingwu.cn-beijing.aliyuncs.com", "2023-09-30", ProtocolType.HTTPS, MethodType.PUT, "/openapi/tingwu/v2/tasks");
        request.putQueryParameter("type", "offline");

        JSONObject root = new JSONObject();
        root.put("AppKey", "Enter the AppKey that you created in the Tingwu console");

        JSONObject input = new JSONObject();
        input.fluentPut("FileUrl", "Enter the URL of the audio file to test")
                .fluentPut("SourceLanguage", "cn")
                .fluentPut("TaskKey", "task" + System.currentTimeMillis());
        root.put("Input", input);

        JSONObject parameters = new JSONObject();
        parameters.put("ContentExtractionEnabled", true);
        JSONObject contentExtraction = new JSONObject();
        contentExtraction.fluentPut("SceneIntroduction", "Offline car outlet sales scenario")
                .fluentPut("ExtractionContents", new JSONArray()
                        .fluentAdd(new JSONObject().fluentPut("Title", "Customer price objections").fluentPut("Content", "Customer mentions car model prices, promotional offers, or concerns about future price drops").fluentPut("Identity", "Customer"))
                        .fluentAdd(new JSONObject().fluentPut("Title", "Competitor model feedback").fluentPut("Content", "Summarize all competitor brands or models that the customer explicitly mentioned, and summarize the customer's thoughts on them"))
                        .fluentAdd(new JSONObject().fluentPut("Title", "Sales script for guiding purchase").fluentPut("Content", "Extract the sales script used to guide the customer to purchase a car from the conversation").fluentPut("Identity", "Sales"))
                );
        parameters.put("ContentExtraction", contentExtraction);
        root.put("Parameters", parameters);
        System.out.println(root.toJSONString());
        request.setHttpContent(root.toJSONString().getBytes(), "utf-8", FormatType.JSON);

        // TODO: Set your AccessKeyId and AccessKeySecret as environment variables.
        DefaultProfile profile = DefaultProfile.getProfile("cn-beijing", System.getenv("ALIBABA_CLOUD_ACCESS_KEY_ID"), System.getenv("ALIBABA_CLOUD_ACCESS_KEY_SECRET"));
        IAcsClient client = new DefaultAcsClient(profile);
        CommonResponse response = client.getCommonResponse(request);
        System.out.println(response.getData());
    }

    public static CommonRequest createCommonRequest(String domain, String version, ProtocolType protocolType, MethodType method, String uri) {
        // Create an API request and set its parameters.
        CommonRequest request = new CommonRequest();
        request.setSysDomain(domain);
        request.setSysVersion(version);
        request.setSysProtocol(protocolType);
        request.setSysMethod(method);
        request.setSysUriPattern(uri);
        request.setHttpContentType(FormatType.JSON);
        return request;
    }
}