AnimeService.php 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690
  1. <?php
  2. namespace App\Services\Anime;
  3. use App\Consts\ErrorConst;
  4. use App\Facade\Site;
  5. use App\Libs\Utils;
  6. use App\Models\MpGeneratePicTask;
  7. use App\Services\AIGeneration\AIImageGenerationService;
  8. use Dflydev\DotAccessData\Util;
  9. use GuzzleHttp\Client;
  10. use Illuminate\Support\Facades\DB;
  11. use Illuminate\Support\Facades\Log;
  12. use Illuminate\Support\Facades\Redis;
  13. use OSS\Core\OssException;
  14. use OSS\OssClient;
  15. class AnimeService
  16. {
  17. protected $aiImageGenerationService;
  18. private $url;
  19. private $api_key;
  20. private $headers;
  21. public function __construct(AIImageGenerationService $aiImageGenerationService) {
  22. $this->aiImageGenerationService = $aiImageGenerationService;
  23. $this->url = 'https://api.deepseek.com/chat/completions';
  24. $this->api_key = env('DEEPSEEK_API_KEY');
  25. $this->headers = [
  26. 'Authorization' => 'Bearer '.$this->api_key,
  27. 'Content-Type' => 'application/json; charset=UTF-8'
  28. ];
  29. }
  30. public function batchSetRoleImg($data) {
  31. $anime_id = getProp($data, 'anime_id');
  32. $episode_number = getProp($data, 'episode_number');
  33. if (!$anime_id) Utils::throwError('1002:请选择对话');
  34. $anime = DB::table('mp_animes')->where('id', $anime_id)->where('is_deleted', 0)->first();
  35. $roles = json_decode(getProp($anime, 'roles'), true);
  36. foreach ($roles as &$role) {
  37. $role_name = getProp($role, 'role');
  38. $description = getProp($role, 'description');
  39. // 参考图地址
  40. $ref_img_url = getProp($role, 'url');
  41. // 此处调用AIImageGenerationService的生成图片任务接口: createImageGenerationTask(注意使用默认参数,prompt使用角色描述),并返回任务id
  42. try {
  43. $params = [
  44. 'prompt' => $description,
  45. 'ref_img_urls' => !empty($ref_img_url) ? [$ref_img_url] : []
  46. ];
  47. $task = $this->aiImageGenerationService->createImageGenerationTask($params);
  48. $task_id = $task->id;
  49. if (!$task_id) {
  50. Utils::throwError("20003: 角色({$role_name})生成图片失败");
  51. }
  52. $role['task_id'] = $task_id;
  53. $role['task_status'] = 'processing';
  54. } catch (\Exception $e) {
  55. Utils::throwError("20003:" . $e->getMessage());
  56. }
  57. }
  58. // 保存角色信息
  59. // $boolen = DB::table('mp_animes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->where('is_default', 1)->update([
  60. $boolen = DB::table('mp_animes')->where('id', $anime_id)->update([
  61. 'roles' => json_encode($roles, 256),
  62. 'generate_status' => '执行中',
  63. 'generate_at' => date('Y-m-d H:i:s'),
  64. 'updated_at' => date('Y-m-d H:i:s')
  65. ]);
  66. if (!$boolen) {
  67. Utils::throwError('20003:保存角色信息失败');
  68. }
  69. return $boolen;
  70. }
  71. public function batchSetSceneImg($data) {
  72. $anime_id = getProp($data, 'anime_id');
  73. if (!$anime_id) Utils::throwError('1002:请选择对话');
  74. $episode_number = getProp($data, 'episode_number');
  75. $anime = DB::table('mp_animes')->where('id', $anime_id)->where('is_deleted', 0)->first();
  76. $scenes = json_decode(getProp($anime, 'scenes'), true);
  77. foreach ($scenes as &$scene) {
  78. $scene_name = getProp($scene, 'scene');
  79. $description = getProp($scene, 'description');
  80. // 参考图地址
  81. $ref_img_url = getProp($scene, 'url');
  82. // 此处调用AIImageGenerationService的生成图片任务接口: createImageGenerationTask(注意使用默认参数,prompt使用角色描述),并返回任务id
  83. try {
  84. $params = [
  85. 'prompt' => $description,
  86. 'ref_img_urls' => !empty($ref_img_url) ? [$ref_img_url] : []
  87. ];
  88. $task = $this->aiImageGenerationService->createImageGenerationTask($params);
  89. $task_id = $task->id;
  90. if (!$task_id) {
  91. Utils::throwError("20003: 角色({$scene_name})生成图片失败");
  92. }
  93. $scene['task_id'] = $task_id;
  94. $scene['task_status'] = 'processing';
  95. } catch (\Exception $e) {
  96. Utils::throwError("20003:" . $e->getMessage());
  97. }
  98. }
  99. // 保存角色信息
  100. // $boolen = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->where('is_default', 1)->update([
  101. $boolen = DB::table('mp_animes')->where('id', $anime_id)->update([
  102. 'scenes' => json_encode($scenes, 256),
  103. 'generate_status' => '执行中',
  104. 'generate_at' => date('Y-m-d H:i:s'),
  105. 'updated_at' => date('Y-m-d H:i:s')
  106. ]);
  107. if (!$boolen) {
  108. Utils::throwError('20003:保存角色信息失败');
  109. }
  110. return $boolen;
  111. }
  112. public function editAnime($data) {
  113. $script_arr =getProp($data, 'script', []);
  114. if (!$script_arr) Utils::throwError('20003:剧本内容不能为空');
  115. $start_episode_sequence = getProp($data, 'start_episode_sequence', 1);
  116. $anime_id = getProp($data, 'anime_id');
  117. try {
  118. DB::beginTransaction();
  119. $table_data = [
  120. 'intro' => isset($script_arr['intro']) ? $script_arr['intro'] : '',
  121. 'highlights' => isset($script_arr['highlights']) ? $script_arr['highlights'] : '',
  122. 'role_relationship' => isset($script_arr['role_relationship']) ? $script_arr['role_relationship'] : '',
  123. 'core_contradiction' => isset($script_arr['core_contradiction']) ? $script_arr['core_contradiction'] : '',
  124. 'roles' => isset($script_arr['roles']) ? json_encode($script_arr['roles'], 256) : '',
  125. 'art_style' => isset($script_arr['art_style']) ? $script_arr['art_style'] : '',
  126. 'scenes' => isset($script_arr['scenes']) ? json_encode($script_arr['scenes'], 256) : '',
  127. 'status' => 3,
  128. 'start_episode_sequence' => $start_episode_sequence,
  129. 'end_episode_sequence' => $start_episode_sequence + count($script_arr['episodes']) - 1,
  130. 'episode_num' => count($script_arr['episodes']),
  131. 'updated_at' => date('Y-m-d H:i:s')
  132. ];
  133. // 保存剧本内容
  134. if (!empty($anime_id)) {
  135. $boolen = DB::table('mp_animes')->where('id', $anime_id)->update($table_data);
  136. if (!$boolen) {
  137. dLog('deepseek')->info('对话保存失败', $table_data);
  138. Utils::throwError('20003:对话保存失败');
  139. }
  140. }else {
  141. $table['created_at'] = $table_data['updated_at'];
  142. $anime_id = DB::table('mp_animes')->insertGetId($table_data);
  143. if (!$anime_id) {
  144. dLog('deepseek')->info('对话保存失败', $table_data);
  145. Utils::throwError('20003:对话保存失败');
  146. }
  147. }
  148. // 保存分集内容
  149. // 删除历史分集内容
  150. $boolen1 = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->delete();
  151. $episodes = [];
  152. $sequence = 1;
  153. foreach ($script_arr['episodes'] as $episode) {
  154. $segment_number = 1;
  155. foreach($episode['segments'] as $segment) {
  156. $episodes[] = [
  157. 'anime_id' => $anime_id,
  158. 'episode_number' => $episode['episode_number'],
  159. 'title' => $episode['title'],
  160. // 'content' => $episode['content'],
  161. 'content' => '',
  162. 'segment_number' => $segment_number,
  163. 'segment_content' => $segment['segment_content'],
  164. 'sequence' => $sequence,
  165. 'created_at' => date('Y-m-d H:i:s'),
  166. 'updated_at' => date('Y-m-d H:i:s')
  167. ];
  168. $sequence++;
  169. $segment_number++;
  170. }
  171. }
  172. $boolen2 = DB::table('mp_anime_episodes')->insert($episodes);
  173. if (!$boolen2) {
  174. dLog('deepseek')->info('分集内容保存失败', $episodes);
  175. Utils::throwError('20003:分集内容保存失败');
  176. }
  177. } catch (\Exception $e) {
  178. DB::rollBack();
  179. Utils::throwError('20003:'.$e->getMessage());
  180. }
  181. DB::commit();
  182. return true;
  183. }
  184. public function animeDetail($data) {
  185. $anime_id = getProp($data, 'anime_id');
  186. $anime = DB::table('mp_animes')->where('id', $anime_id)->where('is_deleted', 0)
  187. ->selectRaw('id as anime_id, anime_name, intro, highlights, roles, role_relationship, core_contradiction, art_style, scenes, status, remark, generate_status, created_at')->first();
  188. if (!$anime) Utils::throwError('20003:该剧本不存在!');
  189. $anime = (array)$anime;
  190. $anime['roles'] = json_decode($anime['roles']);
  191. $anime['scenes'] = json_decode($anime['scenes']);
  192. // 获取分集信息
  193. $episodes = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('is_default', 1)
  194. ->select('id as episode_id', 'episode_number', 'title')
  195. ->orderBy('episode_number')->get()->map(function ($value) {
  196. return (array)$value;
  197. })->toArray();
  198. $anime['episodes'] = $episodes;
  199. return $anime;
  200. }
  201. public function episodeInfo($data) {
  202. $anime_id = getProp($data, 'anime_id');
  203. $episode_number = getProp($data, 'episode_number');
  204. if (!$anime_id || !$episode_number) {
  205. Utils::throwError('1002:请选择剧集');
  206. }
  207. $episode = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->where('is_default', 1)->select('id as episode_id', 'anime_id', 'episode_number', 'title', 'intro', 'art_style', 'roles', 'scenes')->first();
  208. if (!$episode) Utils::throwError('20003: 该剧集不存在!');
  209. $episode = (array)$episode;
  210. $episode['roles'] = json_decode($episode['roles'], true);
  211. $episode['scenes'] = json_decode($episode['scenes'], true);
  212. // 获取分镜信息
  213. $segments = DB::table('mp_episode_segments')->where('anime_id', $anime_id)->where('episode_id', getProp($episode, 'episode_id'))->orderBy('segment_number')->select('act_number', 'act_title', 'segment_id', 'segment_number', 'segment_content')->get()->map(function ($value) {
  214. return (array)$value;
  215. })->toArray();
  216. $acts = [];
  217. foreach($segments as $segment) {
  218. if (isset($acts[getProp($segment, 'act_number')])) {
  219. $acts[getProp($segment, 'act_number')]['segments'][] = [
  220. 'segment_id' => getProp($segment, 'segment_id'),
  221. 'segment_number' => getProp($segment, 'segment_number'),
  222. 'segment_content' => getProp($segment, 'segment_content'),
  223. ];
  224. }else {
  225. $acts[getProp($segment, 'act_number')]= [
  226. 'act_number' => getProp($segment, 'act_number'),
  227. 'act_title' => getProp($segment, 'act_title'),
  228. 'segments' => [
  229. [
  230. 'segment_id' => getProp($segment, 'segment_id'),
  231. 'segment_number' => getProp($segment, 'segment_number'),
  232. 'segment_content' => getProp($segment, 'segment_content'),
  233. ]
  234. ]
  235. ];
  236. }
  237. }
  238. $episode['acts'] = array_values($acts);
  239. return $episode;
  240. }
  241. public function copyEpisodeVersion($data) {
  242. $episode_id = getProp($data, 'episode_id');
  243. $episode = DB::table('mp_anime_episodes')->where('id', $episode_id)->first();
  244. if (!$episode) Utils::throwError('20003:该剧集不存在');
  245. $episode = (array)$episode;
  246. $count = DB::table('mp_anime_episodes')->where('anime_id', $episode['anime_id'])->where('episode_number', $episode['episode_number'])->count('id');
  247. $sequence = $count + 1;
  248. unset($episode['id']);
  249. $episode['sequence'] = $sequence;
  250. $episode['created_at'] = date('Y-m-d H:i:s');
  251. $episode['updated_at'] = date('Y-m-d H:i:s');
  252. $episode['is_default'] = 1;
  253. try {
  254. DB::beginTransaction();
  255. // 移除is_default标志
  256. $boolen = DB::table('mp_anime_episodes')->where('anime_id', $episode['anime_id'])->where('episode_number', $episode['episode_number'])->update(['is_default' => 0, 'updated_at'=>date('Y-m-d H:i:s')]);
  257. if (!$boolen) {
  258. Utils::throwError('20003:更新失败!');
  259. }
  260. // 新增副本
  261. $new_episode_id = DB::table('mp_anime_episodes')->insertGetId($episode);
  262. if (!$new_episode_id) {
  263. Utils::throwError('20003:创建副本失败!');
  264. }
  265. $episode['episode_id'] = $new_episode_id;
  266. // 获取分镜数据
  267. $segments = DB::table('mp_episode_segments')->where('episode_id', $episode_id)->get()->map(function($item) use ($new_episode_id) {
  268. $item = (array)$item;
  269. unset($item['id']);
  270. $item['episode_id'] = $new_episode_id;
  271. $item['segment_id'] = date('YmdHis').mt_rand(1000,9999).str_pad($item['segment_number'], 3, "0", STR_PAD_LEFT);
  272. $item['created_at'] = $item['updated_at'] = date('Y-m-d H:i:s');
  273. return $item;
  274. })->toArray();
  275. // 写入分镜数据
  276. $boolen2 = DB::table('mp_episode_segments')->insert($segments);
  277. if (!$boolen2) {
  278. Utils::throwError('20003:写入分镜数据失败!');
  279. }
  280. }catch (\Exception $e) {
  281. DB::rollBack();
  282. Utils::throwError('20003:'.$e->getMessage());
  283. }
  284. DB::commit();
  285. return $episode;
  286. }
  287. public function episodeVersions($data) {
  288. $anime_id = getProp($data, 'anime_id');
  289. $episode_number = getProp($data, 'episode_number');
  290. if (!$anime_id || !$episode_number) {
  291. Utils::throwError('1002:请选择剧集');
  292. }
  293. $episodes = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->select('anime_id', 'episode_number', 'sequence', 'title', 'intro', 'art_style', 'roles', 'scenes')->get()->map(function($value) {
  294. return (array)$value;
  295. })->toArray();
  296. if (!$episodes) Utils::throwError('20003: 该剧集不存在!');
  297. foreach($episodes as &$episode) {
  298. $episode['version'] = 'V'.$episode['sequence'];
  299. $episode['roles'] = json_decode($episode['roles'], true);
  300. $episode['scenes'] = json_decode($episode['scenes'], true);
  301. }
  302. return $episodes;
  303. }
  304. public function bindEpisodeVersion($data) {
  305. $episode_id = getProp($data, 'episode_id');
  306. $episode = DB::table('mp_anime_episodes')->where('id', $episode_id)->first();
  307. if (!$episode) Utils::throwError('20003:该剧集不存在');
  308. $episode = (array)$episode;
  309. if ((int)$episode['is_default'] === 1) return true;
  310. try {
  311. DB::beginTransaction();
  312. // 移除is_default标志
  313. $boolen = DB::table('mp_anime_episodes')->where('anime_id', $episode['anime_id'])->where('episode_number', $episode['episode_number'])->update(['is_default' => 0, 'updated_at'=>date('Y-m-d H:i:s')]);
  314. if (!$boolen) {
  315. Utils::throwError('20003:更新失败!');
  316. }
  317. // 绑定副本
  318. $boolen2 = DB::table('mp_anime_episodes')->where('id', $episode_id)->update(['is_default' => 1, 'updated_at'=>date('Y-m-d H:i:s')]);
  319. if (!$boolen2) {
  320. Utils::throwError('20003:绑定失败!');
  321. }
  322. }catch (\Exception $e) {
  323. DB::rollBack();
  324. Utils::throwError('20003:'.$e->getMessage());
  325. }
  326. DB::commit();
  327. return true;
  328. }
  329. public function chatChangeImg($data) {
  330. $segment_id = getProp($data, 'segment_id');
  331. $prompt = getProp($data, 'prompt');
  332. if (!$prompt || !$segment_id) {
  333. Utils::throwError('1002:请输入提示词,并且选择需修改的图片');
  334. }
  335. $segment = DB::table("mp_episode_segments")->where('segment_id', $segment_id)->first();
  336. if (!$segment) Utils::throwError('20003:该分镜不存在!');
  337. $ref_img_url = getProp($segment, 'img_url');
  338. if (empty($ref_img_url)) {
  339. $prompt = getProp($segment, 'segment_content');
  340. }
  341. try {
  342. $params = [
  343. 'alias_segment_id' => $segment_id,
  344. 'prompt' => $prompt,
  345. 'ref_img_urls' => !empty($ref_img_url) ? [$ref_img_url] : []
  346. ];
  347. $task = $this->aiImageGenerationService->createImageGenerationTask($params);
  348. $task_id = $task->id;
  349. if (!$task_id) {
  350. Utils::throwError("20003: 生成图片失败");
  351. }
  352. // 更新任务ID
  353. $boolen = DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update(['pic_task_id' => $task_id, 'pic_task_status' => '生成中', 'updated_at' => date('Y-m-d H:i:s')]);
  354. if (!$boolen) Utils::throwError('20003:更新任务ID失败');
  355. } catch (\Exception $e) {
  356. Utils::throwError("20003:" . $e->getMessage());
  357. }
  358. $start_count = 0;
  359. $img_url = '';
  360. while ($start_count <= 30) {
  361. sleep(3);
  362. $task = MpGeneratePicTask::where('id', $task_id)->first();
  363. $statusInfo = $this->aiImageGenerationService->queryTaskStatus($task);
  364. if ($statusInfo['status'] === 'success') {
  365. $task->updateStatus('success', [
  366. 'result_url' => $statusInfo['result_url'],
  367. 'result_json' => $statusInfo['result_json'] ?? []
  368. ]);
  369. // 同步调整分镜图片状态和结果
  370. if (getProp($task, 'alias_segment_id')) {
  371. DB::table('mp_episode_segments')->where('segment_id', getProp($task, 'alias_segment_id'))->update([
  372. 'img_url' => $statusInfo['result_url'][0],
  373. 'pic_task_status' => '已完成',
  374. ]);
  375. }
  376. $img_url = $statusInfo['result_url'][0];
  377. break;
  378. } elseif ($statusInfo['status'] === 'failed') {
  379. $task->updateStatus('failed', [
  380. 'error_message' => $statusInfo['error_message'],
  381. 'result_json' => $statusInfo['result_json'] ?? []
  382. ]);
  383. if (getProp($task, 'alias_segment_id')) {
  384. DB::table('mp_episode_segments')->where('segment_id', getProp($task, 'alias_segment_id'))->update([
  385. 'pic_task_status' => '失败',
  386. ]);
  387. }
  388. break;
  389. }
  390. $start_count+=3;
  391. }
  392. if (!$img_url) Utils::throwError('20003:图片生成中,请稍后刷新');
  393. return $img_url;
  394. }
  395. public function batchSetSegmentTasks($data) {
  396. $anime_id = getProp($data, 'anime_id');
  397. $episode_number = getProp($data, 'episode_number');
  398. if (!$anime_id || !$episode_number) {
  399. Utils::throwError('1002:请选择剧集');
  400. }
  401. // 获取剧集信息
  402. $episode = DB::table('mp_anime_episodes')
  403. ->where('anime_id', $anime_id)
  404. ->where('episode_number', $episode_number)
  405. ->where('is_default', 1)
  406. ->select('id as episode_id', 'anime_id', 'episode_number')
  407. ->first();
  408. if (!$episode) {
  409. Utils::throwError('20003:该剧集不存在');
  410. }
  411. // 获取动漫的角色信息(包含参考图)
  412. $anime = DB::table('mp_animes')
  413. ->where('id', $anime_id)
  414. ->select('roles', 'scenes')
  415. ->first();
  416. if (!$anime || !$anime->roles) {
  417. Utils::throwError('20003:未找到角色信息');
  418. }
  419. $roles = json_decode($anime->roles, true);
  420. if (!$roles) {
  421. Utils::throwError('20003:角色信息格式错误');
  422. }
  423. // 构建角色名称到参考图的映射
  424. $role_map = [];
  425. foreach ($roles as $role) {
  426. $role_name = getProp($role, 'role');
  427. $role_url = getProp($role, 'url');
  428. if (!empty($role_name) && !empty($role_url)) {
  429. // // 处理角色名称,去除装扮后缀(如:许芸-校服装 -> 许芸)- 修复中文乱码问题(弃用,容易产生重复图片)
  430. // $base_name = preg_replace('/[--].*$/u', '', $role_name);
  431. // $role_map[$base_name] = $role_url;
  432. // 同时保存完整名称的映射
  433. $role_map[$role_name] = $role_url;
  434. }
  435. }
  436. $scenes = json_decode($anime->scenes, true);
  437. if (!$scenes) {
  438. Utils::throwError('20003:角色信息格式错误');
  439. }
  440. // 构建角色名称到参考图的映射
  441. $scene_map = [];
  442. foreach ($scenes as $scene) {
  443. $scene_name = getProp($scene, 'scene');
  444. $scene_url = getProp($scene, 'url');
  445. if (!empty($scene_name) && !empty($scene_url)) {
  446. $scene_map[$scene_name] = $scene_url;
  447. }
  448. }
  449. // 获取所有分镜信息
  450. $segments = DB::table('mp_episode_segments')
  451. ->where('anime_id', $anime_id)
  452. ->where('episode_id', $episode->episode_id)
  453. ->orderBy('segment_number')
  454. ->select('segment_id', 'segment_number', 'segment_content', 'scene', 'characters')
  455. ->limit(3)
  456. ->get()
  457. ->toArray();
  458. if (empty($segments)) {
  459. Utils::throwError('20003:未找到分镜信息');
  460. }
  461. $updated_segments = [];
  462. $failed_segments = [];
  463. // 批量为每个分镜生成图片任务
  464. foreach ($segments as $segment) {
  465. $segment_id = $segment->segment_id;
  466. $segment_content = $segment->segment_content;
  467. if (empty($segment_content)) {
  468. $failed_segments[] = [
  469. 'segment_id' => $segment_id,
  470. 'error' => '分镜内容为空'
  471. ];
  472. continue;
  473. }
  474. try {
  475. // 解析分镜内容,提取画面描述作为主要提示词
  476. $prompt = $segment_content;
  477. $ref_img_urls = [];
  478. // 分镜场景
  479. $scene = getProp($segment, 'scene');
  480. $matched_scene = '';
  481. if (isset($scene_map[$scene])) {
  482. $img_index = count($ref_img_urls) + 1;
  483. $prompt = "场景:$scene(图$img_index)\n$prompt";
  484. $ref_img_urls[] = $scene_map[$scene];
  485. $matched_scene = $scene;
  486. }
  487. // 分镜角色
  488. $characters = getProp($segment, 'characters');
  489. $characters = str_replace(['、', ',', ';'], ',', $characters);
  490. $roles = explode(',', $characters);
  491. // 从分镜内容中提取涉及的角色
  492. $segment_characters = $this->extractCharactersFromSegment($segment_content, array_keys($role_map), $roles);
  493. // 获取匹配角色的参考图
  494. foreach ($segment_characters as $character) {
  495. if (isset($role_map[$character])) {
  496. // 将$prompt中出现的角色替换成角色(图1、2、3,序号是$ref_img_urls的个数加1)
  497. $img_index = count($ref_img_urls) + 1;
  498. $prompt = str_replace($character, $character . '(图' . $img_index . ')', $prompt);
  499. $ref_img_urls[] = $role_map[$character];
  500. }
  501. }
  502. // 如果没有找到匹配的角色参考图,不使用参考图
  503. if (empty($ref_img_urls)) {
  504. $ref_img_urls = [];
  505. }
  506. // 准备生成任务参数
  507. $params = [
  508. 'alias_segment_id' => $segment_id,
  509. 'prompt' => $prompt,
  510. 'ref_img_urls' => $ref_img_urls
  511. ];
  512. // $task_id = mt_rand(100000, 999999);
  513. // 调用AI图片生成服务
  514. $task = $this->aiImageGenerationService->createImageGenerationTask($params);
  515. $task_id = $task->id;
  516. if (!$task_id) {
  517. $failed_segments[] = [
  518. 'segment_id' => $segment_id,
  519. 'error' => '创建生成任务失败'
  520. ];
  521. continue;
  522. }
  523. // 更新分镜的任务信息
  524. $update_result = DB::table('mp_episode_segments')
  525. ->where('segment_id', $segment_id)
  526. ->update([
  527. 'pic_task_id' => $task_id,
  528. 'pic_task_status' => '生成中',
  529. 'updated_at' => date('Y-m-d H:i:s')
  530. ]);
  531. if ($update_result) {
  532. $updated_segments[] = [
  533. 'segment_id' => $segment_id,
  534. 'task_id' => $task_id,
  535. 'status' => '生成中',
  536. 'matched_scenes' => $scene,
  537. 'matched_roles' => $segment_characters,
  538. 'ref_urls_count' => count($ref_img_urls)
  539. ];
  540. } else {
  541. $failed_segments[] = [
  542. 'segment_id' => $segment_id,
  543. 'error' => '更新分镜任务状态失败'
  544. ];
  545. }
  546. } catch (\Exception $e) {
  547. $failed_segments[] = [
  548. 'segment_id' => $segment_id,
  549. 'error' => $e->getMessage()
  550. ];
  551. }
  552. }
  553. // // 更新剧集生成状态
  554. // if (!empty($updated_segments)) {
  555. // DB::table('mp_anime_episodes')
  556. // ->where('id', $episode->episode_id)
  557. // ->update([
  558. // 'generate_status' => '执行中',
  559. // 'updated_at' => date('Y-m-d H:i:s')
  560. // ]);
  561. // }
  562. return [
  563. 'success_count' => count($updated_segments),
  564. 'failed_count' => count($failed_segments),
  565. 'success_segments' => $updated_segments,
  566. 'failed_segments' => $failed_segments,
  567. 'total_segments' => count($segments)
  568. ];
  569. }
  570. /**
  571. * 从分镜内容中提取涉及的角色
  572. */
  573. private function extractCharactersFromSegment($segment_content, $available_characters, $roles) {
  574. $found_characters = [];
  575. foreach ($available_characters as $character) {
  576. // 检查角色名称是否在分镜内容中出现
  577. if (strpos($segment_content, $character) !== false) {
  578. $found_characters[] = $character;
  579. }
  580. }
  581. if (!$found_characters) {
  582. foreach ($roles as $character) {
  583. // 检查角色名称是否在分镜内容中出现
  584. if (strpos($segment_content, $character) !== false) {
  585. $found_characters[] = $character;
  586. }
  587. }
  588. }
  589. return array_unique($found_characters);
  590. }
  591. }