AIVideoGenerationService.php 82 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980
  1. <?php
  2. namespace App\Services\AIGeneration;
  3. use App\Libs\Utils;
  4. use App\Models\MpGenerateVideoTask;
  5. use App\Models\AIElement;
  6. use App\Services\VolcEngineService;
  7. use GuzzleHttp\Client;
  8. use Illuminate\Support\Facades\DB;
  9. class AIVideoGenerationService
  10. {
  11. private $volcEngineService;
  12. public function __construct(VolcEngineService $volcEngineService)
  13. {
  14. $this->volcEngineService = $volcEngineService;
  15. }
  16. /**
  17. * 异步创建视频生成任务
  18. *
  19. * @param array $params
  20. * @return MpGenerateVideoTask
  21. */
  22. public function createJimengTask(array $params): MpGenerateVideoTask
  23. {
  24. // 生成唯一的任务ID
  25. $taskId = 'vid_gen_' . time() . '_' . uniqid();
  26. // 创建任务记录
  27. $task = MpGenerateVideoTask::create([
  28. 'task_id' => $taskId,
  29. 'prompt' => $params['prompt'] ?? '',
  30. 'first_frame_url' => $params['first_frame_url'] ?? null,
  31. 'tail_frame_url' => $params['tail_frame_url'] ?? null,
  32. 'seed' => $params['seed'] ?? -1,
  33. 'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : 121,
  34. 'video_resolution' => $params['video_resolution'] ?? '720P',
  35. 'status' => MpGenerateVideoTask::STATUS_PENDING,
  36. 'api_type' => 'jimeng',
  37. ]);
  38. // 发送异步请求到即梦AI API创建任务
  39. $this->submitTaskToApi($task);
  40. return $task;
  41. }
  42. /**
  43. * 提交任务到即梦AI API
  44. *
  45. * @param MpGenerateVideoTask $task
  46. * @return void
  47. */
  48. private function submitTaskToApi(MpGenerateVideoTask $task): void
  49. {
  50. try {
  51. // 验证环境变量配置
  52. $accessKey = env('VOLC_AI_AK');
  53. $secretKey = env('VOLC_AI_SK');
  54. if (empty($accessKey) || empty($secretKey)) {
  55. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  56. 'error_message' => '火山引擎访问密钥未配置,请检查环境变量AK和SK'
  57. ]);
  58. return;
  59. }
  60. // 根据任务类型选择对应的req_key
  61. if ($task->first_frame_url && $task->tail_frame_url) {
  62. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_tail_v30' : 'jimeng_i2v_first_tail_v30_1080';
  63. }else if($task->first_frame_url){
  64. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_v30' : 'jimeng_i2v_first_v30_1080';
  65. }else{
  66. $req_key = $task->video_resolution == '720P' ? 'jimeng_t2v_v30' : 'jimeng_t2v_v30_1080p';
  67. }
  68. // 构建即梦AI 3.0 视频生成API请求参数
  69. $apiParams = [
  70. 'req_key' => $req_key,
  71. 'prompt' => $task->prompt,
  72. 'seed' => (int)$task->seed,
  73. 'frames' => $task->frames,
  74. ];
  75. // 添加图片URL参数
  76. if ($task->first_frame_url) {
  77. $apiParams['image_urls'] = [
  78. $task->first_frame_url
  79. ];
  80. if ($task->tail_frame_url) $apiParams['image_urls'][] = $task->tail_frame_url;
  81. }
  82. dLog('generate')->info('图生视频参数: ', $apiParams);
  83. logDB('generate', 'info', '即梦AI视频生成任务提交', ['id' => $task->id, 'params' => $apiParams]);
  84. // 调用即梦AI 3.0 视频生成API提交任务
  85. $response = $this->volcEngineService->request(
  86. 'POST',
  87. 'visual.volcengineapi.com', // 即梦AI API域名
  88. '/', // 使用根路径
  89. [], // Query参数
  90. json_encode($apiParams),
  91. $accessKey,
  92. $secretKey,
  93. 'cv', // 服务标识
  94. 'cn-north-1', // 区域
  95. 'CVSync2AsyncSubmitTask',
  96. '2022-08-31',
  97. 'application/json'
  98. );
  99. $responseData = json_decode($response['body'], true);
  100. dLog('generate')->info('即梦AI 3.0 视频生成API提交任务响应: ', $responseData);
  101. logDB('generate', 'info', '即梦AI视频生成API响应', ['id' => $task->id, 'response' => $responseData]);
  102. if ($responseData['code'] !== 10000) {
  103. // API返回错误
  104. logDB('generate', 'error', '即梦AI视频生成任务提交失败', ['id' => $task->id, 'error' => $responseData['message'] ?? 'API Error']);
  105. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  106. 'error_message' => $responseData['message'] ?? 'API Error',
  107. 'extra_params' => $responseData['data'] ?? []
  108. ]);
  109. } else {
  110. // 任务提交成功,更新状态为处理中
  111. $updateData = [];
  112. // 存储API返回的任务ID
  113. if (isset($responseData['data']['task_id'])) {
  114. $updateData['task_id'] = $responseData['data']['task_id'];
  115. $updateData['extra_params'] = $responseData['data'] ?? [];
  116. $updateData['extra_params']['req_key'] = $req_key;
  117. $updateData['extra_params']['request_id'] = isset($responseData['request_id']) ? $responseData['request_id'] : '';
  118. }
  119. $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
  120. }
  121. } catch (\Exception $e) {
  122. // 记录错误
  123. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  124. 'error_message' => 'API请求失败: ' . $e->getMessage()
  125. ]);
  126. Utils::throwError('1001:'.$e->getMessage());
  127. }
  128. }
  129. /**
  130. * 查询任务状态(即梦AI)
  131. *
  132. * @param MpGenerateVideoTask $task
  133. * @return array
  134. */
  135. public function queryJimengTaskStatus(MpGenerateVideoTask $task): array
  136. {
  137. try {
  138. // 获取API任务ID
  139. $apiTaskId = $task->task_id ?? null;
  140. if (!$apiTaskId) {
  141. return [
  142. 'status' => 'failed',
  143. 'error_message' => 'API任务ID不存在',
  144. 'result_url' => null
  145. ];
  146. }
  147. // 根据任务类型选择对应的req_key
  148. if ($task->first_frame_url && $task->tail_frame_url) {
  149. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_tail_v30' : 'jimeng_i2v_first_tail_v30_1080';
  150. }else if($task->first_frame_url){
  151. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_v30' : 'jimeng_i2v_first_v30_1080';
  152. }else{
  153. $req_key = $task->video_resolution == '720P' ? 'jimeng_t2v_v30' : 'jimeng_t2v_v30_1080p';
  154. }
  155. // 构建查询参数
  156. $apiParams = [
  157. 'req_key' => $req_key,
  158. 'task_id' => $apiTaskId,
  159. // 'req_json' => '{"return_url":true}' // 返回URL格式
  160. ];
  161. // 调用即梦AI 3.0 视频生成API查询任务状态
  162. $response = $this->volcEngineService->request(
  163. 'POST',
  164. 'visual.volcengineapi.com',
  165. '/', // 使用根路径
  166. [], // Query参数
  167. json_encode($apiParams),
  168. env('VOLC_AI_AK'),
  169. env('VOLC_AI_SK'),
  170. 'cv',
  171. 'cn-north-1',
  172. 'CVSync2AsyncGetResult',
  173. '2022-08-31',
  174. 'application/json'
  175. );
  176. $responseData = json_decode($response['body'], true);
  177. dLog('generate')->info('即梦AI 3.0 视频生成API查询任务状态响应: ', $responseData);
  178. logDB('generate', 'info', '即梦AI视频任务状态查询', ['id' => $task->id, 'response' => $responseData]);
  179. if ($responseData['code'] !== 10000) {
  180. // API返回错误
  181. logDB('generate', 'error', '即梦AI视频任务状态查询失败', ['id' => $task->id, 'error' => $responseData['message'] ?? 'API Error']);
  182. return [
  183. 'status' => 'failed',
  184. 'error_message' => $responseData['message'] ?? 'API Error',
  185. 'result_url' => null,
  186. 'result_json' => $responseData ?? []
  187. ];
  188. }
  189. // 解析API响应
  190. $result = $responseData['data'] ?? [];
  191. $taskStatus = $result['status'] ?? 'failed';
  192. $returnData = [
  193. 'status' => $taskStatus,
  194. 'error_message' => null,
  195. 'result_url' => null,
  196. 'result_json' => $responseData ?? []
  197. ];
  198. // 如果任务成功,获取结果URL
  199. if ($taskStatus === 'done' && isset($result['video_url'])) {
  200. if (isset($result['video_url'])) {
  201. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  202. $video_ext = getVideoExtFromUrl($result['video_url']);
  203. $video_name = $video_name . $video_ext;
  204. // 将视频另存到tos
  205. $url = uploadStreamByTos('video', file_get_contents($result['video_url']), $video_name);
  206. $returnData['result_url'] = $url;
  207. }
  208. // 计算视频时长
  209. $video_duration = 0;
  210. if (isset($result['duration'])) {
  211. $video_duration = (int)$result['duration'];
  212. } elseif (isset($result['frames']) && isset($result['framespersecond'])) {
  213. $frames = (int)$result['frames'];
  214. $fps = (int)$result['framespersecond'];
  215. if ($fps > 0) {
  216. $video_duration = (int)floor($frames / $fps);
  217. }
  218. }
  219. // 如果没有计算出时长,使用任务表中的设定值
  220. if ($video_duration <= 0 && !empty($task->extra_params)) {
  221. $extraParams = is_array($task->extra_params) ? $task->extra_params : json_decode($task->extra_params, true);
  222. if (isset($extraParams['duration'])) {
  223. $video_duration = (int)$extraParams['duration'];
  224. }
  225. }
  226. // 只有当video_duration > 0时才添加到返回数据中
  227. if ($video_duration > 0) {
  228. $returnData['video_duration'] = $video_duration;
  229. $returnData['video_time_point_start'] = 0;
  230. $returnData['video_time_point_end'] = $video_duration;
  231. }
  232. $returnData['status'] = 'success';
  233. } elseif (in_array($taskStatus, ['not_found', 'expired'])) {
  234. $returnData['status'] = 'failed';
  235. $returnData['error_message'] = '任务未找到或已过期';
  236. } elseif ($taskStatus === 'failed') {
  237. $returnData['status'] = 'failed';
  238. $returnData['error_message'] = '任务执行失败';
  239. }
  240. // in_queue 或 generating状态保持原status
  241. return $returnData;
  242. } catch (\Exception $e) {
  243. return [
  244. 'status' => 'failed',
  245. 'error_message' => $e->getMessage(),
  246. 'result_url' => null
  247. ];
  248. }
  249. }
  250. /**
  251. * 更新所有待处理任务的状态
  252. *
  253. * @return void
  254. */
  255. public function updatePendingTasks(): void
  256. {
  257. // 获取所有处理中的任务
  258. $tasks = MpGenerateVideoTask::where('status', MpGenerateVideoTask::STATUS_PROCESSING)
  259. ->get();
  260. foreach ($tasks as $task) {
  261. $api_type = getProp($task, 'api_type');
  262. switch($api_type) {
  263. case 'seedance':
  264. $this->updateSeedanceTask($task);
  265. break;
  266. case 'jimeng':
  267. $this->updateJimengTask($task);
  268. break;
  269. case 'keling':
  270. $this->updateKelingOmniTask($task);
  271. break;
  272. default:
  273. break;
  274. }
  275. }
  276. }
  277. // 更新即梦AI视频任务状态
  278. private function updateJimengTask($task) {
  279. $statusInfo = $this->queryJimengTaskStatus($task);
  280. if (!isset($statusInfo['status'])) return;
  281. if ($statusInfo['status'] === 'success') {
  282. logDB('generate', 'info', '即梦AI视频生成任务成功', ['id' => $task->id, 'result_url' => $statusInfo['result_url']]);
  283. $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
  284. // 同步调整分镜视频状态和结果
  285. $segment_id = getProp($task, 'alias_segment_id');
  286. if ($segment_id && isset($statusInfo['result_url'])) {
  287. try {
  288. DB::beginTransaction();
  289. $now = date('Y-m-d H:i:s');
  290. // 构建更新数据
  291. $segmentUpdateData = [
  292. 'video_url' => $statusInfo['result_url'],
  293. 'video_task_status' => '已完成',
  294. 'current_type' => 2,
  295. 'last_frame_url' => $statusInfo['last_frame_url'] ?? '',
  296. 'updated_at' => $now
  297. ];
  298. // 只有当video_duration存在且大于0时才更新
  299. if (isset($statusInfo['video_duration']) && $statusInfo['video_duration'] > 0) {
  300. $segmentUpdateData['video_duration'] = $statusInfo['video_duration'];
  301. $segmentUpdateData['video_time_point_start'] = 0;
  302. // $segmentUpdateData['video_time_point_end'] = $statusInfo['video_duration'];
  303. $audio_duration = DB::table('mp_episode_segments')->where('segment_id', $segment_id)->value('audio_duration');
  304. $segmentUpdateData['video_time_point_end'] = $audio_duration ?? 0;
  305. }
  306. // 更新分镜表
  307. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update($segmentUpdateData);
  308. // 添加视频生成对话记录
  309. $this->addVideoGenerationRecords($segment_id, $statusInfo['result_url']);
  310. DB::commit();
  311. } catch (\Exception $e) {
  312. DB::rollBack();
  313. dLog('generate')->error('即梦视频任务处理失败', [
  314. 'segment_id' => $segment_id,
  315. 'task_id' => $task->id,
  316. 'error' => $e->getMessage()
  317. ]);
  318. }
  319. }
  320. } elseif ($statusInfo['status'] === 'failed') {
  321. logDB('generate', 'error', '即梦AI视频生成任务失败', ['id' => $task->id, 'error' => $statusInfo['error_message']]);
  322. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, $statusInfo);
  323. // 同步更新分镜表状态为失败
  324. $segment_id = getProp($task, 'alias_segment_id');
  325. if ($segment_id) {
  326. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  327. 'video_task_status' => '失败',
  328. 'updated_at' => date('Y-m-d H:i:s')
  329. ]);
  330. }
  331. }
  332. // 如果仍然是处理中状态,不做任何操作
  333. // 处理:如果任务处理超过12小时,标记为失败
  334. $processingTime = now()->diffInHours($task->created_at);
  335. if ($processingTime > 12) {
  336. logDB('generate', 'warning', '即梦AI视频生成任务超时', ['id' => $task->id, 'processing_hours' => $processingTime]);
  337. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  338. 'error_message' => '任务处理超时(超过12小时)'
  339. ]);
  340. // 同步更新分镜表状态为失败
  341. $segment_id = getProp($task, 'alias_segment_id');
  342. if ($segment_id) {
  343. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  344. 'video_task_status' => '失败',
  345. 'updated_at' => date('Y-m-d H:i:s')
  346. ]);
  347. }
  348. }
  349. }
  350. public function getTaskList($data)
  351. {
  352. $task_id = getProp($data, 'task_id');
  353. $status = getProp($data, 'status');
  354. $query = MpGenerateVideoTask::select('*');
  355. if ($task_id) {
  356. $query->where('id', $task_id);
  357. }
  358. if ($status) {
  359. $query->where('status', $status);
  360. }
  361. return $query->orderBy('created_at', 'desc')->paginate();
  362. }
  363. /**
  364. * 创建 Seedance 1.5 Pro 视频生成任务
  365. *
  366. * @param array $params
  367. * @return MpGenerateVideoTask
  368. */
  369. public function createSeedanceTask(array $params): MpGenerateVideoTask
  370. {
  371. // 生成唯一的任务ID
  372. $taskId = 'seedance_' . time() . '_' . uniqid();
  373. // 创建任务记录,复用现有表结构
  374. $task = MpGenerateVideoTask::create([
  375. 'task_id' => $taskId,
  376. 'alias_segment_id' => $params['alias_segment_id'] ?? '',
  377. 'prompt' => $params['prompt'] ?? '',
  378. 'first_frame_url' => $params['first_frame_url'] ?? '',
  379. 'tail_frame_url' => $params['tail_frame_url'] ?? '',
  380. 'seed' => $params['seed'] ?? -1,
  381. 'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : null,
  382. 'video_resolution' => $params['video_resolution'] ?? '720P',
  383. 'status' => MpGenerateVideoTask::STATUS_PENDING,
  384. 'api_type' => 'seedance', // 标识为 Seedance 1.5 Pro API
  385. 'extra_params' => [
  386. 'model' => $params['model'] ?? 'doubao-seedance-1-5-pro-251215',
  387. 'content' => $params['content'],
  388. 'callback_url' => $params['callback_url'] ?? '',
  389. 'return_last_frame' => $params['return_last_frame'] ?? true,
  390. 'service_tier' => $params['service_tier'] ?? 'default',
  391. 'execution_expires_after' => $params['execution_expires_after'] ?? 172800,
  392. 'generate_audio' => $params['generate_audio'] ?? false,
  393. 'draft' => $params['draft'] ?? false,
  394. 'ratio' => $params['ratio'] ?? '9:16',
  395. 'duration' => $params['video_duration'] ?? 5,
  396. 'camera_fixed' => $params['camera_fixed'] ?? false,
  397. 'watermark' => $params['watermark'] ?? false,
  398. ]
  399. ]);
  400. // 提交任务到火山引擎 API
  401. $this->submitSeedanceTaskToApi($task);
  402. return $task;
  403. }
  404. /**
  405. * 提交 Seedance 任务到火山引擎 API
  406. *
  407. * @param MpGenerateVideoTask $task
  408. * @return void
  409. */
  410. private function submitSeedanceTaskToApi(MpGenerateVideoTask $task): void
  411. {
  412. $client = new Client(['verify'=>false,'timeout'=>120]);
  413. try {
  414. // 验证环境变量配置
  415. $apiKey = env('VOLC_AI_API_KEY');
  416. if (empty($apiKey)) {
  417. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  418. 'error_message' => '火山引擎访问密钥未配置,请检查环境变量API Key'
  419. ]);
  420. return;
  421. }
  422. $extraParams = $task->extra_params;
  423. // 构建火山引擎 Seedance API 请求参数
  424. $apiParams = [
  425. 'model' => isset(\App\Consts\BaseConst::MODELS_MAP[$extraParams['model']]) ? \App\Consts\BaseConst::MODELS_MAP[$extraParams['model']] : $extraParams['model'],
  426. 'content' => $extraParams['content'],
  427. ];
  428. // 添加可选参数
  429. if (!empty($extraParams['callback_url'])) {
  430. $apiParams['callback_url'] = $extraParams['callback_url'];
  431. }
  432. if (isset($extraParams['return_last_frame'])) {
  433. $apiParams['return_last_frame'] = $extraParams['return_last_frame'];
  434. }
  435. if (!empty($extraParams['service_tier']) && !in_array($extraParams['model'], ['doubao-seedance-2-0-260128', 'doubao-seedance-2-0-fast-260128'])) {
  436. $apiParams['service_tier'] = $extraParams['service_tier'];
  437. }
  438. if (!empty($extraParams['execution_expires_after'])) {
  439. $apiParams['execution_expires_after'] = $extraParams['execution_expires_after'];
  440. }
  441. if (isset($extraParams['generate_audio'])) {
  442. $apiParams['generate_audio'] = $extraParams['generate_audio'];
  443. }
  444. if (isset($extraParams['draft'])) {
  445. $apiParams['draft'] = $extraParams['draft'];
  446. }
  447. if (!empty($extraParams['video_resolution'])) {
  448. $apiParams['resolution'] = strtolower($task->video_resolution);
  449. }
  450. if (!empty($extraParams['ratio'])) {
  451. $apiParams['ratio'] = $extraParams['ratio'];
  452. }
  453. if (!empty($extraParams['duration'])) {
  454. $apiParams['duration'] = (int)$extraParams['duration'];
  455. }
  456. if (isset($task->frames)) {
  457. // $apiParams['frames'] = $task->frames;
  458. }
  459. if ($task->seed !== -1) {
  460. $apiParams['seed'] = (int)$task->seed;
  461. }
  462. if (isset($extraParams['camera_fixed'])) {
  463. $apiParams['camera_fixed'] = $extraParams['camera_fixed'];
  464. }
  465. if (isset($extraParams['watermark'])) {
  466. $apiParams['watermark'] = $extraParams['watermark'];
  467. }
  468. dLog('generate')->info('Seedance 1.5 Pro 视频生成参数: ', $apiParams);
  469. logDB('generate', 'info', 'Seedance视频生成任务提交', ['id' => $task->id, 'params' => $apiParams]);
  470. // 调用火山引擎 Seedance API
  471. $response = $client->post('https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks', [
  472. 'headers' => [
  473. 'Authorization' => 'Bearer ' . $apiKey,
  474. 'Content-Type' => 'application/json',
  475. ],
  476. 'json' => $apiParams
  477. ]);
  478. $responseData = json_decode($response->getBody(), true);
  479. dLog('generate')->info('Seedance 1.5 Pro API 响应: ', $responseData);
  480. logDB('generate', 'info', 'Seedance视频生成API响应', ['id' => $task->id, 'response' => $responseData]);
  481. if (isset($responseData['error'])) {
  482. // API返回错误
  483. logDB('generate', 'error', 'Seedance视频生成任务提交失败', ['id' => $task->id, 'error' => $responseData['error']['message'] ?? 'API Error']);
  484. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  485. 'error_message' => $responseData['error']['message'] ?? 'API Error',
  486. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  487. ]);
  488. } elseif (isset($responseData['id'])) {
  489. // 任务提交成功,更新状态为处理中
  490. $updateData = [
  491. 'task_id' => $responseData['id'], // 使用 API 返回的任务 ID
  492. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  493. ];
  494. $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
  495. } else {
  496. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  497. 'error_message' => '未知的 API 响应格式',
  498. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  499. ]);
  500. }
  501. } catch (\Exception $e) {
  502. // 记录错误
  503. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  504. 'error_message' => 'API请求失败: ' . $e->getMessage()
  505. ]);
  506. Utils::throwError('1001:'.$e->getMessage());
  507. }
  508. }
  509. // 更新Seedance视频任务状态
  510. private function updateSeedanceTask($task) {
  511. $statusInfo = $this->querySeedanceTaskStatus($task);
  512. if (!isset($statusInfo['status'])) return;
  513. if ($statusInfo['status'] === 'success') {
  514. logDB('generate', 'info', 'Seedance视频生成任务成功', ['id' => $task->id, 'result_url' => $statusInfo['result_url']]);
  515. $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
  516. // 同步调整分镜视频状态和结果
  517. $segment_id = getProp($task, 'alias_segment_id');
  518. if ($segment_id && isset($statusInfo['result_url'])) {
  519. try {
  520. DB::beginTransaction();
  521. $now = date('Y-m-d H:i:s');
  522. // 构建更新数据
  523. $segmentUpdateData = [
  524. 'video_url' => $statusInfo['result_url'],
  525. 'video_task_status' => '已完成',
  526. 'current_type' => 2,
  527. 'last_frame_url' => $statusInfo['last_frame_url'] ?? '',
  528. 'updated_at' => $now
  529. ];
  530. // 只有当video_duration存在且大于0时才更新
  531. if (isset($statusInfo['video_duration']) && $statusInfo['video_duration'] > 0) {
  532. $segmentUpdateData['video_duration'] = $statusInfo['video_duration'];
  533. $segmentUpdateData['video_time_point_start'] = 0;
  534. // $segmentUpdateData['video_time_point_end'] = $statusInfo['video_duration'];
  535. $audio_duration = DB::table('mp_episode_segments')->where('segment_id', $segment_id)->value('audio_duration');
  536. $segmentUpdateData['video_time_point_end'] = $audio_duration ?? 0;
  537. }
  538. // 更新分镜表
  539. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update($segmentUpdateData);
  540. // 添加视频生成对话记录
  541. $this->addVideoGenerationRecords($segment_id, $statusInfo['result_url']);
  542. DB::commit();
  543. } catch (\Exception $e) {
  544. DB::rollBack();
  545. dLog('generate')->error('Seedance视频任务处理失败', [
  546. 'segment_id' => $segment_id,
  547. 'task_id' => $task->id,
  548. 'error' => $e->getMessage()
  549. ]);
  550. }
  551. }
  552. } elseif ($statusInfo['status'] === 'failed') {
  553. logDB('generate', 'error', 'Seedance视频生成任务失败', ['id' => $task->id, 'error' => $statusInfo['error_message']]);
  554. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  555. 'error_message' => $statusInfo['error_message'],
  556. 'result_json' => $statusInfo['result_json'] ?? []
  557. ]);
  558. // 同步更新分镜表状态为失败
  559. $segment_id = getProp($task, 'alias_segment_id');
  560. if ($segment_id) {
  561. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  562. 'video_task_status' => '失败',
  563. 'updated_at' => date('Y-m-d H:i:s')
  564. ]);
  565. }
  566. }
  567. // 如果仍然是处理中状态,不做任何操作
  568. // 处理:如果任务处理超过48小时,标记为失败
  569. $processingTime = now()->diffInHours($task->created_at);
  570. if ($processingTime > 48) {
  571. logDB('generate', 'warning', 'Seedance视频生成任务超时', ['id' => $task->id, 'processing_hours' => $processingTime]);
  572. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  573. 'error_message' => '任务处理超时(超过48小时)'
  574. ]);
  575. // 同步更新分镜表状态为失败
  576. $segment_id = getProp($task, 'alias_segment_id');
  577. if ($segment_id) {
  578. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  579. 'video_task_status' => '失败',
  580. 'updated_at' => date('Y-m-d H:i:s')
  581. ]);
  582. }
  583. }
  584. }
  585. /**
  586. * 查询 Seedance 任务状态
  587. *
  588. * @param MpGenerateVideoTask $task
  589. * @return array
  590. */
  591. public function querySeedanceTaskStatus(MpGenerateVideoTask $task): array
  592. {
  593. $client = new Client(['verify'=>false,'timeout'=>120]);
  594. try {
  595. // 如果任务已完成,直接返回结果
  596. if (in_array($task->status, [MpGenerateVideoTask::STATUS_SUCCESS, MpGenerateVideoTask::STATUS_FAILED])) {
  597. return [
  598. 'success' => true,
  599. 'data' => [
  600. 'id' => $task->id,
  601. 'status' => $this->mapTaskStatus($task->status),
  602. 'result_url' => $task->result_url,
  603. 'error_message' => $task->error_message,
  604. 'created_at' => $task->created_at,
  605. 'started_at' => $task->started_at,
  606. 'completed_at' => $task->completed_at
  607. ]
  608. ];
  609. }
  610. // 调用火山引擎 API 查询任务状态
  611. $apiKey = env('VOLC_AI_API_KEY');
  612. if (empty($apiKey)) {
  613. return [
  614. 'status' => MpGenerateVideoTask::STATUS_FAILED,
  615. 'result_json' => [],
  616. 'error_message' => '火山引擎访问密钥未配置'
  617. ];
  618. }
  619. // 构建查询 URL
  620. $url = 'https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks/' . $task->task_id;
  621. $response = $client->get($url, [
  622. 'headers' => [
  623. 'Authorization' => 'Bearer ' . $apiKey,
  624. 'Content-Type' => 'application/json',
  625. ]
  626. ]);
  627. $responseData = json_decode($response->getBody(), true);
  628. dLog('generate')->info('Seedance 任务状态查询响应: ', $responseData);
  629. logDB('generate', 'info', 'Seedance任务状态查询', ['id' => $task->id, 'response' => $responseData]);
  630. if (isset($responseData['error'])) {
  631. logDB('generate', 'error', 'Seedance任务状态查询失败', ['id' => $task->id, 'error' => $responseData['error']['message'] ?? 'API查询失败']);
  632. return [
  633. 'status' => MpGenerateVideoTask::STATUS_FAILED,
  634. 'result_json' => $responseData,
  635. 'error_message' => $responseData['error']['message'] ?? 'API查询失败'
  636. ];
  637. }
  638. // 解析任务状态
  639. $apiStatus = $responseData['status'] ?? 'unknown';
  640. $taskStatus = $this->mapApiStatusToTaskStatus($apiStatus);
  641. $updateData = ['result_json' => $responseData];
  642. if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
  643. $updateData['status'] = MpGenerateVideoTask::STATUS_SUCCESS;
  644. // 任务成功,处理结果
  645. if (isset($responseData['content']['video_url'])) {
  646. $video_url = $responseData['content']['video_url'];
  647. // 下载并保存视频
  648. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  649. $video_ext = getVideoExtFromUrl($video_url);
  650. $video_name = $video_name . $video_ext;
  651. $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
  652. $updateData['result_url'] = $url;
  653. }
  654. // 计算视频时长
  655. $video_duration = 0;
  656. if (isset($responseData['duration'])) {
  657. // 如果直接返回duration,使用该值
  658. $video_duration = (int)$responseData['duration'];
  659. } elseif (isset($responseData['frames']) && isset($responseData['framespersecond'])) {
  660. // 如果返回frames和framespersecond,计算时长
  661. $frames = (int)$responseData['frames'];
  662. $fps = (int)$responseData['framespersecond'];
  663. if ($fps > 0) {
  664. $video_duration = (int)floor($frames / $fps);
  665. }
  666. }
  667. // 如果没有计算出时长,使用任务表中的设定值
  668. if ($video_duration <= 0 && !empty($task->extra_params)) {
  669. $extraParams = is_array($task->extra_params) ? $task->extra_params : json_decode($task->extra_params, true);
  670. if (isset($extraParams['duration'])) {
  671. $video_duration = (int)$extraParams['duration'];
  672. }
  673. }
  674. // 只有当video_duration > 0时才添加到更新数据中
  675. if ($video_duration > 0) {
  676. $updateData['video_duration'] = $video_duration;
  677. $updateData['video_time_point_start'] = 0;
  678. $updateData['video_time_point_end'] = $video_duration;
  679. }
  680. // 处理尾帧图片(如果启用了 return_last_frame)
  681. if (isset($responseData['content']['last_frame_url']) && !empty($responseData['content']['last_frame_url'])) {
  682. $pic_url = $responseData['content']['last_frame_url'];
  683. $pic_name = 'ai_generation_' . time() . '_' . uniqid();
  684. $pic_ext = getImgExtFromUrl($pic_url);
  685. $pic_name = $pic_name . $pic_ext;
  686. if ($pic_ext === '.png') {
  687. $comporessed_data = compressRemoteImageUrlToSize($pic_url);
  688. $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
  689. }else {
  690. // 将图片另存到tos
  691. $url = uploadStreamByTos('image', file_get_contents($pic_url), $pic_name);
  692. }
  693. $updateData['last_frame_url'] = $url;
  694. }
  695. return $updateData;
  696. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
  697. $updateData['status'] = MpGenerateVideoTask::STATUS_FAILED;
  698. $updateData['error_message'] = $responseData['error_message'] ?? '任务执行失败';
  699. return $updateData;
  700. }
  701. // 其他状态保持不变
  702. return [
  703. 'success' => true,
  704. 'data' => [
  705. 'id' => $task->id,
  706. 'status' => $this->mapTaskStatus($task->status),
  707. 'result_url' => $task->result_url,
  708. 'error_message' => $task->error_message,
  709. 'created_at' => $task->created_at,
  710. 'started_at' => $task->started_at,
  711. 'completed_at' => $task->completed_at,
  712. 'api_response' => $responseData
  713. ]
  714. ];
  715. } catch (\Exception $e) {
  716. dLog('generate')->info('Seedance 任务状态查询异常: '.$e->getMessage());
  717. return [
  718. 'success' => false,
  719. 'message' => '查询失败: ' . $e->getMessage()
  720. ];
  721. }
  722. }
  723. /**
  724. * 映射 API 状态到任务状态
  725. *
  726. * @param string $apiStatus
  727. * @return string
  728. */
  729. private function mapApiStatusToTaskStatus(string $apiStatus): string
  730. {
  731. switch ($apiStatus) {
  732. case 'queued':
  733. case 'running':
  734. return MpGenerateVideoTask::STATUS_PROCESSING;
  735. case 'succeeded':
  736. return MpGenerateVideoTask::STATUS_SUCCESS;
  737. case 'failed':
  738. case 'expired':
  739. return MpGenerateVideoTask::STATUS_FAILED;
  740. default:
  741. return MpGenerateVideoTask::STATUS_PROCESSING;
  742. }
  743. }
  744. public function seedanceCallback($data) {
  745. try {
  746. dLog('generate')->info('Seedance 回调数据: ', $data);
  747. logDB('generate', 'info', 'Seedance回调接收', ['task_id' => $data['id'] ?? null, 'status' => $data['status'] ?? null]);
  748. // 从回调数据中获取任务ID
  749. $taskId = $data['id'] ?? null;
  750. if (!$taskId) {
  751. dLog('generate')->error('Seedance 回调缺少任务ID');
  752. logDB('generate', 'error', 'Seedance回调缺少任务ID', ['data' => $data]);
  753. return ['success' => false, 'message' => '缺少任务ID'];
  754. }
  755. // 查找对应的任务记录
  756. $task = MpGenerateVideoTask::where('task_id', $taskId)->first();
  757. if (!$task) {
  758. dLog('generate')->error('Seedance 回调找不到任务: ' . $taskId);
  759. logDB('generate', 'error', 'Seedance回调找不到任务', ['task_id' => $taskId]);
  760. return ['success' => false, 'message' => '任务不存在'];
  761. }
  762. // 解析任务状态
  763. $apiStatus = $data['status'] ?? 'unknown';
  764. $taskStatus = $this->mapApiStatusToTaskStatus($apiStatus);
  765. // 准备更新数据
  766. $updateData = [
  767. 'result_json' => $data,
  768. 'updated_at' => now()
  769. ];
  770. if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
  771. $updateData['status'] = MpGenerateVideoTask::STATUS_SUCCESS;
  772. $updateData['completed_at'] = now();
  773. // 处理视频结果
  774. if (isset($data['content']['video_url'])) {
  775. $video_url = $data['content']['video_url'];
  776. try {
  777. // 下载并保存视频
  778. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  779. $video_ext = getVideoExtFromUrl($video_url);
  780. $video_name = $video_name . $video_ext;
  781. $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
  782. $updateData['result_url'] = $url;
  783. dLog('generate')->info('Seedance 视频保存成功: ' . $url);
  784. } catch (\Exception $e) {
  785. dLog('generate')->error('Seedance 视频保存失败: ' . $e->getMessage());
  786. // 如果保存失败,直接使用原始URL
  787. $updateData['result_url'] = $video_url;
  788. }
  789. }
  790. // 处理尾帧图片(如果启用了 return_last_frame)
  791. if (isset($data['content']['last_frame_url']) && !empty($data['content']['last_frame_url'])) {
  792. $pic_url = $data['content']['last_frame_url'];
  793. try {
  794. $pic_name = 'ai_generation_' . time() . '_' . uniqid();
  795. $pic_ext = getImgExtFromUrl($pic_url);
  796. $pic_name = $pic_name . $pic_ext;
  797. if ($pic_ext === '.png') {
  798. $compressed_data = compressRemoteImageUrlToSize($pic_url);
  799. $url = uploadStreamByTos('image', $compressed_data, $pic_name);
  800. } else {
  801. $url = uploadStreamByTos('image', file_get_contents($pic_url), $pic_name);
  802. }
  803. $updateData['last_frame_url'] = $url;
  804. dLog('generate')->info('Seedance 尾帧图片保存成功: ' . $url);
  805. } catch (\Exception $e) {
  806. dLog('generate')->error('Seedance 尾帧图片保存失败: ' . $e->getMessage());
  807. // 如果保存失败,直接使用原始URL
  808. $updateData['last_frame_url'] = $pic_url;
  809. }
  810. }
  811. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
  812. $updateData['status'] = MpGenerateVideoTask::STATUS_FAILED;
  813. $updateData['completed_at'] = now();
  814. $updateData['error_message'] = $data['error_message'] ?? $data['failure_reason'] ?? '任务执行失败';
  815. } else {
  816. // 处理中状态
  817. $updateData['status'] = $taskStatus;
  818. if ($taskStatus === MpGenerateVideoTask::STATUS_PROCESSING && !$task->started_at) {
  819. $updateData['started_at'] = now();
  820. }
  821. }
  822. // 更新任务记录
  823. $task->update($updateData);
  824. // 更新分镜表的视频任务状态
  825. $this->updateSegmentVideoStatus($task, $taskStatus, $updateData);
  826. dLog('generate')->info('Seedance 回调处理完成: ' . $taskId . ', 状态: ' . $taskStatus);
  827. logDB('generate', 'info', 'Seedance回调处理完成', ['task_id' => $taskId, 'status' => $taskStatus]);
  828. return [
  829. 'success' => true,
  830. 'message' => '回调处理成功',
  831. 'task_id' => $taskId,
  832. 'status' => $taskStatus
  833. ];
  834. } catch (\Exception $e) {
  835. dLog('generate')->error('Seedance 回调处理异常: ' . $e->getMessage());
  836. logDB('generate', 'error', 'Seedance回调处理异常', ['error' => $e->getMessage()]);
  837. return [
  838. 'success' => false,
  839. 'message' => '回调处理失败: ' . $e->getMessage()
  840. ];
  841. }
  842. }
  843. /**
  844. * 更新分镜表的视频任务状态
  845. *
  846. * @param MpGenerateVideoTask $task
  847. * @param string $taskStatus
  848. * @param array $updateData
  849. */
  850. private function updateSegmentVideoStatus(MpGenerateVideoTask $task, string $taskStatus, array $updateData) {
  851. try {
  852. // 查找关联的分镜记录
  853. $segment = DB::table('mp_episode_segments')
  854. ->where('video_task_id', $task->id)
  855. ->first();
  856. if (!$segment) {
  857. dLog('generate')->info('未找到关联的分镜记录: ' . $task->id);
  858. return;
  859. }
  860. $segmentUpdateData = ['updated_at' => date('Y-m-d H:i:s')];
  861. if (isset($updateData['last_frame_url']) && $updateData['last_frame_url']) {
  862. $segmentUpdateData['last_frame_url'] = $updateData['last_frame_url'];
  863. }
  864. if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
  865. $segmentUpdateData['video_task_status'] = '已完成';
  866. if (isset($updateData['result_url'])) {
  867. $segmentUpdateData['video_url'] = $updateData['result_url'];
  868. }
  869. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
  870. $segmentUpdateData['video_task_status'] = '失败';
  871. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_PROCESSING) {
  872. $segmentUpdateData['video_task_status'] = '生成中';
  873. }
  874. // 更新分镜表
  875. DB::table('mp_episode_segments')
  876. ->where('video_task_id', $task->id)
  877. ->update($segmentUpdateData);
  878. dLog('generate')->info('分镜视频状态更新成功: ' . $segment->segment_id . ', 状态: ' . ($segmentUpdateData['video_task_status'] ?? '未知'));
  879. } catch (\Exception $e) {
  880. dLog('generate')->error('更新分镜视频状态失败: ' . $e->getMessage());
  881. }
  882. }
  883. /**
  884. * 生成可灵AI的JWT Token
  885. *
  886. * @return string
  887. */
  888. private function generateKelingJwtToken(): string
  889. {
  890. $ak = env('KELING_AK');
  891. $sk = env('KELING_SK');
  892. if (empty($ak) || empty($sk)) {
  893. throw new \Exception('可灵AI访问密钥未配置,请检查环境变量KELING_AK和KELING_SK');
  894. }
  895. // JWT头部
  896. $header = [
  897. 'alg' => 'HS256',
  898. 'typ' => 'JWT'
  899. ];
  900. // JWT载荷
  901. $payload = [
  902. 'iss' => $ak, // 签发者,使用Access Key
  903. 'exp' => time() + 1800, // 过期时间,30分钟后过期
  904. 'nbf' => time() - 5 // 生效时间,5秒前开始生效
  905. ];
  906. // Base64URL编码
  907. $headerEncoded = $this->base64UrlEncode(json_encode($header));
  908. $payloadEncoded = $this->base64UrlEncode(json_encode($payload));
  909. // 创建签名
  910. $signature = hash_hmac('sha256', $headerEncoded . '.' . $payloadEncoded, $sk, true);
  911. $signatureEncoded = $this->base64UrlEncode($signature);
  912. // 返回完整的JWT token
  913. return $headerEncoded . '.' . $payloadEncoded . '.' . $signatureEncoded;
  914. }
  915. /**
  916. * Base64URL编码(JWT标准编码)
  917. *
  918. * @param string $data
  919. * @return string
  920. */
  921. private function base64UrlEncode(string $data): string
  922. {
  923. return rtrim(strtr(base64_encode($data), '+/', '-_'), '=');
  924. }
  925. /**
  926. * 创建可灵AI Omni视频生成任务
  927. *
  928. * @param array $params
  929. * @return MpGenerateVideoTask
  930. */
  931. public function createKelingOmniTask(array $params): MpGenerateVideoTask
  932. {
  933. // 生成唯一的任务ID
  934. $taskId = 'keling_omni_' . time() . '_' . uniqid();
  935. // 创建任务记录,复用现有表结构
  936. $task = MpGenerateVideoTask::create([
  937. 'task_id' => $taskId,
  938. 'prompt' => $params['prompt'] ?? '',
  939. 'first_frame_url' => $params['first_frame_url'] ?? '',
  940. 'tail_frame_url' => $params['tail_frame_url'] ?? '',
  941. 'seed' => $params['seed'] ?? -1,
  942. 'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : null,
  943. 'video_resolution' => $params['video_resolution'] ?? '720P',
  944. 'status' => MpGenerateVideoTask::STATUS_PENDING,
  945. 'api_type' => 'keling', // 标识为可灵AI Omni API
  946. 'extra_params' => [
  947. 'api_type' => 'keling',
  948. 'model' => $params['model'] ?? 'kling-video-o1',
  949. 'prompt' => $params['prompt'] ?? '',
  950. 'multi_prompt' => $params['multi_prompt'] ?? [],
  951. 'shot_type' => $params['shot_type'] ?? 'customize',
  952. 'image_list' => $params['image_list'] ?? [],
  953. 'element_list' => $params['element_list'] ?? [],
  954. 'sound' => $params['sound'] ?? 'off',
  955. // 'aspect_ratio' => $params['aspect_ratio'] ?? '16:9',
  956. 'duration' => $params['video_duration'] ?? 5,
  957. 'mode' => $params['mode'] ?? 'std',
  958. 'callback_url' => $params['callback_url'] ?? '',
  959. ]
  960. ]);
  961. // 提交任务到可灵AI Omni API
  962. $this->submitKelingOmniTaskToApi($task);
  963. return $task;
  964. }
  965. /**
  966. * 提交可灵AI Omni任务到API
  967. *
  968. * @param MpGenerateVideoTask $task
  969. * @return void
  970. */
  971. private function submitKelingOmniTaskToApi(MpGenerateVideoTask $task): void
  972. {
  973. $client = new Client(['verify'=>false,'timeout'=>120]);
  974. try {
  975. // 生成JWT token
  976. $jwtToken = $this->generateKelingJwtToken();
  977. $extraParams = $task->extra_params;
  978. // 构建可灵AI Omni API请求参数
  979. $apiParams = [
  980. 'model_name' => $extraParams['model'],
  981. 'prompt' => $extraParams['prompt'],
  982. 'mode' => $extraParams['mode'],
  983. // 'aspect_ratio' => $extraParams['aspect_ratio'],
  984. 'duration' => (int)$extraParams['duration'],
  985. ];
  986. // 添加可选参数
  987. if (!empty($extraParams['multi_prompt'])) {
  988. $apiParams['multi_prompt'] = $extraParams['multi_prompt'];
  989. }
  990. if (!empty($extraParams['shot_type'])) {
  991. $apiParams['shot_type'] = $extraParams['shot_type'];
  992. }
  993. if (!empty($extraParams['image_list'])) {
  994. $apiParams['image_list'] = $extraParams['image_list'];
  995. }
  996. if (!empty($extraParams['element_list'])) {
  997. $apiParams['element_list'] = json_decode($extraParams['element_list'], true);
  998. }
  999. if (isset($extraParams['sound'])) {
  1000. $apiParams['sound'] = $extraParams['sound'];
  1001. }
  1002. // if (!empty($extraParams['callback_url'])) {
  1003. // $apiParams['callback_url'] = $extraParams['callback_url'];
  1004. // }
  1005. dLog('generate')->info('可灵AI Omni视频生成参数: ', $apiParams);
  1006. logDB('generate', 'info', '可灵AI Omni视频生成任务提交', ['id' => $task->id, 'params' => $apiParams]);
  1007. // 调用可灵AI Omni API
  1008. $response = $client->post('https://api-beijing.klingai.com/v1/videos/omni-video', [
  1009. 'headers' => [
  1010. 'Authorization' => 'Bearer ' . $jwtToken,
  1011. 'Content-Type' => 'application/json',
  1012. ],
  1013. 'json' => $apiParams
  1014. ]);
  1015. $responseData = json_decode($response->getBody(), true);
  1016. dLog('generate')->info('可灵AI Omni API响应: ', $responseData);
  1017. logDB('generate', 'info', '可灵AI Omni API响应', ['id' => $task->id, 'response' => $responseData]);
  1018. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1019. // API返回错误
  1020. $errorMessage = $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error';
  1021. logDB('generate', 'error', '可灵AI Omni任务提交失败', ['id' => $task->id, 'error' => $errorMessage]);
  1022. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1023. 'error_message' => $errorMessage,
  1024. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  1025. ]);
  1026. } elseif (isset($responseData['data']['task_id'])) {
  1027. // 任务提交成功,更新状态为处理中
  1028. $updateData = [
  1029. 'task_id' => $responseData['data']['task_id'], // 使用API返回的任务ID
  1030. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  1031. ];
  1032. $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
  1033. } else {
  1034. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1035. 'error_message' => '未知的API响应格式',
  1036. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  1037. ]);
  1038. }
  1039. } catch (\Exception $e) {
  1040. // 记录错误
  1041. logDB('generate', 'error', '可灵AI Omni任务异常', ['id' => $task->id, 'error' => $e->getMessage()]);
  1042. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1043. 'error_message' => 'API请求失败: ' . $e->getMessage()
  1044. ]);
  1045. dLog('generate')->error('可灵AI Omni API请求异常: ' . $e->getMessage());
  1046. }
  1047. }
  1048. /**
  1049. * 查询可灵AI Omni任务状态
  1050. *
  1051. * @param MpGenerateVideoTask $task
  1052. * @return array
  1053. */
  1054. public function queryKelingOmniTaskStatus(MpGenerateVideoTask $task): array
  1055. {
  1056. $client = new Client(['verify'=>false,'timeout'=>120]);
  1057. try {
  1058. // 如果任务已完成,直接返回结果
  1059. if (in_array($task->status, [MpGenerateVideoTask::STATUS_SUCCESS, MpGenerateVideoTask::STATUS_FAILED])) {
  1060. return [
  1061. 'success' => true,
  1062. 'data' => [
  1063. 'task_id' => $task->task_id,
  1064. 'task_status' => $this->mapTaskStatusToKelingStatus($task->status),
  1065. 'result_url' => $task->result_url,
  1066. 'error_message' => $task->error_message,
  1067. 'created_at' => $task->created_at,
  1068. 'updated_at' => $task->updated_at
  1069. ]
  1070. ];
  1071. }
  1072. // 生成JWT token
  1073. $jwtToken = $this->generateKelingJwtToken();
  1074. // 构建查询URL
  1075. $url = 'https://api-beijing.klingai.com/v1/videos/omni-video/' . $task->task_id;
  1076. $response = $client->get($url, [
  1077. 'headers' => [
  1078. 'Authorization' => 'Bearer ' . $jwtToken,
  1079. 'Content-Type' => 'application/json',
  1080. ]
  1081. ]);
  1082. $responseData = json_decode($response->getBody(), true);
  1083. dLog('generate')->info('可灵AI Omni任务状态查询响应: ', $responseData);
  1084. logDB('generate', 'info', '可灵AI Omni任务状态查询', ['id' => $task->id, 'response' => $responseData]);
  1085. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1086. logDB('generate', 'error', '可灵AI Omni任务状态查询失败', ['id' => $task->id, 'error' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败']);
  1087. return [
  1088. 'status' => 'failed',
  1089. 'error_message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1090. 'result_json' => $responseData
  1091. ];
  1092. }
  1093. // 解析任务状态
  1094. $taskData = $responseData['data'] ?? [];
  1095. $taskStatus = $taskData['task_status'] ?? 'unknown';
  1096. $updateData = ['result_json' => $responseData];
  1097. if ($taskStatus === 'succeed') {
  1098. $updateData['status'] = 'success';
  1099. // 任务成功,处理结果
  1100. if (isset($taskData['task_result']['videos']) && !empty($taskData['task_result']['videos'])) {
  1101. $video = $taskData['task_result']['videos'][0];
  1102. $video_url = $video['url'];
  1103. // 下载并保存视频
  1104. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  1105. $video_ext = getVideoExtFromUrl($video_url);
  1106. $video_name = $video_name . $video_ext;
  1107. $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
  1108. $updateData['result_url'] = $url;
  1109. // 计算视频时长
  1110. $video_duration = 0;
  1111. if (isset($video['duration'])) {
  1112. $video_duration = (int)$video['duration'];
  1113. } elseif (isset($taskData['duration'])) {
  1114. $video_duration = (int)$taskData['duration'];
  1115. } elseif (isset($video['frames']) && isset($video['framespersecond'])) {
  1116. $frames = (int)$video['frames'];
  1117. $fps = (int)$video['framespersecond'];
  1118. if ($fps > 0) {
  1119. $video_duration = (int)floor($frames / $fps);
  1120. }
  1121. }
  1122. // 如果没有计算出时长,使用任务表中的设定值
  1123. if ($video_duration <= 0 && !empty($task->extra_params)) {
  1124. $extraParams = is_array($task->extra_params) ? $task->extra_params : json_decode($task->extra_params, true);
  1125. if (isset($extraParams['duration'])) {
  1126. $video_duration = (int)$extraParams['duration'];
  1127. }
  1128. }
  1129. // 只有当video_duration > 0时才添加到更新数据中
  1130. if ($video_duration > 0) {
  1131. $updateData['video_duration'] = $video_duration;
  1132. $updateData['video_time_point_start'] = 0;
  1133. $updateData['video_time_point_end'] = $video_duration;
  1134. }
  1135. }
  1136. return $updateData;
  1137. } elseif ($taskStatus === 'failed') {
  1138. $updateData = [
  1139. 'status' => 'failed',
  1140. 'error_message' => '任务执行失败',
  1141. 'result_json' => $responseData
  1142. ];
  1143. return $updateData;
  1144. }
  1145. // 其他状态(processing等)保持不变
  1146. return [
  1147. 'success' => true,
  1148. 'data' => [
  1149. 'task_id' => $task->task_id,
  1150. 'task_status' => $taskStatus,
  1151. 'result_url' => $task->result_url,
  1152. 'error_message' => $task->error_message,
  1153. 'created_at' => $task->created_at,
  1154. 'updated_at' => $task->updated_at,
  1155. 'api_response' => $responseData
  1156. ]
  1157. ];
  1158. } catch (\Exception $e) {
  1159. dLog('generate')->error('可灵AI Omni任务状态查询异常: ' . $e->getMessage());
  1160. return [
  1161. 'success' => false,
  1162. 'message' => '查询失败: ' . $e->getMessage()
  1163. ];
  1164. }
  1165. }
  1166. /**
  1167. * 映射任务状态到可灵AI状态格式
  1168. *
  1169. * @param string $taskStatus
  1170. * @return string
  1171. */
  1172. private function mapTaskStatusToKelingStatus(string $taskStatus): string
  1173. {
  1174. switch ($taskStatus) {
  1175. case MpGenerateVideoTask::STATUS_PENDING:
  1176. return 'submitted';
  1177. case MpGenerateVideoTask::STATUS_PROCESSING:
  1178. return 'processing';
  1179. case MpGenerateVideoTask::STATUS_SUCCESS:
  1180. return 'succeed';
  1181. case MpGenerateVideoTask::STATUS_FAILED:
  1182. return 'failed';
  1183. default:
  1184. return 'unknown';
  1185. }
  1186. }
  1187. /**
  1188. * 更新可灵AI Omni视频任务状态
  1189. *
  1190. * @param MpGenerateVideoTask $task
  1191. * @return void
  1192. */
  1193. private function updateKelingOmniTask($task): void
  1194. {
  1195. $statusInfo = $this->queryKelingOmniTaskStatus($task);
  1196. if (!isset($statusInfo['status'])) return;
  1197. if ($statusInfo['status'] === 'success') {
  1198. logDB('generate', 'info', '可灵AI Omni视频生成任务成功', ['id' => $task->id, 'result_url' => $statusInfo['result_url']]);
  1199. $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
  1200. // 同步调整分镜视频状态和结果
  1201. $segment_id = getProp($task, 'alias_segment_id');
  1202. if ($segment_id && isset($statusInfo['result_url'])) {
  1203. try {
  1204. DB::beginTransaction();
  1205. $now = date('Y-m-d H:i:s');
  1206. // 构建更新数据
  1207. $segmentUpdateData = [
  1208. 'video_url' => $statusInfo['result_url'],
  1209. 'video_task_status' => '已完成',
  1210. 'current_type' => 2,
  1211. 'last_frame_url' => $statusInfo['last_frame_url'] ?? '',
  1212. 'updated_at' => $now
  1213. ];
  1214. // 只有当video_duration存在且大于0时才更新
  1215. if (isset($statusInfo['video_duration']) && $statusInfo['video_duration'] > 0) {
  1216. $segmentUpdateData['video_duration'] = $statusInfo['video_duration'];
  1217. $segmentUpdateData['video_time_point_start'] = 0;
  1218. // $segmentUpdateData['video_time_point_end'] = $statusInfo['video_duration'];
  1219. $audio_duration = DB::table('mp_episode_segments')->where('segment_id', $segment_id)->value('audio_duration');
  1220. $segmentUpdateData['video_time_point_end'] = $audio_duration ?? 0;
  1221. }
  1222. // 更新分镜表
  1223. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update($segmentUpdateData);
  1224. // 添加视频生成对话记录
  1225. $this->addVideoGenerationRecords($segment_id, $statusInfo['result_url']);
  1226. DB::commit();
  1227. } catch (\Exception $e) {
  1228. DB::rollBack();
  1229. dLog('generate')->error('可灵视频任务处理失败', [
  1230. 'segment_id' => $segment_id,
  1231. 'task_id' => $task->id,
  1232. 'error' => $e->getMessage()
  1233. ]);
  1234. }
  1235. }
  1236. } elseif ($statusInfo['status'] === 'failed') {
  1237. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1238. 'error_message' => $statusInfo['error_message'],
  1239. 'result_json' => $statusInfo['result_json'] ?? []
  1240. ]);
  1241. // 同步更新分镜表状态为失败
  1242. $segment_id = getProp($task, 'alias_segment_id');
  1243. if ($segment_id) {
  1244. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  1245. 'video_task_status' => '失败',
  1246. 'updated_at' => date('Y-m-d H:i:s')
  1247. ]);
  1248. }
  1249. }
  1250. // 如果仍然是处理中状态,不做任何操作
  1251. // 处理:如果任务处理超过24小时,标记为失败
  1252. $processingTime = now()->diffInHours($task->created_at);
  1253. if ($processingTime > 24) {
  1254. logDB('generate', 'warning', '可灵AI Omni视频生成任务超时', ['id' => $task->id, 'processing_hours' => $processingTime]);
  1255. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1256. 'error_message' => '任务处理超时(超过24小时)'
  1257. ]);
  1258. // 同步更新分镜表状态为失败
  1259. $segment_id = getProp($task, 'alias_segment_id');
  1260. if ($segment_id) {
  1261. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  1262. 'video_task_status' => '失败',
  1263. 'updated_at' => date('Y-m-d H:i:s')
  1264. ]);
  1265. }
  1266. }
  1267. }
  1268. /**
  1269. * 映射任务状态到前端显示状态
  1270. *
  1271. * @param string $taskStatus
  1272. * @return string
  1273. */
  1274. private function mapTaskStatus(string $taskStatus): string
  1275. {
  1276. switch ($taskStatus) {
  1277. case MpGenerateVideoTask::STATUS_PENDING:
  1278. return 'queued';
  1279. case MpGenerateVideoTask::STATUS_PROCESSING:
  1280. return 'running';
  1281. case MpGenerateVideoTask::STATUS_SUCCESS:
  1282. return 'succeeded';
  1283. case MpGenerateVideoTask::STATUS_FAILED:
  1284. return 'failed';
  1285. default:
  1286. return 'unknown';
  1287. }
  1288. }
  1289. // ==================== 主体管理相关方法 ====================
  1290. /**
  1291. * 创建主体
  1292. *
  1293. * @param array $params
  1294. * @return AIElement
  1295. */
  1296. public function createElement(array $params): AIElement
  1297. {
  1298. // 生成唯一的任务ID
  1299. $taskId = 'element_' . time() . '_' . uniqid();
  1300. // 创建主体记录
  1301. $element = AIElement::create([
  1302. 'task_id' => $taskId,
  1303. 'external_task_id' => $params['external_task_id'] ?? null,
  1304. 'element_name' => $params['element_name'],
  1305. 'element_description' => $params['element_description'],
  1306. 'reference_type' => $params['reference_type'],
  1307. 'element_image_list' => $params['element_image_list'] ?? null,
  1308. 'element_video_list' => $params['element_video_list'] ?? null,
  1309. 'element_voice_id' => $params['element_voice_id'] ?? null,
  1310. 'tag_list' => $params['tag_list'] ?? null,
  1311. 'callback_url' => $params['callback_url'] ?? null,
  1312. 'task_status' => AIElement::STATUS_SUBMITTED,
  1313. ]);
  1314. // 提交到可灵AI API
  1315. $this->submitElementToKelingApi($element);
  1316. return $element;
  1317. }
  1318. /**
  1319. * 提交主体创建任务到可灵AI API
  1320. *
  1321. * @param AIElement $element
  1322. * @return void
  1323. */
  1324. private function submitElementToKelingApi(AIElement $element): void
  1325. {
  1326. $client = new Client(['verify' => false, 'timeout' => 120]);
  1327. try {
  1328. // 生成JWT token
  1329. $jwtToken = $this->generateKelingJwtToken();
  1330. // 构建API请求参数
  1331. $apiParams = [
  1332. 'element_name' => $element->element_name,
  1333. 'element_description' => $element->element_description,
  1334. 'reference_type' => $element->reference_type,
  1335. ];
  1336. // 添加图片参考
  1337. if ($element->reference_type === AIElement::REFERENCE_TYPE_IMAGE && $element->element_image_list) {
  1338. $apiParams['element_image_list'] = $element->element_image_list;
  1339. }
  1340. // 添加视频参考
  1341. if ($element->reference_type === AIElement::REFERENCE_TYPE_VIDEO && $element->element_video_list) {
  1342. $apiParams['element_video_list'] = $element->element_video_list;
  1343. }
  1344. // 添加音色ID
  1345. if ($element->element_voice_id) {
  1346. $apiParams['element_voice_id'] = $element->element_voice_id;
  1347. }
  1348. // 添加标签
  1349. if ($element->tag_list) {
  1350. $apiParams['tag_list'] = $element->tag_list;
  1351. }
  1352. // 添加回调URL
  1353. if ($element->callback_url) {
  1354. $apiParams['callback_url'] = $element->callback_url;
  1355. }
  1356. // 添加外部任务ID
  1357. if ($element->external_task_id) {
  1358. $apiParams['external_task_id'] = $element->external_task_id;
  1359. }
  1360. dLog('generate')->info('可灵AI创建主体参数: ', $apiParams);
  1361. // 调用可灵AI创建主体API
  1362. $response = $client->post('https://api-beijing.klingai.com/v1/general/advanced-custom-elements', [
  1363. 'headers' => [
  1364. 'Authorization' => 'Bearer ' . $jwtToken,
  1365. 'Content-Type' => 'application/json',
  1366. ],
  1367. 'json' => $apiParams
  1368. ]);
  1369. $responseData = json_decode($response->getBody(), true);
  1370. dLog('generate')->info('可灵AI创建主体API响应: ', $responseData);
  1371. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1372. // API返回错误
  1373. $errorMessage = $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error';
  1374. $element->update([
  1375. 'task_status' => AIElement::STATUS_FAILED,
  1376. 'task_status_msg' => $errorMessage,
  1377. ]);
  1378. } elseif (isset($responseData['data']['task_id'])) {
  1379. // 任务提交成功,更新状态为处理中
  1380. $element->update([
  1381. 'task_id' => $responseData['data']['task_id'], // 使用API返回的任务ID
  1382. 'task_status' => AIElement::STATUS_PROCESSING,
  1383. 'created_at' => isset($responseData['data']['created_at']) ?
  1384. date('Y-m-d H:i:s', $responseData['data']['created_at'] / 1000) : now(),
  1385. 'updated_at' => isset($responseData['data']['updated_at']) ?
  1386. date('Y-m-d H:i:s', $responseData['data']['updated_at'] / 1000) : now(),
  1387. ]);
  1388. } else {
  1389. $element->update([
  1390. 'task_status' => AIElement::STATUS_FAILED,
  1391. 'task_status_msg' => '未知的API响应格式',
  1392. ]);
  1393. }
  1394. } catch (\Exception $e) {
  1395. // 记录错误
  1396. $element->update([
  1397. 'task_status' => AIElement::STATUS_FAILED,
  1398. 'task_status_msg' => 'API请求失败: ' . $e->getMessage(),
  1399. ]);
  1400. dLog('generate')->error('可灵AI创建主体API请求异常: ' . $e->getMessage());
  1401. }
  1402. }
  1403. /**
  1404. * 查询单个主体状态
  1405. *
  1406. * @param string $taskId
  1407. * @param string|null $externalTaskId
  1408. * @return array
  1409. */
  1410. public function queryElementStatus(string $taskId, string $externalTaskId = null): array
  1411. {
  1412. $client = new Client(['verify' => false, 'timeout' => 120]);
  1413. try {
  1414. // 生成JWT token
  1415. $jwtToken = $this->generateKelingJwtToken();
  1416. // 构建查询URL
  1417. $url = 'https://api-beijing.klingai.com/v1/general/advanced-custom-elements/' . $taskId;
  1418. $queryParams = [];
  1419. if ($externalTaskId) {
  1420. $queryParams['external_task_id'] = $externalTaskId;
  1421. }
  1422. if (!empty($queryParams)) {
  1423. $url .= '?' . http_build_query($queryParams);
  1424. }
  1425. $response = $client->get($url, [
  1426. 'headers' => [
  1427. 'Authorization' => 'Bearer ' . $jwtToken,
  1428. 'Content-Type' => 'application/json',
  1429. ]
  1430. ]);
  1431. $responseData = json_decode($response->getBody(), true);
  1432. dLog('generate')->info('可灵AI查询主体状态响应: ', $responseData);
  1433. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1434. return [
  1435. 'success' => false,
  1436. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1437. 'data' => null
  1438. ];
  1439. }
  1440. // 更新本地数据库记录
  1441. if (isset($responseData['data'])) {
  1442. $this->updateLocalElementFromApi($responseData['data']);
  1443. }
  1444. return [
  1445. 'success' => true,
  1446. 'data' => $responseData['data']
  1447. ];
  1448. } catch (\Exception $e) {
  1449. dLog('generate')->error('可灵AI查询主体状态异常: ' . $e->getMessage());
  1450. return [
  1451. 'success' => false,
  1452. 'message' => '查询失败: ' . $e->getMessage(),
  1453. 'data' => null
  1454. ];
  1455. }
  1456. }
  1457. /**
  1458. * 根据API响应更新本地主体记录
  1459. *
  1460. * @param array $apiData
  1461. * @return void
  1462. */
  1463. private function updateLocalElementFromApi(array $apiData): void
  1464. {
  1465. $element = AIElement::where('task_id', $apiData['task_id'])->first();
  1466. if (!$element) {
  1467. return;
  1468. }
  1469. $updateData = [
  1470. 'task_status' => $apiData['task_status'],
  1471. 'task_status_msg' => $apiData['task_status_msg'] ?? null,
  1472. 'final_unit_deduction' => $apiData['final_unit_deduction'] ?? null,
  1473. 'updated_at' => isset($apiData['updated_at']) ?
  1474. date('Y-m-d H:i:s', $apiData['updated_at'] / 1000) : now(),
  1475. ];
  1476. // 如果任务成功,更新主体信息
  1477. if ($apiData['task_status'] === AIElement::STATUS_SUCCEED && isset($apiData['task_result']['elements'])) {
  1478. $elementData = $apiData['task_result']['elements'][0];
  1479. $updateData = array_merge($updateData, [
  1480. 'element_id' => $elementData['element_id'],
  1481. 'element_voice_info' => $elementData['element_voice_info'] ?? null,
  1482. 'owned_by' => $elementData['owned_by'] ?? 'custom',
  1483. 'status' => $elementData['status'] ?? AIElement::ELEMENT_STATUS_SUCCEED,
  1484. ]);
  1485. }
  1486. $element->update($updateData);
  1487. }
  1488. /**
  1489. * 查询自定义主体列表
  1490. *
  1491. * @param int $pageNum
  1492. * @param int $pageSize
  1493. * @return array
  1494. */
  1495. public function queryCustomElementsList(int $pageNum = 1, int $pageSize = 30): array
  1496. {
  1497. $client = new Client(['verify' => false, 'timeout' => 120]);
  1498. try {
  1499. // 生成JWT token
  1500. $jwtToken = $this->generateKelingJwtToken();
  1501. // 构建查询URL
  1502. $url = 'https://api-beijing.klingai.com/v1/general/advanced-custom-elements';
  1503. $queryParams = [
  1504. 'pageNum' => $pageNum,
  1505. 'pageSize' => $pageSize,
  1506. ];
  1507. $url .= '?' . http_build_query($queryParams);
  1508. $response = $client->get($url, [
  1509. 'headers' => [
  1510. 'Authorization' => 'Bearer ' . $jwtToken,
  1511. 'Content-Type' => 'application/json',
  1512. ]
  1513. ]);
  1514. $responseData = json_decode($response->getBody(), true);
  1515. dLog('generate')->info('可灵AI查询自定义主体列表响应: ', $responseData);
  1516. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1517. return [
  1518. 'success' => false,
  1519. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1520. 'data' => []
  1521. ];
  1522. }
  1523. // 更新本地数据库记录
  1524. if (isset($responseData['data']) && is_array($responseData['data'])) {
  1525. foreach ($responseData['data'] as $elementData) {
  1526. $this->updateLocalElementFromApi($elementData);
  1527. }
  1528. }
  1529. return [
  1530. 'success' => true,
  1531. 'data' => $responseData['data'] ?? []
  1532. ];
  1533. } catch (\Exception $e) {
  1534. dLog('generate')->error('可灵AI查询自定义主体列表异常: ' . $e->getMessage());
  1535. return [
  1536. 'success' => false,
  1537. 'message' => '查询失败: ' . $e->getMessage(),
  1538. 'data' => []
  1539. ];
  1540. }
  1541. }
  1542. /**
  1543. * 查询官方主体列表
  1544. *
  1545. * @param int $pageNum
  1546. * @param int $pageSize
  1547. * @return array
  1548. */
  1549. public function queryOfficialElementsList(int $pageNum = 1, int $pageSize = 30): array
  1550. {
  1551. $client = new Client(['verify' => false, 'timeout' => 120]);
  1552. try {
  1553. // 生成JWT token
  1554. $jwtToken = $this->generateKelingJwtToken();
  1555. // 构建查询URL
  1556. $url = 'https://api-beijing.klingai.com/v1/general/advanced-presets-elements';
  1557. $queryParams = [
  1558. 'pageNum' => $pageNum,
  1559. 'pageSize' => $pageSize,
  1560. ];
  1561. $url .= '?' . http_build_query($queryParams);
  1562. $response = $client->get($url, [
  1563. 'headers' => [
  1564. 'Authorization' => 'Bearer ' . $jwtToken,
  1565. 'Content-Type' => 'application/json',
  1566. ]
  1567. ]);
  1568. $responseData = json_decode($response->getBody(), true);
  1569. dLog('generate')->info('可灵AI查询官方主体列表响应: ', $responseData);
  1570. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1571. return [
  1572. 'success' => false,
  1573. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1574. 'data' => []
  1575. ];
  1576. }
  1577. return [
  1578. 'success' => true,
  1579. 'data' => $responseData['data'] ?? []
  1580. ];
  1581. } catch (\Exception $e) {
  1582. dLog('generate')->error('可灵AI查询官方主体列表异常: ' . $e->getMessage());
  1583. return [
  1584. 'success' => false,
  1585. 'message' => '查询失败: ' . $e->getMessage(),
  1586. 'data' => []
  1587. ];
  1588. }
  1589. }
  1590. /**
  1591. * 删除自定义主体
  1592. *
  1593. * @param string $elementId
  1594. * @return array
  1595. */
  1596. public function deleteElement(string $elementId): array
  1597. {
  1598. $client = new Client(['verify' => false, 'timeout' => 120]);
  1599. try {
  1600. // 生成JWT token
  1601. $jwtToken = $this->generateKelingJwtToken();
  1602. // 构建API请求参数
  1603. $apiParams = [
  1604. 'element_id' => $elementId,
  1605. ];
  1606. dLog('generate')->info('可灵AI删除主体参数: ', $apiParams);
  1607. // 调用可灵AI删除主体API
  1608. $response = $client->post('https://api-beijing.klingai.com/v1/general/delete-elements', [
  1609. 'headers' => [
  1610. 'Authorization' => 'Bearer ' . $jwtToken,
  1611. 'Content-Type' => 'application/json',
  1612. ],
  1613. 'json' => $apiParams
  1614. ]);
  1615. $responseData = json_decode($response->getBody(), true);
  1616. dLog('generate')->info('可灵AI删除主体API响应: ', $responseData);
  1617. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1618. // API返回错误
  1619. return [
  1620. 'success' => false,
  1621. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error'
  1622. ];
  1623. } else {
  1624. // 删除成功,更新本地记录状态
  1625. AIElement::where('element_id', $elementId)->update([
  1626. 'status' => AIElement::ELEMENT_STATUS_DELETED,
  1627. 'updated_at' => now(),
  1628. ]);
  1629. return [
  1630. 'success' => true,
  1631. 'message' => '主体删除成功',
  1632. 'data' => $responseData['data'] ?? []
  1633. ];
  1634. }
  1635. } catch (\Exception $e) {
  1636. dLog('generate')->error('可灵AI删除主体API请求异常: ' . $e->getMessage());
  1637. return [
  1638. 'success' => false,
  1639. 'message' => 'API请求失败: ' . $e->getMessage()
  1640. ];
  1641. }
  1642. }
  1643. /**
  1644. * 获取本地主体列表(包含自定义和官方)
  1645. *
  1646. * @param array $filters
  1647. * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
  1648. */
  1649. public function getLocalElementsList(array $filters = [])
  1650. {
  1651. $query = AIElement::query();
  1652. // 筛选条件
  1653. if (isset($filters['owned_by'])) {
  1654. $query->where('owned_by', $filters['owned_by']);
  1655. }
  1656. if (isset($filters['reference_type'])) {
  1657. $query->where('reference_type', $filters['reference_type']);
  1658. }
  1659. if (isset($filters['status'])) {
  1660. $query->where('status', $filters['status']);
  1661. }
  1662. if (isset($filters['task_status'])) {
  1663. $query->where('task_status', $filters['task_status']);
  1664. }
  1665. // 搜索关键词
  1666. if (isset($filters['keyword']) && !empty($filters['keyword'])) {
  1667. $query->where(function ($q) use ($filters) {
  1668. $q->where('element_name', 'like', '%' . $filters['keyword'] . '%')
  1669. ->orWhere('element_description', 'like', '%' . $filters['keyword'] . '%');
  1670. });
  1671. }
  1672. return $query->orderBy('created_at', 'desc')
  1673. ->paginate($filters['page_size'] ?? 20);
  1674. }
  1675. /**
  1676. * 添加视频生成对话记录
  1677. *
  1678. * @param int $segmentId 分镜ID
  1679. * @param string $videoUrl 视频URL
  1680. * @return void
  1681. */
  1682. private function addVideoGenerationRecords($segmentId, $videoUrl)
  1683. {
  1684. try {
  1685. // 获取分镜信息
  1686. $segment = DB::table('mp_episode_segments as a')->leftJoin('mp_animes as b', 'a.anime_id', 'b.id')
  1687. ->where('a.segment_id', $segmentId)
  1688. ->select('b.user_id as uid', 'a.anime_id', 'a.episode_number', 'a.tail_frame')
  1689. ->first();
  1690. if (!$segment) {
  1691. dLog('generate')->warning('分镜信息不存在,无法添加对话记录', ['segment_id' => $segmentId]);
  1692. return;
  1693. }
  1694. $now = date('Y-m-d H:i:s');
  1695. // 保存对话记录
  1696. $records = [
  1697. [
  1698. 'uid' => $segment->uid,
  1699. 'anime_id' => $segment->anime_id,
  1700. 'sequence' => $segment->episode_number,
  1701. 'role' => 'user',
  1702. 'content' => '图片转视频',
  1703. 'segment_id' => $segmentId,
  1704. 'video_url' => '',
  1705. 'created_at' => $now,
  1706. 'updated_at' => $now
  1707. ],
  1708. [
  1709. 'uid' => $segment->uid,
  1710. 'anime_id' => $segment->anime_id,
  1711. 'sequence' => $segment->episode_number,
  1712. 'role' => 'assistant',
  1713. 'content' => $segment->tail_frame ?: '',
  1714. 'segment_id' => $segmentId,
  1715. 'video_url' => $videoUrl,
  1716. 'created_at' => $now,
  1717. 'updated_at' => $now
  1718. ]
  1719. ];
  1720. DB::table('mp_anime_records')->insert($records);
  1721. dLog('generate')->info('视频生成对话记录添加成功', [
  1722. 'segment_id' => $segmentId,
  1723. 'video_url' => $videoUrl,
  1724. 'uid' => $segment->uid,
  1725. 'anime_id' => $segment->anime_id
  1726. ]);
  1727. } catch (\Exception $e) {
  1728. dLog('generate')->error('添加视频生成对话记录失败: ' . $e->getMessage(), [
  1729. 'segment_id' => $segmentId,
  1730. 'video_url' => $videoUrl,
  1731. 'error' => $e->getMessage()
  1732. ]);
  1733. }
  1734. }
  1735. }