xunfei.py

# -*- coding: utf-8 -*-
import base64
import hashlib
import hmac
import json
import os
import time
import requests
import urllib

lfasr_host = 'https://raasr.xfyun.cn/v2/api'
# 请求的接口名
api_upload = '/upload'
api_get_result = '/getResult'


class RequestApi(object):
    def __init__(self, appid, secret_key, upload_file_path):
        self.appid = appid
        self.secret_key = secret_key
        self.upload_file_path = upload_file_path
        self.ts = str(int(time.time()))
        self.signa = self.get_signa()

    def get_signa(self):
        appid = self.appid
        secret_key = self.secret_key
        m2 = hashlib.md5()
        m2.update((appid + self.ts).encode('utf-8'))
        md5 = m2.hexdigest()
        md5 = bytes(md5, encoding='utf-8')
        # 以secret_key为key, 上面的md5为msg， 使用hashlib.sha1加密结果为signa
        signa = hmac.new(secret_key.encode('utf-8'), md5, hashlib.sha1).digest()
        signa = base64.b64encode(signa)
        signa = str(signa, 'utf-8')
        return signa


    def upload(self):
        print("上传部分：")
        upload_file_path = self.upload_file_path
        file_len = os.path.getsize(upload_file_path)
        file_name = os.path.basename(upload_file_path)

        param_dict = {}
        param_dict['appId'] = self.appid
        param_dict['signa'] = self.signa
        param_dict['ts'] = self.ts
        param_dict["fileSize"] = file_len
        param_dict["fileName"] = file_name
        param_dict["duration"] = "200"
        print("upload参数：", param_dict)
        data = open(upload_file_path, 'rb').read(file_len)

        response = requests.post(url =lfasr_host + api_upload+"?"+urllib.parse.urlencode(param_dict),
                                headers = {"Content-type":"application/json"},data=data)
        print("upload_url:",response.request.url)
        result = json.loads(response.text)
        print("upload resp:", result)
        return result


    def get_result(self):
        uploadresp = self.upload()
        orderId = uploadresp['content']['orderId']
        param_dict = {}
        param_dict['appId'] = self.appid
        param_dict['signa'] = self.signa
        param_dict['ts'] = self.ts
        param_dict['orderId'] = orderId
        param_dict['resultType'] = "transfer,predict"
        print("")
        print("查询部分：")
        print("get result参数：", param_dict)
        status = 3
        # 建议使用回调的方式查询结果，查询接口有请求频率限制
        while status == 3:
            response = requests.post(url=lfasr_host + api_get_result + "?" + urllib.parse.urlencode(param_dict),
                                     headers={"Content-type": "application/json"})
            # print("get_result_url:",response.request.url)
            result = json.loads(response.text)
            print(result)
            status = result['content']['orderInfo']['status']
            print("status=",status)
            if status == 4:
                break
            time.sleep(5)
        print("get_result resp:",result)
        return result




# 输入讯飞开放平台的appid，secret_key和待转写的文件路径
def doRequest(folder, filename):
    api = RequestApi(appid="1a1c1793",
                     secret_key="***", # 请填写您自己的appid和KEY
                     upload_file_path=rf"audio/slice/{folder}/{filename}")

    res = api.get_result()
    print(res)
    return res

import json
def extract_and_format_transcription_from_string(json_string):
    """
    This function takes a JSON string as input, parses it, 
    extracts the transcription result, and formats it into a paragraph.

    :param json_string: JSON string containing response data
    :return: A string representing a formatted paragraph
    """
    # Parse the JSON string
    json_data = json.loads(json_string)

    # Extracting transcription result
    order_result_str = json_data.get("content", {}).get("orderResult", "{}")
    order_result = json.loads(order_result_str)

    # Parsing the transcription result
    sentences = []
    for lattice in order_result.get("lattice", []):
        json_1best_str = lattice.get("json_1best", "{}")
        json_1best = json.loads(json_1best_str)
        
        # Extract and process each word
        for rt in json_1best.get("st", {}).get("rt", []):
            sentence = ''.join([cw[0]["w"] for ws in rt["ws"] for cw in ws["cw"]])
            sentences.append(sentence)

    # Joining sentences to form a paragraph
    paragraph = ' '.join(sentences)
    return paragraph