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

from Config.Config import ALY_AK, ALY_SK


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():
    body = dict()
    body['AppKey'] = 'VQRY3Ij1D9MZvym6'

    # 基本请求参数
    input = dict()
    input['SourceLanguage'] = 'cn'
    input['TaskKey'] = 'task' + datetime.datetime.now().strftime('%Y%m%d%H%M%S')
    # https://yx.ccsjy.cn/ChangChunCloudSchool/index.html#/course-detail/831993494a63a7087ac1c752dcd25d19
    # 《数轴的复习》
    input['FileUrl'] = 'https://ccschool.edusoa.com/cloud_file/project/ccyx-0012/material/26/7c/267c936afcf3e965d463f65c7275d60b.mp4'
    body['Input'] = input

    # AI相关参数，按需设置即可
    parameters = dict()

    # 音视频转换相关
    transcoding = dict()
    # 将原音视频文件转成mp3文件，用以后续浏览器播放
    # transcoding['TargetAudioFormat'] = 'mp3'
    # transcoding['SpectrumEnabled'] = False
    # parameters['Transcoding'] = transcoding

    # 语音识别控制相关
    transcription = dict()
    # 角色分离 ： 可选
    transcription['DiarizationEnabled'] = True
    diarization = dict()
    diarization['SpeakerCount'] = 2
    transcription['Diarization'] = diarization
    parameters['Transcription'] = transcription

    # 文本翻译控制相关 ： 可选
    parameters['TranslationEnabled'] = True
    translation = dict()
    translation['TargetLanguages'] = ['en']  # 假设翻译成英文
    parameters['Translation'] = translation

    # 章节速览相关 ： 可选，包括： 标题、议程摘要
    parameters['AutoChaptersEnabled'] = True

    # 智能纪要相关 ： 可选，包括： 待办、关键信息(关键词、重点内容、场景识别)
    parameters['MeetingAssistanceEnabled'] = True
    meetingAssistance = dict()
    meetingAssistance['Types'] = ['Actions', 'KeyInformation']
    parameters['MeetingAssistance'] = meetingAssistance

    # 摘要控制相关 ： 可选，包括： 全文摘要、发言人总结摘要、问答摘要(问答回顾)
    parameters['SummarizationEnabled'] = True
    summarization = dict()
    summarization['Types'] = ['Paragraph', 'Conversational', 'QuestionsAnswering', 'MindMap']
    parameters['Summarization'] = summarization

    # ppt抽取和ppt总结 ： 可选
    parameters['PptExtractionEnabled'] = True

    # 口语书面化 ： 可选
    parameters['TextPolishEnabled'] = True

    # 大模型后处理任务全局参数 ： 可选
    parameters['Model'] = 'qwq'
    parameters['LlmOutputLanguage'] = 'en'

    body['Parameters'] = parameters
    return body


body = init_parameters()
print(body)

# TODO  请通过环境变量设置您的 AccessKeyId 和 AccessKeySecret
credentials = AccessKeyCredential(ALY_AK, ALY_SK)
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))


"""
{'AppKey': 'VQRY3Ij1D9MZvym6', 'Input': {'SourceLanguage': 'cn', 'TaskKey': 'task20250820145400', 'FileUrl': 'https://ccschool.edusoa.com/cloud_file/project/ccyx-0012/material/26/7c/267c936afcf3e965d463f65c7275d60b.mp4'}, 'Parameters': {'Transcription': {'DiarizationEnabled': True, 'Diarization': {'SpeakerCount': 2}}, 'TranslationEnabled': True, 'Translation': {'TargetLanguages': ['en']}, 'AutoChaptersEnabled': True, 'MeetingAssistanceEnabled': True, 'MeetingAssistance': {'Types': ['Actions', 'KeyInformation']}, 'SummarizationEnabled': True, 'Summarization': {'Types': ['Paragraph', 'Conversational', 'QuestionsAnswering', 'MindMap']}, 'PptExtractionEnabled': True, 'TextPolishEnabled': True, 'Model': 'qwq', 'LlmOutputLanguage': 'en'}}
response: 
{
    "Code": "0",
    "Data": {
        "TaskId": "93d5f86d16b74a63aa0f2ecb80297819",
        "TaskKey": "task20250820145400",
        "TaskStatus": "ONGOING"
    },
    "Message": "success",
    "RequestId": "27485F01-F17F-53AC-9530-BC1B886362EE"
}
"""