tingwu

backend/ai_services/services.py

@@ -0,0 +1,115 @@
+import json
+import logging
+import time
+import oss2
+from aliyunsdkcore.client import AcsClient
+from aliyunsdkcore.acs_exception.exceptions import ClientException, ServerException
+# 尝试导入最新的 API 版本，如果有问题可能需要调整
+try:
+    from aliyunsdktingwu.request.v20230930 import CreateTaskRequest, GetTaskInfoRequest
+except ImportError:
+    # Fallback or error handling if version differs
+    pass
+    
+from django.conf import settings
+
+logger = logging.getLogger(__name__)
+
+class AliyunTingwuService:
+    def __init__(self):
+        self.access_key_id = settings.ALIYUN_ACCESS_KEY_ID
+        self.access_key_secret = settings.ALIYUN_ACCESS_KEY_SECRET
+        self.oss_bucket_name = settings.ALIYUN_OSS_BUCKET_NAME
+        self.oss_endpoint = settings.ALIYUN_OSS_ENDPOINT
+        self.tingwu_app_key = settings.ALIYUN_TINGWU_APP_KEY
+        self.region_id = "cn-beijing" # 听悟服务主要在北京
+
+        # 初始化 OSS Bucket
+        if self.access_key_id and self.access_key_secret and self.oss_endpoint:
+             auth = oss2.Auth(self.access_key_id, self.access_key_secret)
+             self.bucket = oss2.Bucket(auth, self.oss_endpoint, self.oss_bucket_name)
+        else:
+            self.bucket = None
+            logger.warning("Aliyun OSS configuration missing.")
+        
+        # 初始化听悟 Client
+        if self.access_key_id and self.access_key_secret:
+            self.client = AcsClient(
+                self.access_key_id,
+                self.access_key_secret,
+                self.region_id
+            )
+        else:
+            self.client = None
+            logger.warning("Aliyun AccessKey configuration missing.")
+
+    def upload_to_oss(self, file_obj, file_name):
+        """
+        上传文件到 OSS 并返回带签名的 URL (有效期 3 小时)
+        """
+        if not self.bucket:
+            raise Exception("OSS Client not initialized")
+            
+        try:
+            # 上传文件
+            # file_obj 应该是打开的文件对象或字节流
+            self.bucket.put_object(file_name, file_obj)
+            
+            # 生成签名 URL，有效期 3 小时 (3600 * 3)
+            url = self.bucket.sign_url('GET', file_name, 3600 * 3)
+            return url
+        except Exception as e:
+            logger.error(f"OSS Upload failed: {e}")
+            raise e
+
+    def create_transcription_task(self, file_url, language="cn"):
+        """
+        创建听悟转写任务
+        """
+        if not self.client:
+             raise Exception("Tingwu Client not initialized")
+
+        request = CreateTaskRequest.CreateTaskRequest()
+        request.set_AppKey(self.tingwu_app_key)
+        
+        # 配置 Input
+        input_param = {
+            "FileUrl": file_url,
+            "SourceLanguage": language,
+            "TaskKey": "transcription_task" 
+        }
+        request.set_Input(json.dumps(input_param))
+        
+        # 配置 Parameters (开启自动章节和摘要)
+        parameters = {
+            "Transcoding": {
+                "TargetAudioFormat": "mp3"
+            },
+            "AutoChaptersEnabled": True,
+            "SummarizationEnabled": True 
+        }
+        request.set_Parameters(json.dumps(parameters))
+
+        try:
+            response = self.client.do_action_with_exception(request)
+            return json.loads(response)
+        except (ClientException, ServerException) as e:
+            logger.error(f"Tingwu CreateTask failed: {e}")
+            raise e
+
+    def get_task_info(self, task_id):
+        """
+        查询任务状态和结果
+        """
+        if not self.client:
+             raise Exception("Tingwu Client not initialized")
+
+        request = GetTaskInfoRequest.GetTaskInfoRequest()
+        request.set_TaskId(task_id)
+        
+        try:
+            response = self.client.do_action_with_exception(request)
+            return json.loads(response)
+        except (ClientException, ServerException) as e:
+            logger.error(f"Tingwu GetTaskInfo failed: {e}")
+            raise e