douyin/utils/music_analysis.py

import json
from http import HTTPStatus
from dashscope.audio.asr import Transcription
import dashscope
import requests
import os
from openai import OpenAI

dashscope.api_key = os.getenv("DASHSCOPE_API_KEY", "*****")
content = """
将文本的链接提取出来，只留链接的有效信息
# 输出
只输出链接不需要分析过程
"""


def music_analysis(music_url):
    transcribe_response = Transcription.async_call(
        model='paraformer-v2',
        file_urls=[music_url],
        language_hints=['zh', 'en']  # “language_hints”只支持paraformer-v2模型
    )

    while True:
        if transcribe_response.output.task_status == 'SUCCEEDED' or transcribe_response.output.task_status == 'FAILED':
            break
        transcribe_response = Transcription.fetch(task=transcribe_response.output.task_id)

    if transcribe_response.status_code == HTTPStatus.OK:
        url = transcribe_response.output['results'][0]['transcription_url']
        print(url)

        # 发送GET请求
        response = requests.get(url)
        text = ''

        # 验证响应状态
        if response.status_code == 200:
            # 解析JSON数据
            data = response.json()
            for transcripts in data['transcripts']:
                text += transcripts['text']

        else:
            print(f"请求失败，状态码：{response.status_code}")
    return text


def chat_analysis(video_url):
    text = ''
    client = OpenAI(
        # 若没有配置环境变量，请用百炼API Key将下行替换为：api_key="sk-xxx",
        api_key=os.getenv("DASHSCOPE_API_KEY", "*****"),
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    )
    completion = client.chat.completions.create(
        model="qwen-plus",
        # 此处以qwen-plus为例，可按需更换模型名称。模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
        messages=[
            {
                'role': 'system',
                'content': content
            },
            {'role': 'user', 'content': video_url}],
    )

    print(f'语言模型返回数据: {completion.model_dump()}')
    for choices in completion.model_dump()['choices']:
        text += choices['message']['content']
    return text

def video_url(video_url):
    print(video_url)
if __name__ == '__main__':
    # music_analysis('https://lf26-music-east.douyinstatic.com/obj/ies-music-hj/7494207652008839996.mp3')
    chat_analysis(
        '原始视频url为：9.79 复制打开抖音，看看【学钓鱼的佳琪的作品】少与人纠缠，多跟鱼拉扯 # dou是钓鱼人 # 爱... https://v.douyin.com/1skvRRYPEgA/ DHI:/ n@D.uf 05/06')