AIImageGenerationService.php 83 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074
  1. <?php
  2. namespace App\Services\AIGeneration;
  3. use App\Facade\Site;
  4. use App\Consts\BaseConst;
  5. use App\Models\MpGeneratePicTask;
  6. use App\Models\MpAsset;
  7. use App\Services\PointsService;
  8. use App\Services\VolcEngineService;
  9. use GuzzleHttp\Client;
  10. use GuzzleHttp\Exception\GuzzleException;
  11. use Illuminate\Support\Facades\DB;
  12. use Illuminate\Support\Facades\Log;
  13. use Illuminate\Support\Facades\Redis;
  14. class AIImageGenerationService
  15. {
  16. private $volcEngineService;
  17. private $httpClient;
  18. /** @var PointsService */
  19. private $pointsService;
  20. // 模型常量
  21. const MODEL_JIMENG_4 = 'jimeng_4.0';
  22. const MODEL_NANO_BANANA_2 = 'NanoBanana2';
  23. const MODEL_NANO_BANANA_PRO = 'NanoBananaPro';
  24. const MODEL_GPT_IMAGE_2 = 'GptImage2';
  25. // 队列派发去重相关常量
  26. /** 派发锁在 Redis 中的 key 前缀 */
  27. const DISPATCH_LOCK_PREFIX = 'ai_pic_dispatch:';
  28. /** 派发锁使用的 Redis 连接,与队列保持一致 */
  29. const DISPATCH_LOCK_CONNECTION = 'default';
  30. /** 派发锁有效期(秒):worker 长时间未消费时的兜底重派时间 */
  31. const DISPATCH_LOCK_TTL = 600;
  32. /** 每轮最多派发的任务数,避免一次性推送过多任务 */
  33. const DISPATCH_BATCH_LIMIT = 50;
  34. /** 每轮扫描的候选任务数,需大于 DISPATCH_BATCH_LIMIT,避免已入队的任务挡住后面的任务 */
  35. const DISPATCH_CANDIDATE_LIMIT = 500;
  36. /** 派发锁分组:火山API */
  37. const DISPATCH_GROUP_VOLC = 'volc_pics';
  38. /** 派发锁分组:GPT-Image2 */
  39. const DISPATCH_GROUP_GPT_IMAGE2 = 'gpt_image2_pics';
  40. public function __construct(VolcEngineService $volcEngineService, PointsService $pointsService)
  41. {
  42. $this->volcEngineService = $volcEngineService;
  43. $this->pointsService = $pointsService;
  44. $this->httpClient = new Client([
  45. 'timeout' => 300,
  46. ]);
  47. }
  48. /**
  49. * 构建图片任务计费信息
  50. *
  51. * @param array $params
  52. * @param string $model
  53. * @return array
  54. */
  55. private function buildImageChargeInfo(array $params, string $model): array
  56. {
  57. $uid = 0;
  58. try {
  59. $uid = (int)Site::getUid();
  60. } catch (\Throwable $e) {
  61. // 非请求上下文(如命令行)下可能无法获取用户ID,置为0
  62. }
  63. $width = (int)getProp($params, 'width', 1600);
  64. $height = (int)getProp($params, 'height', 2848);
  65. return [
  66. 'user_id' => $uid,
  67. 'model' => $model,
  68. 'width' => $width,
  69. 'height' => $height,
  70. 'resolution' => $this->pointsService->normalizeImageResolutionKey($width, $height),
  71. 'image_num' => (int)getProp($params, 'image_num', 1),
  72. 'source' => !empty($params['alias_segment_id']) ? 'segment' : (!empty($params['alias_act_id']) ? 'act' : ''),
  73. 'alias_segment_id' => $params['alias_segment_id'] ?? '',
  74. 'created_at' => date('Y-m-d H:i:s'),
  75. ];
  76. }
  77. /**
  78. * 图片生成余额预检
  79. *
  80. * 有用户上下文时校验本次生成所需积分(单价 × 生成张数),不足直接抛错;
  81. * 后台命令/队列等无用户上下文的场景跳过预检。
  82. *
  83. * @param array $chargeInfo
  84. * @return void
  85. */
  86. private function preCheckImageBalance(array $chargeInfo): void
  87. {
  88. $uid = (int)($chargeInfo['user_id'] ?? 0);
  89. if (!$uid) {
  90. return;
  91. }
  92. $imageNum = (int)($chargeInfo['image_num'] ?? 1);
  93. if ($imageNum <= 0) {
  94. $imageNum = 1;
  95. }
  96. $needPoints = $this->pointsService->getImageChargePoints($chargeInfo) * $imageNum;
  97. if ($needPoints <= 0) {
  98. return;
  99. }
  100. $this->pointsService->checkUserPointsEnough($needPoints, $uid);
  101. }
  102. /**
  103. * 图片生成成功统一处理(事务保证原子性)
  104. *
  105. * 在同一事务内完成:任务状态更新 + 积分计费 + 分镜表同步;
  106. * 任一步失败整体回滚,任务保持原状态,由后续定时任务/超时重置重试。
  107. *
  108. * @param MpGeneratePicTask $task
  109. * @param array $resultUrls
  110. * @param array $resultJson
  111. * @return void
  112. */
  113. private function handleImageTaskSuccess(MpGeneratePicTask $task, array $resultUrls, array $resultJson): void
  114. {
  115. try {
  116. DB::beginTransaction();
  117. // 更新任务状态为成功
  118. $task->updateStatus(MpGeneratePicTask::STATUS_SUCCESS, [
  119. 'result_url' => $resultUrls,
  120. 'result_json' => $resultJson,
  121. ]);
  122. // 记录计费明细并扣减用户积分(真实失败则抛错回滚,避免“成功但未计费”)
  123. $chargeResult = $this->pointsService->recordImageTaskCharge($task);
  124. if (empty($chargeResult['charged'])
  125. && !in_array((string)($chargeResult['reason'] ?? ''), ['already_charged', 'no_charge_info'], true)) {
  126. throw new \Exception('图片计费失败: ' . (string)($chargeResult['reason'] ?? 'unknown'));
  127. }
  128. // 同步调整分镜图片状态和结果
  129. $segmentId = getProp($task, 'alias_segment_id');
  130. if ($segmentId) {
  131. $updateSegmentResult = DB::table('mp_episode_segments')->where('segment_id', $segmentId)->update([
  132. 'img_url' => $resultUrls[0],
  133. 'pic_task_status' => '已完成',
  134. ]);
  135. // 本应更新却未更新到(分镜被删除等),抛错触发整体回滚重试
  136. if ($updateSegmentResult === false || $updateSegmentResult === 0) {
  137. throw new \Exception('更新分镜图片状态失败: 未找到对应的分镜记录');
  138. }
  139. // 如果是首帧图ID则更新anime表
  140. if (Redis::sismember('anime_first_frame_urls', $segmentId)) {
  141. $animeId = DB::table('mp_episode_segments')->where('segment_id', $segmentId)->value('anime_id');
  142. DB::table('mp_animes')->where('id', $animeId)->update(['first_frame_url' => $resultUrls[0], 'updated_at' => date('Y-m-d H:i:s')]);
  143. Redis::srem('anime_first_frame_urls', $segmentId);
  144. }
  145. }
  146. DB::commit();
  147. } catch (\Exception $e) {
  148. DB::rollBack();
  149. dLog('generate')->error('图片任务成功处理失败,已回滚: ' . $e->getMessage(), ['task_id' => $task->id]);
  150. logDB('generate', 'error', '图片任务成功处理失败,已回滚', [
  151. 'task_id' => $task->id,
  152. 'error' => $e->getMessage(),
  153. ]);
  154. }
  155. }
  156. /**
  157. * 异步创建图片生成任务
  158. *
  159. * @param array $params
  160. * @return MpGeneratePicTask
  161. */
  162. public function createImageGenerationTask(array $params): MpGeneratePicTask
  163. {
  164. // // 获取模型类型,默认使用即梦AI 4.0
  165. // $model = $params['model'] ?? self::MODEL_JIMENG_4;
  166. // 获取模型类型,默认使用即梦seedream5.0-lite
  167. $model = $params['model'] ?? 'gpt-image-2';
  168. // 构建计费信息并余额预检(不足直接报错)
  169. $chargeInfo = $this->buildImageChargeInfo($params, $model);
  170. $this->preCheckImageBalance($chargeInfo);
  171. // 检查是否有正在处理的任务(仅即梦AI需要排队)
  172. if ($model === self::MODEL_JIMENG_4 && $this->hasProcessingTask()) {
  173. // 如果有正在处理的任务,创建排队任务
  174. return $this->createQueuedTask($params);
  175. }
  176. // 生成唯一的任务ID
  177. $taskId = 'img_gen_' . time() . '_' . uniqid();
  178. // 创建任务记录
  179. $task = MpGeneratePicTask::create([
  180. 'task_id' => $taskId,
  181. 'alias_segment_id' => $params['alias_segment_id'] ?? 0,
  182. 'prompt' => $params['prompt'] ?? '',
  183. 'width' => $params['width'] ?? 1600,
  184. 'height' => $params['height'] ?? 2848,
  185. 'image_num' => $params['image_num'] ?? 1,
  186. 'scale' => $params['scale'] ?? 50,
  187. 'ref_img_url' => $params['ref_img_urls'] ?? null,
  188. 'mask_img_url' => $params['mask_img_url'] ?? null,
  189. 'extra_params' => $params['extra_params'] ?? null,
  190. 'status' => MpGeneratePicTask::STATUS_PENDING,
  191. 'model' => $model,
  192. 'charge_info' => $chargeInfo,
  193. ]);
  194. // 即梦AI和NanoBanana系列立即提交任务,火山API在定时任务中提交
  195. if ($model === self::MODEL_JIMENG_4) {
  196. $this->submitTaskToJimengApi($task);
  197. } elseif (in_array($model, BaseConst::NANO_BANANA_MODELS)) {
  198. $this->submitTaskToNanoBananaApi($task);
  199. }
  200. return $task;
  201. }
  202. /**
  203. * 检查是否有正在处理的任务
  204. *
  205. * @return bool
  206. */
  207. private function hasProcessingTask(): bool
  208. {
  209. return MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PROCESSING)
  210. ->exists();
  211. }
  212. /**
  213. * 创建排队任务
  214. *
  215. * @param array $params
  216. * @return MpGeneratePicTask
  217. */
  218. private function createQueuedTask(array $params): MpGeneratePicTask
  219. {
  220. // 生成唯一的任务ID
  221. $taskId = 'img_gen_' . time() . '_' . uniqid();
  222. // 获取模型类型
  223. $model = $params['model'] ?? self::MODEL_JIMENG_4;
  224. // 构建计费信息并余额预检(不足直接报错)
  225. $chargeInfo = $this->buildImageChargeInfo($params, $model);
  226. $this->preCheckImageBalance($chargeInfo);
  227. // 创建排队状态的任务记录
  228. return MpGeneratePicTask::create([
  229. 'task_id' => $taskId,
  230. 'alias_segment_id' => $params['alias_segment_id'] ?? 0,
  231. 'prompt' => $params['prompt'] ?? '',
  232. 'width' => $params['width'] ?? 1600,
  233. 'height' => $params['height'] ?? 2848,
  234. 'image_num' => $params['image_num'] ?? 1,
  235. 'scale' => $params['scale'] ?? 50,
  236. 'ref_img_url' => $params['ref_img_urls'] ?? null,
  237. 'mask_img_url' => $params['mask_img_url'] ?? null,
  238. 'extra_params' => $params['extra_params'] ?? null,
  239. 'status' => MpGeneratePicTask::STATUS_PENDING,
  240. 'model' => $model,
  241. 'charge_info' => $chargeInfo,
  242. ]);
  243. }
  244. /**
  245. * 提交任务到即梦AI API
  246. *
  247. * @param MpGeneratePicTask $task
  248. * @return void
  249. */
  250. private function submitTaskToJimengApi(MpGeneratePicTask $task): void
  251. {
  252. try {
  253. // 验证环境变量配置
  254. $accessKey = env('VOLC_AI_AK');
  255. $secretKey = env('VOLC_AI_SK');
  256. if (empty($accessKey) || empty($secretKey)) {
  257. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  258. 'error_message' => '火山引擎访问密钥未配置,请检查环境变量AK和SK'
  259. ]);
  260. return;
  261. }
  262. //构建即梦AI 4.0 API请求参数
  263. $apiParams = [
  264. 'req_key' => 'jimeng_t2i_v40',
  265. 'prompt' => $task->prompt,
  266. 'scale' => $task->scale / 100, //为0-1范围
  267. ];
  268. // 添加可选参数
  269. if ($task->image_num > 1) {
  270. $apiParams['force_single'] = false;
  271. } else {
  272. $apiParams['force_single'] = true;
  273. }
  274. // 参考图片
  275. if ($task->ref_img_url) {
  276. $apiParams['image_urls'] = $task->ref_img_url;
  277. }
  278. // if ($task->mask_img_url) {
  279. // $apiParams['image_urls'] = $task->mask_img_url;
  280. // }
  281. // 尺寸参数
  282. $area = $task->width * $task->height; //计算面积
  283. if ($area >= 1024*1024 && $area <= 4096*4096) {
  284. $apiParams['size'] = (int)$area;
  285. }
  286. dLog('generate')->info('即梦AI文生图参数: ', $apiParams);
  287. logDB('generate', 'info', '即梦AI文生图任务提交', ['task_id' => $task->task_id, 'params' => $apiParams]);
  288. //调用即梦AI 4.0 API提交任务
  289. $response = $this->volcEngineService->request(
  290. 'POST',
  291. 'visual.volcengineapi.com', //即梦AI API域名
  292. '/', // 使用根路径
  293. [], // Query参数
  294. json_encode($apiParams),
  295. $accessKey,
  296. $secretKey,
  297. 'cv', // 服务标识
  298. 'cn-north-1', //区域
  299. 'CVSync2AsyncSubmitTask',
  300. '2022-08-31',
  301. 'application/json'
  302. );
  303. $responseData = json_decode($response['body'], true);
  304. dLog('generate')->info('即梦AI 4.0 API提交任务响应: ', $responseData);
  305. logDB('generate', 'info', '即梦AI 4.0 API提交任务响应', ['task_id' => $task->task_id, 'response' => $responseData]);
  306. if ($responseData['code'] !== 10000) {
  307. // API返回错误
  308. logDB('generate', 'error', '即梦AI任务提交失败', ['task_id' => $task->task_id, 'error' => $responseData['message'] ?? 'API Error']);
  309. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  310. 'error_message' => $responseData['message'] ?? 'API Error',
  311. 'extra_params' => $responseData['data'] ?? []
  312. ]);
  313. } else {
  314. // 任务提交成功,更新状态为处理中
  315. $updateData = [];
  316. //存储API返回的任务ID
  317. if (isset($responseData['data']['task_id'])) {
  318. $updateData['task_id'] = $responseData['data']['task_id'];
  319. $updateData['extra_params'] = $responseData ?? [];
  320. }
  321. $task->updateStatus(MpGeneratePicTask::STATUS_PROCESSING, $updateData);
  322. }
  323. } catch (\Exception $e) {
  324. // 记录错误
  325. logDB('generate', 'error', '即梦AI任务提交异常', ['task_id' => $task->task_id, 'error' => $e->getMessage()]);
  326. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  327. 'error_message' => 'API请求失败: ' . $e->getMessage()
  328. ]);
  329. }
  330. }
  331. /**
  332. * 提交任务到GPT-Image2 API
  333. *
  334. * @param MpGeneratePicTask $task
  335. * @return void
  336. */
  337. public function submitTaskToGptImage2Api(MpGeneratePicTask $task): void
  338. {
  339. try {
  340. // 验证环境变量配置
  341. $apiKey = env('GPT_IMAGE2_API_KEY');
  342. if (empty($apiKey)) {
  343. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  344. 'error_message' => 'GPT-Image2 API KEY未配置,请检查环境变量GPT_IMAGE2_API_KEY'
  345. ]);
  346. return;
  347. }
  348. // 更新状态为处理中
  349. $task->updateStatus(MpGeneratePicTask::STATUS_PROCESSING);
  350. // $api_url = 'https://token.ithinkai.cn/v1/images/generations'; // 备用地址1
  351. // $api_url = 'https://api.nonelinear.com/v1/images/generations'; // 备用地址2
  352. $api_url = 'https://ai-api.kkidc.com/v1/images/generations';
  353. $isEditMode = false;
  354. // 参考图片 - 如果有参考图则使用 /edits 接口
  355. if ($task->ref_img_url) {
  356. // $api_url = 'https://token.ithinkai.cn/v1/images/edits'; // 备用地址1
  357. // $api_url = 'https://api.nonelinear.com/v1/images/generations'; // 备用地址2
  358. $api_url = 'https://ai-api.kkidc.com/v1/images/edits';
  359. $isEditMode = true;
  360. }
  361. dLog('generate')->info('开始调用GPT-Image2 API', ['task_id' => $task->task_id, 'mode' => $isEditMode ? 'edit' : 'generation']);
  362. logDB('generate', 'info', '开始调用GPT-Image2 API', ['task_id' => $task->task_id, 'mode' => $isEditMode ? 'edit' : 'generation']);
  363. // 创建专用客户端,优化配置
  364. $gptClient = new Client([
  365. 'timeout' => 1800, // 超时
  366. 'connect_timeout' => 120, // 连接超时10秒
  367. 'http_errors' => false, // 不自动抛出HTTP错误
  368. 'verify' => false, // 禁用SSL验证(如果有证书问题)
  369. 'allow_redirects' => true,
  370. ]);
  371. // 根据是否有参考图选择不同的请求方式
  372. if ($isEditMode) {
  373. // 使用原生 cURL 处理 multipart/form-data 格式
  374. $imageUrls = is_array($task->ref_img_url) ? $task->ref_img_url : [$task->ref_img_url];
  375. // 限制最多4张参考图
  376. $imageUrls = array_slice($imageUrls, 0, 4);
  377. dLog('generate')->info('下载参考图片', ['image_urls' => $imageUrls, 'count' => count($imageUrls)]);
  378. // 下载所有参考图片并保存为临时文件
  379. $tempFiles = [];
  380. foreach ($imageUrls as $index => $imageUrl) {
  381. $imageContent = @file_get_contents($imageUrl);
  382. if ($imageContent === false) {
  383. dLog('generate')->warning('下载参考图片失败,跳过', ['image_url' => $imageUrl]);
  384. continue;
  385. }
  386. dLog('generate')->info('参考图片下载完成', ['index' => $index, 'size' => strlen($imageContent)]);
  387. // 保存为临时文件(cURL 需要真实文件路径)
  388. $extension = pathinfo(parse_url($imageUrl, PHP_URL_PATH), PATHINFO_EXTENSION) ?: 'png';
  389. $tempFile = tempnam(sys_get_temp_dir(), 'gpt_img_') . '.' . $extension;
  390. file_put_contents($tempFile, $imageContent);
  391. $tempFiles[] = [
  392. 'path' => $tempFile,
  393. 'extension' => $extension
  394. ];
  395. dLog('generate')->info('临时文件创建成功', ['file' => $tempFile, 'size' => filesize($tempFile)]);
  396. }
  397. if (empty($tempFiles)) {
  398. throw new \Exception('所有参考图片下载失败');
  399. }
  400. // 构建 POST 字段
  401. $postFields = [
  402. 'prompt' => $task->prompt,
  403. 'model' => 'gpt-image-2',
  404. 'n' => (string)$task->image_num, // 新接口要求字符串类型
  405. 'response_format' => 'b64_json', // 新接口必填参数
  406. ];
  407. // 尺寸参数(必填)
  408. if ($task->width && $task->height) {
  409. $postFields['size'] = $task->width . 'x' . $task->height;
  410. } else {
  411. $postFields['size'] = '1024x1024'; // 默认尺寸
  412. }
  413. // 添加参考图片文件(支持多图,新接口要求使用 image 字段名)
  414. foreach ($tempFiles as $index => $tempFile) {
  415. $fieldName = 'image'; // 新接口统一使用 image 字段名
  416. $postFields[$fieldName] = new \CURLFile(
  417. $tempFile['path'],
  418. 'image/' . $tempFile['extension'],
  419. 'image' . ($index + 1) . '.' . $tempFile['extension']
  420. );
  421. }
  422. dLog('generate')->info('GPT-Image2 API参数(编辑模式-cURL)', [
  423. 'api_url' => $api_url,
  424. 'prompt' => $task->prompt,
  425. 'size' => $postFields['size'],
  426. 'n' => $postFields['n'],
  427. 'response_format' => $postFields['response_format'],
  428. 'image_count' => count($tempFiles)
  429. ]);
  430. logDB('generate', 'info', 'GPT-Image2任务提交(编辑模式)', [
  431. 'task_id' => $task->task_id,
  432. 'api_url' => $api_url,
  433. 'image_urls' => $imageUrls,
  434. 'prompt' => $task->prompt
  435. ]);
  436. $startTime = microtime(true);
  437. // 使用原生 cURL
  438. $ch = curl_init();
  439. curl_setopt_array($ch, [
  440. CURLOPT_URL => $api_url,
  441. CURLOPT_RETURNTRANSFER => true,
  442. CURLOPT_POST => true,
  443. CURLOPT_POSTFIELDS => $postFields,
  444. CURLOPT_HTTPHEADER => [
  445. 'Authorization: Bearer ' . $apiKey,
  446. 'Accept: application/json',
  447. 'Expect:', // 移除 Expect: 100-continue,避免等待
  448. ],
  449. CURLOPT_TIMEOUT => 600,
  450. CURLOPT_CONNECTTIMEOUT => 30,
  451. CURLOPT_SSL_VERIFYPEER => false,
  452. CURLOPT_SSL_VERIFYHOST => false,
  453. CURLOPT_HTTP_VERSION => CURL_HTTP_VERSION_1_1, // 强制使用 HTTP/1.1
  454. ]);
  455. dLog('generate')->info('开始发送cURL请求');
  456. $responseBody = curl_exec($ch);
  457. $statusCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  458. $curlError = curl_error($ch);
  459. $curlErrno = curl_errno($ch);
  460. curl_close($ch);
  461. // 删除所有临时文件
  462. foreach ($tempFiles as $tempFile) {
  463. @unlink($tempFile['path']);
  464. }
  465. $elapsed = round(microtime(true) - $startTime, 2);
  466. dLog('generate')->info('cURL请求完成', ['elapsed' => $elapsed . 's', 'http_code' => $statusCode, 'curl_errno' => $curlErrno]);
  467. if ($curlErrno !== 0) {
  468. throw new \Exception('cURL错误: ' . $curlError . ' (错误码: ' . $curlErrno . ')');
  469. }
  470. if (empty($responseBody)) {
  471. throw new \Exception('API返回空响应,HTTP状态码: ' . $statusCode);
  472. }
  473. } else {
  474. // 使用 JSON 格式 (原有的文生图模式)
  475. $apiParams = [
  476. 'model' => 'gpt-image-2',
  477. 'prompt' => $task->prompt,
  478. 'n' => (int)$task->image_num,
  479. 'quality' => 'auto', // 新接口必填参数:图像质量等级
  480. 'response_format' => 'b64_json', // 新接口必填参数:返回格式
  481. ];
  482. // 尺寸参数
  483. if ($task->width && $task->height) {
  484. $apiParams['size'] = $task->width . 'x' . $task->height;
  485. } else {
  486. $apiParams['size'] = 'auto'; // 新接口必填参数
  487. }
  488. dLog('generate')->info('GPT-Image2 API参数', $apiParams);
  489. logDB('generate', 'info', 'GPT-Image2任务提交', ['task_id' => $task->task_id, 'api_url' => $api_url, 'params' => $apiParams]);
  490. $startTime = microtime(true);
  491. // 调用GPT-Image2 API
  492. $response = $gptClient->post($api_url, [
  493. 'headers' => [
  494. 'Authorization' => 'Bearer ' . $apiKey,
  495. 'Content-Type' => 'application/json',
  496. ],
  497. 'json' => $apiParams,
  498. ]);
  499. $statusCode = $response->getStatusCode();
  500. $responseBody = $response->getBody()->getContents();
  501. $elapsed = round(microtime(true) - $startTime, 2);
  502. dLog('generate')->info('Guzzle请求完成', ['elapsed' => $elapsed . 's', 'http_code' => $statusCode]);
  503. if (empty($responseBody)) {
  504. throw new \Exception('API返回空响应,HTTP状态码: ' . $statusCode);
  505. }
  506. }
  507. // 解析响应
  508. dLog('generate')->info('响应数据长度', ['length' => strlen($responseBody)]);
  509. $responseData = json_decode($responseBody, true);
  510. if (json_last_error() !== JSON_ERROR_NONE) {
  511. dLog('generate')->error('JSON解析失败', ['error' => json_last_error_msg(), 'body' => substr($responseBody, 0, 500)]);
  512. throw new \Exception('JSON解析失败: ' . json_last_error_msg());
  513. }
  514. // 移除 b64_json 字段用于日志记录,避免生成超大日志
  515. $logResponseData = $this->removeB64JsonFromResponse($responseData);
  516. dLog('generate')->info('GPT-Image2 API响应', $logResponseData);
  517. logDB('generate', 'info', 'GPT-Image2 API响应', ['task_id' => $task->task_id, 'response' => $logResponseData]);
  518. if (isset($responseData['error'])) {
  519. // API返回错误
  520. logDB('generate', 'error', 'GPT-Image2任务失败', ['task_id' => $task->task_id, 'error' => $responseData['error']['message'] ?? 'API Error']);
  521. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  522. 'error_message' => $responseData['error']['message'] ?? 'API Error',
  523. 'result_json' => $logResponseData
  524. ]);
  525. } else {
  526. // 任务提交成功,直接处理结果
  527. dLog('generate')->info('开始处理GPT-Image2响应结果');
  528. $this->processGptImage2ApiResponse($task, $responseData);
  529. }
  530. } catch (\Exception $e) {
  531. // 记录错误
  532. dLog('generate')->error('GPT-Image2 API请求失败: ' . $e->getMessage(), [
  533. 'task_id' => $task->task_id,
  534. 'trace' => $e->getTraceAsString()
  535. ]);
  536. logDB('generate', 'error', 'GPT-Image2任务异常', [
  537. 'task_id' => $task->task_id,
  538. 'error' => $e->getMessage(),
  539. 'file' => $e->getFile(),
  540. 'line' => $e->getLine()
  541. ]);
  542. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  543. 'error_message' => 'API请求失败: ' . $e->getMessage()
  544. ]);
  545. }
  546. }
  547. /**
  548. * 处理GPT-Image2 API响应结果
  549. *
  550. * @param MpGeneratePicTask $task
  551. * @param array $responseData
  552. * @return void
  553. */
  554. private function processGptImage2ApiResponse(MpGeneratePicTask $task, array $responseData): void
  555. {
  556. try {
  557. if (!isset($responseData['data']) || empty($responseData['data'])) {
  558. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  559. 'error_message' => '未返回图片数据',
  560. 'result_json' => $responseData
  561. ]);
  562. return;
  563. }
  564. $result_urls = [];
  565. foreach ($responseData['data'] as $index => $imageData) {
  566. // 优先处理 b64_json 格式(新接口返回格式)
  567. if (isset($imageData['b64_json']) && !empty($imageData['b64_json'])) {
  568. $base64Data = $imageData['b64_json'];
  569. $imageStream = base64_decode($base64Data);
  570. if ($imageStream === false) {
  571. dLog('generate')->error('Base64解码失败', ['task_id' => $task->task_id, 'index' => $index]);
  572. continue;
  573. }
  574. $pic_name = 'ai_generation_gpt_' . time() . '_' . uniqid() . '.png';
  575. // 压缩PNG图片流
  576. $compressedData = $this->compressImageStream($imageStream);
  577. $url = uploadStreamByTos('image', $compressedData, $pic_name);
  578. $result_urls[] = $url;
  579. // 记录优化后的提示词(如果有)
  580. if (isset($imageData['revised_prompt'])) {
  581. dLog('generate')->info('图片生成优化提示词', [
  582. 'task_id' => $task->task_id,
  583. 'index' => $index,
  584. 'revised_prompt' => $imageData['revised_prompt']
  585. ]);
  586. }
  587. } elseif (isset($imageData['url']) && !empty($imageData['url'])) {
  588. // 兼容旧的 url 格式(如果接口仍返回)
  589. $url = $imageData['url'];
  590. $pic_name = 'ai_generation_gpt_' . time() . '_' . uniqid();
  591. $pic_ext = getImgExtFromUrl($url);
  592. $pic_name = $pic_name . $pic_ext;
  593. if ($pic_ext === '.png') {
  594. $comporessed_data = compressRemoteImageUrlToSize($url);
  595. $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
  596. } else {
  597. $url = uploadStreamByTos('image', file_get_contents($url), $pic_name);
  598. }
  599. $result_urls[] = $url;
  600. }
  601. }
  602. if (empty($result_urls)) {
  603. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  604. 'error_message' => '所有图片生成失败',
  605. 'result_json' => $responseData
  606. ]);
  607. return;
  608. }
  609. // 移除 b64_json 字段以避免存储大量 Base64 数据
  610. $cleanedResponseData = $responseData;
  611. if (isset($cleanedResponseData['data'])) {
  612. foreach ($cleanedResponseData['data'] as &$imageData) {
  613. unset($imageData['b64_json']); // 移除 Base64 数据
  614. }
  615. unset($imageData); // 解除引用
  616. }
  617. // 图片生成成功统一处理(事务:状态更新 + 计费 + 分镜同步)
  618. $this->handleImageTaskSuccess($task, $result_urls, $cleanedResponseData);
  619. } catch (\Exception $e) {
  620. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  621. 'error_message' => '处理响应失败: ' . $e->getMessage()
  622. ]);
  623. }
  624. }
  625. /**
  626. * 提交任务到火山图片生成API
  627. *
  628. * @param MpGeneratePicTask $task
  629. * @return void
  630. */
  631. public function submitTaskToVolcApi(MpGeneratePicTask $task): void
  632. {
  633. try {
  634. // 验证环境变量配置
  635. $apiKey = env('VOLC_AI_API_KEY');
  636. if (empty($apiKey)) {
  637. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  638. 'error_message' => '火山引擎API KEY未配置,请检查环境变量VOLC_AI_API_KEY'
  639. ]);
  640. return;
  641. }
  642. // 更新状态为处理中
  643. $task->updateStatus(MpGeneratePicTask::STATUS_PROCESSING);
  644. // 构建火山图片生成API请求参数
  645. $apiParams = [
  646. 'model' => isset(BaseConst::MODELS_MAP[$task->model]) ? BaseConst::MODELS_MAP[$task->model] : $task->model,
  647. 'prompt' => $task->prompt,
  648. 'watermark' => false,
  649. ];
  650. // 尺寸参数
  651. if ($task->width && $task->height) {
  652. $apiParams['size'] = $task->width . 'x' . $task->height;
  653. }
  654. // 参考图片
  655. if ($task->ref_img_url) {
  656. if (is_array($task->ref_img_url)) {
  657. // $apiParams['image'] = count($task->ref_img_url) === 1 ? $task->ref_img_url[0] : $task->ref_img_url;
  658. $apiParams['image'] = $task->ref_img_url;
  659. } else {
  660. $apiParams['image'] = $task->ref_img_url;
  661. }
  662. }
  663. // 生成多图或者单图
  664. $imageNum = (int)$task->image_num;
  665. if ($imageNum === 1) {
  666. $apiParams['sequential_image_generation'] = 'disabled';
  667. }else if ($imageNum > 1) {
  668. // 组图模式下由模型决定实际生成张数,这里在提示词末尾明确要求生成对应数量,并将组图上限设置为目标张数
  669. $apiParams['prompt'] = rtrim((string)$task->prompt) . "\n生成{$imageNum}张图片";
  670. $apiParams['sequential_image_generation'] = 'auto';
  671. $apiParams['sequential_image_generation_options']['max_images'] = $imageNum;
  672. }
  673. dLog('generate')->info('火山图片生成API参数: ', $apiParams);
  674. logDB('generate', 'info', '火山图片生成任务提交', ['task_id' => $task->task_id, 'params' => $apiParams]);
  675. // 调用火山图片生成API
  676. $response = $this->httpClient->post('https://ark.cn-beijing.volces.com/api/v3/images/generations', [
  677. 'headers' => [
  678. 'Authorization' => 'Bearer ' . $apiKey,
  679. 'Content-Type' => 'application/json',
  680. ],
  681. 'json' => $apiParams,
  682. ]);
  683. $responseData = json_decode($response->getBody()->getContents(), true);
  684. dLog('generate')->info('火山图片生成API响应: ', $responseData);
  685. logDB('generate', 'info', '火山图片生成API响应', ['task_id' => $task->task_id, 'response' => $responseData]);
  686. if (isset($responseData['error'])) {
  687. // API返回错误
  688. logDB('generate', 'error', '火山图片生成任务失败', ['task_id' => $task->task_id, 'error' => $responseData['error']['message'] ?? 'API Error']);
  689. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  690. 'error_message' => $responseData['error']['message'] ?? 'API Error',
  691. 'result_json' => $responseData
  692. ]);
  693. } else {
  694. // 任务提交成功,直接处理结果
  695. $this->processVolcApiResponse($task, $responseData);
  696. }
  697. } catch (\Exception $e) {
  698. // 记录错误
  699. dLog('generate')->error('火山API请求失败: ' . $e->getMessage());
  700. logDB('generate', 'error', '火山图片生成任务异常', ['task_id' => $task->task_id, 'error' => $e->getMessage()]);
  701. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  702. 'error_message' => 'API请求失败: ' . $e->getMessage()
  703. ]);
  704. }
  705. }
  706. /**
  707. * 处理火山API响应结果
  708. *
  709. * @param MpGeneratePicTask $task
  710. * @param array $responseData
  711. * @return void
  712. */
  713. private function processVolcApiResponse(MpGeneratePicTask $task, array $responseData): void
  714. {
  715. try {
  716. if (!isset($responseData['data']) || empty($responseData['data'])) {
  717. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  718. 'error_message' => '未返回图片数据',
  719. 'result_json' => $responseData
  720. ]);
  721. return;
  722. }
  723. $result_urls = [];
  724. foreach ($responseData['data'] as $imageData) {
  725. if (isset($imageData['error'])) {
  726. // 单张图片生成失败
  727. dLog('generate')->warning('单张图片生成失败: ', $imageData['error']);
  728. continue;
  729. }
  730. if (isset($imageData['url'])) {
  731. $url = $imageData['url'];
  732. $pic_name = 'ai_generation_' . time() . '_' . uniqid();
  733. $pic_ext = getImgExtFromUrl($url);
  734. $pic_name = $pic_name . $pic_ext;
  735. if ($pic_ext === '.png') {
  736. $comporessed_data = compressRemoteImageUrlToSize($url);
  737. $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
  738. } else {
  739. // 将图片另存到tos
  740. $url = uploadStreamByTos('image', file_get_contents($url), $pic_name);
  741. }
  742. $result_urls[] = $url;
  743. }
  744. }
  745. if (empty($result_urls)) {
  746. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  747. 'error_message' => '所有图片生成失败',
  748. 'result_json' => $responseData
  749. ]);
  750. return;
  751. }
  752. // 图片生成成功统一处理(事务:状态更新 + 计费 + 分镜同步)
  753. $this->handleImageTaskSuccess($task, $result_urls, $responseData);
  754. } catch (\Exception $e) {
  755. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  756. 'error_message' => '处理响应失败: ' . $e->getMessage()
  757. ]);
  758. }
  759. }
  760. public function getTaskList($data)
  761. {
  762. $task_id = getProp($data, 'task_id');
  763. $status = getProp($data, 'status');
  764. $query = MpGeneratePicTask::select('*');
  765. if ($task_id) {
  766. $query->where('id', $task_id);
  767. }
  768. if ($status) {
  769. $query->where('status', $status);
  770. }
  771. return $query->orderBy('created_at', 'desc')->paginate();
  772. }
  773. /**
  774. * 查询任务状态
  775. *
  776. * @param MpGeneratePicTask $task
  777. * @return array
  778. */
  779. public function queryTaskStatus(MpGeneratePicTask $task): array
  780. {
  781. // 根据模型类型调用不同的查询方法
  782. if ($task->model === self::MODEL_JIMENG_4) {
  783. return $this->queryJimengTaskStatus($task);
  784. } elseif (in_array($task->model, BaseConst::NANO_BANANA_MODELS)) {
  785. return $this->queryNanoBananaTaskStatus($task);
  786. } else {
  787. // 火山API是同步返回结果,不需要查询
  788. return [
  789. 'status' => $task->status === MpGeneratePicTask::STATUS_SUCCESS ? 'success' : 'failed',
  790. 'error_message' => $task->error_message,
  791. 'result_url' => $task->result_url,
  792. 'result_json' => $task->result_json ?? []
  793. ];
  794. }
  795. }
  796. /**
  797. * 查询即梦AI任务状态
  798. *
  799. * @param MpGeneratePicTask $task
  800. * @return array
  801. */
  802. private function queryJimengTaskStatus(MpGeneratePicTask $task): array
  803. {
  804. try {
  805. // 获取API任务ID
  806. $apiTaskId = $task->task_id ?? null;
  807. if (!$apiTaskId) {
  808. return [
  809. 'status' => 'failed',
  810. 'error_message' => 'API任务ID不存在',
  811. 'result_url' => null
  812. ];
  813. }
  814. //构建查询参数
  815. $apiParams = [
  816. 'req_key' => 'jimeng_t2i_v40',
  817. 'task_id' => $apiTaskId,
  818. 'req_json' => '{"return_url":true}' // 返回URL格式
  819. ];
  820. //调用即梦AI 4.0 API查询任务状态
  821. $response = $this->volcEngineService->request(
  822. 'POST',
  823. 'visual.volcengineapi.com',
  824. '/', // 使用根路径
  825. [], // Query参数
  826. json_encode($apiParams),
  827. env('VOLC_AI_AK'),
  828. env('VOLC_AI_SK'),
  829. 'cv',
  830. 'cn-north-1',
  831. 'CVSync2AsyncGetResult',
  832. '2022-08-31',
  833. 'application/json'
  834. );
  835. $responseData = json_decode($response['body'], true);
  836. dLog('generate')->info('即梦AI 4.0 API查询任务状态响应: ', $responseData);
  837. logDB('generate', 'info', '即梦AI任务状态查询', ['task_id' => $apiTaskId, 'response' => $responseData]);
  838. if ($responseData['code'] !== 10000) {
  839. // API返回错误
  840. logDB('generate', 'error', '即梦AI任务状态查询失败', ['task_id' => $apiTaskId, 'error' => $responseData['message'] ?? 'API Error']);
  841. return [
  842. 'status' => 'failed',
  843. 'error_message' => $responseData['message'] ?? 'API Error',
  844. 'result_url' => null,
  845. 'result_json' => $responseData ?? []
  846. ];
  847. }
  848. // 解析API响应
  849. $result = $responseData['data'] ?? [];
  850. $taskStatus = $result['status'] ?? 'failed';
  851. $returnData = [
  852. 'status' => $taskStatus,
  853. 'error_message' => null,
  854. 'result_url' => null,
  855. 'result_json' => $responseData ?? []
  856. ];
  857. // 如果任务成功,获取结果URL
  858. if ($taskStatus === 'done' && isset($result['image_urls']) && count($result['image_urls']) > 0) {
  859. $result_urls = [];
  860. foreach ($result['image_urls'] as $url) {
  861. $pic_name = 'ai_generation_' . time() . '_' . uniqid();
  862. $pic_ext = getImgExtFromUrl($url);
  863. $pic_name = $pic_name . $pic_ext;
  864. if ($pic_ext === '.png') {
  865. $comporessed_data = compressRemoteImageUrlToSize($url);
  866. $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
  867. }else {
  868. // 将图片另存到tos
  869. $url = uploadStreamByTos('image', file_get_contents($url), $pic_name);
  870. }
  871. $result_urls[] = $url;
  872. }
  873. $returnData['result_url'] = $result_urls;
  874. $returnData['status'] = 'success';
  875. } elseif (in_array($taskStatus, ['not_found', 'expired'])) {
  876. $returnData['status'] = 'failed';
  877. $returnData['error_message'] = '任务未找到或已过期';
  878. } elseif ($taskStatus === 'failed') {
  879. $returnData['status'] = 'failed';
  880. $returnData['error_message'] = '任务执行失败';
  881. }
  882. // in_queue 或 generating状态保持原status
  883. return $returnData;
  884. } catch (\Exception $e) {
  885. return [
  886. 'status' => 'failed',
  887. 'error_message' => $e->getMessage(),
  888. 'result_url' => null
  889. ];
  890. }
  891. }
  892. /**
  893. * 提交任务到NanoBanana API
  894. *
  895. * @param MpGeneratePicTask $task
  896. * @return void
  897. */
  898. private function submitTaskToNanoBananaApi(MpGeneratePicTask $task): void
  899. {
  900. try {
  901. // 验证环境变量配置
  902. $apiKey = env('NANO_BANANA_API_KEY');
  903. if (empty($apiKey)) {
  904. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  905. 'error_message' => 'NanoBanana API KEY未配置,请检查环境变量NANO_BANANA_API_KEY'
  906. ]);
  907. return;
  908. }
  909. // 根据模型选择API地址
  910. switch ($task->model) {
  911. case self::MODEL_NANO_BANANA_2:
  912. $apiUrl = 'https://api.wuyinkeji.com/api/async/image_nanoBanana2';
  913. break;
  914. case self::MODEL_NANO_BANANA_PRO:
  915. $apiUrl = 'https://api.wuyinkeji.com/api/async/image_nanoBanana_pro';
  916. break;
  917. case self::MODEL_GPT_IMAGE_2:
  918. $apiUrl = 'https://api.wuyinkeji.com/api/async/image_gpt';
  919. break;
  920. default:
  921. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  922. 'error_message' => '不支持的模型类型: ' . $task->model
  923. ]);
  924. logDB('generate', 'error', '模型类型错误', ['task_id' => $task->task_id, 'model' => $task->model]);
  925. return;
  926. }
  927. // 构建API请求参数
  928. $apiParams = [
  929. 'key' => $apiKey,
  930. 'prompt' => $task->prompt,
  931. ];
  932. if ($task->model == self::MODEL_GPT_IMAGE_2) {
  933. // 该模型下,size需设置为9:16类似的数值,请根据如下比例反推
  934. $apiParams['size'] = $this->getRatioFromSize($task->width, $task->height);
  935. }else {
  936. // 尺寸参数(根据宽高判断)
  937. $area = $task->width * $task->height;
  938. if ($area >= 1024*1024 && $area < 2048*2048) {
  939. $apiParams['size'] = '1k';
  940. } elseif ($area >= 2048*2048) {
  941. $apiParams['size'] = '2k';
  942. } else {
  943. $apiParams['size'] = '1k'; // 默认1k
  944. }
  945. // 画面比例
  946. if ($task->width && $task->height) {
  947. $ratio = $task->width . ':' . $task->height;
  948. // 简化比例
  949. $gcd = $this->gcd($task->width, $task->height);
  950. $ratio = ($task->width / $gcd) . ':' . ($task->height / $gcd);
  951. $apiParams['aspectRatio'] = $ratio;
  952. }
  953. }
  954. // 参考图片
  955. if ($task->ref_img_url) {
  956. $apiParams['urls'] = is_array($task->ref_img_url) ? $task->ref_img_url : [$task->ref_img_url];
  957. }
  958. dLog('generate')->info('NanoBanana API参数: ', $apiParams);
  959. logDB('generate', 'info', 'NanoBanana任务提交', ['task_id' => $task->task_id, 'model' => $task->model, 'params' => $apiParams]);
  960. // 调用NanoBanana API
  961. $response = $this->httpClient->post($apiUrl, [
  962. 'query' => ['key' => $apiKey],
  963. 'json' => $apiParams,
  964. ]);
  965. $responseData = json_decode($response->getBody()->getContents(), true);
  966. dLog('generate')->info('NanoBanana API响应: ', $responseData);
  967. logDB('generate', 'info', 'NanoBanana API响应', ['task_id' => $task->task_id, 'response' => $responseData]);
  968. if ($responseData['code'] !== 200) {
  969. // API返回错误
  970. logDB('generate', 'error', 'NanoBanana任务提交失败', ['task_id' => $task->task_id, 'error' => $responseData['msg'] ?? 'API Error']);
  971. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  972. 'error_message' => $responseData['msg'] ?? 'API Error',
  973. 'extra_params' => $responseData
  974. ]);
  975. } else {
  976. // 任务提交成功,更新状态为处理中
  977. $updateData = [];
  978. // 存储API返回的任务ID
  979. if (isset($responseData['data']['id'])) {
  980. $updateData['task_id'] = $responseData['data']['id'];
  981. $updateData['extra_params'] = $responseData;
  982. }
  983. $task->updateStatus(MpGeneratePicTask::STATUS_PROCESSING, $updateData);
  984. }
  985. } catch (\Exception $e) {
  986. // 记录错误
  987. logDB('generate', 'error', 'NanoBanana任务提交异常', ['task_id' => $task->task_id, 'error' => $e->getMessage()]);
  988. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  989. 'error_message' => 'API请求失败: ' . $e->getMessage()
  990. ]);
  991. }
  992. }
  993. /**
  994. * 查询NanoBanana任务状态
  995. *
  996. * @param MpGeneratePicTask $task
  997. * @return array
  998. */
  999. private function queryNanoBananaTaskStatus(MpGeneratePicTask $task): array
  1000. {
  1001. try {
  1002. // 获取API任务ID
  1003. $apiTaskId = $task->task_id ?? null;
  1004. $apiKey = env('NANO_BANANA_API_KEY');
  1005. if (!$apiTaskId) {
  1006. return [
  1007. 'status' => 'failed',
  1008. 'error_message' => 'API任务ID不存在',
  1009. 'result_url' => null
  1010. ];
  1011. }
  1012. if (empty($apiKey)) {
  1013. return [
  1014. 'status' => 'failed',
  1015. 'error_message' => 'NanoBanana API KEY未配置',
  1016. 'result_url' => null
  1017. ];
  1018. }
  1019. // 调用统一查询接口
  1020. $response = $this->httpClient->get('https://api.wuyinkeji.com/api/async/detail', [
  1021. 'query' => [
  1022. 'key' => $apiKey,
  1023. 'id' => $apiTaskId
  1024. ]
  1025. ]);
  1026. $responseData = json_decode($response->getBody()->getContents(), true);
  1027. dLog('generate')->info('NanoBanana任务状态查询响应: ', $responseData);
  1028. logDB('generate', 'info', 'NanoBanana任务状态查询', ['task_id' => $apiTaskId, 'response' => $responseData]);
  1029. if ($responseData['code'] !== 200) {
  1030. // API返回错误
  1031. logDB('generate', 'error', 'NanoBanana任务状态查询失败', ['task_id' => $apiTaskId, 'error' => $responseData['msg'] ?? 'API Error']);
  1032. return [
  1033. 'status' => 'failed',
  1034. 'error_message' => $responseData['msg'] ?? 'API Error',
  1035. 'result_url' => null,
  1036. 'result_json' => $responseData
  1037. ];
  1038. }
  1039. // 解析API响应
  1040. $data = $responseData['data'] ?? [];
  1041. $taskStatus = $data['status'] ?? 0;
  1042. $returnData = [
  1043. 'status' => 'processing',
  1044. 'error_message' => null,
  1045. 'result_url' => null,
  1046. 'result_json' => $responseData
  1047. ];
  1048. // status: 2表示成功
  1049. if ($taskStatus === 2 && isset($data['result']) && !empty($data['result'])) {
  1050. $result_urls = [];
  1051. foreach ($data['result'] as $url) {
  1052. $pic_name = 'ai_generation_nanobanana_' . time() . '_' . uniqid();
  1053. $pic_ext = getImgExtFromUrl($url);
  1054. $pic_name = $pic_name . $pic_ext;
  1055. if ($pic_ext === '.png') {
  1056. $comporessed_data = compressRemoteImageUrlToSize($url);
  1057. $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
  1058. } else {
  1059. // 将图片另存到tos
  1060. $url = uploadStreamByTos('image', file_get_contents($url), $pic_name);
  1061. }
  1062. $result_urls[] = $url;
  1063. }
  1064. $returnData['result_url'] = $result_urls;
  1065. $returnData['status'] = 'success';
  1066. } elseif ($taskStatus === 3) {
  1067. // status: 3表示失败
  1068. $returnData['status'] = 'failed';
  1069. $returnData['error_message'] = $data['message'] ?? '任务执行失败';
  1070. }
  1071. // status: 0或1表示处理中,保持processing状态
  1072. return $returnData;
  1073. } catch (\Exception $e) {
  1074. return [
  1075. 'status' => 'failed',
  1076. 'error_message' => $e->getMessage(),
  1077. 'result_url' => null
  1078. ];
  1079. }
  1080. }
  1081. /**
  1082. * 计算最大公约数(用于简化比例)
  1083. *
  1084. * @param int $a
  1085. * @param int $b
  1086. * @return int
  1087. */
  1088. private function gcd(int $a, int $b): int
  1089. {
  1090. while ($b != 0) {
  1091. $temp = $b;
  1092. $b = $a % $b;
  1093. $a = $temp;
  1094. }
  1095. return $a;
  1096. }
  1097. /**
  1098. * 创建素材
  1099. *
  1100. * @param array $params
  1101. * @return array
  1102. */
  1103. public function createAsset(array $params): array
  1104. {
  1105. try {
  1106. // 验证必填参数
  1107. if (empty($params['url'])) {
  1108. return [
  1109. 'code' => 1,
  1110. 'msg' => '素材URL不能为空',
  1111. 'data' => null
  1112. ];
  1113. }
  1114. // 先获取表中是否已有该url(仅限智帧AI来源的素材)
  1115. $asset = DB::table('mp_assets')->where('url', $params['url'])->where('source_type', 'zhizhen')->first();
  1116. if ($asset) {
  1117. return [
  1118. 'code' => 0,
  1119. 'msg' => '创建成功',
  1120. 'data' => [
  1121. 'id' => $asset->id,
  1122. 'asset_id' => $asset->asset_id,
  1123. 'group_id' => $asset->group_id,
  1124. 'group_name' => $asset->group_name,
  1125. ]
  1126. ];
  1127. }
  1128. // 获取API配置
  1129. $apiKey = env('ZHIZHEN_AI_SK');
  1130. $apiUrl = 'https://model-relay.zzengine.net';
  1131. if (empty($apiKey)) {
  1132. return [
  1133. 'code' => 1002,
  1134. 'msg' => '智帧AI API KEY未配置,请检查环境变量ZHIZHEN_AI_SK',
  1135. 'data' => null
  1136. ];
  1137. }
  1138. // 构建请求参数
  1139. $requestData = [
  1140. 'url' => $params['url'],
  1141. 'asset_type' => $params['asset_type'] ?? MpAsset::TYPE_IMAGE,
  1142. ];
  1143. // 可选参数
  1144. if (!empty($params['name'])) {
  1145. $requestData['name'] = $params['name'];
  1146. }
  1147. if (!empty($params['group_id'])) {
  1148. $requestData['group_id'] = $params['group_id'];
  1149. }
  1150. dLog('generate')->info('智帧AI创建素材请求参数: ', $requestData);
  1151. // 调用智帧AI API
  1152. $response = $this->httpClient->post($apiUrl . '/api/v1/asset/create', [
  1153. 'headers' => [
  1154. 'X-Api-Key' => $apiKey,
  1155. 'Content-Type' => 'application/json',
  1156. ],
  1157. 'json' => $requestData,
  1158. ]);
  1159. $responseData = json_decode($response->getBody()->getContents(), true);
  1160. dLog('generate')->info('智帧AI创建素材响应: ', $responseData);
  1161. logDB('generate', 'info', '智帧AI创建素材', ['params' => $requestData, 'response' => $responseData]);
  1162. if ($responseData['code'] !== 0) {
  1163. return [
  1164. 'code' => $responseData['code'],
  1165. 'msg' => $responseData['msg'] ?? '创建素材失败',
  1166. 'data' => null
  1167. ];
  1168. }
  1169. // 保存素材记录到数据库
  1170. $assetData = $responseData['data'] ?? [];
  1171. $asset = MpAsset::create([
  1172. 'asset_id' => $assetData['asset_id'] ?? '',
  1173. 'group_id' => $assetData['group_id'] ?? null,
  1174. 'group_name' => $assetData['group_name'] ?? null,
  1175. 'name' => $params['name'] ?? null,
  1176. 'url' => $params['url'],
  1177. 'asset_type' => $params['asset_type'] ?? MpAsset::TYPE_IMAGE,
  1178. 'source_type' => 'zhizhen',
  1179. 'status' => MpAsset::STATUS_PROCESSING,
  1180. 'extra_params' => $responseData,
  1181. ]);
  1182. return [
  1183. 'code' => 0,
  1184. 'msg' => '创建成功',
  1185. 'data' => [
  1186. 'id' => $asset->id,
  1187. 'asset_id' => $asset->asset_id,
  1188. 'group_id' => $asset->group_id,
  1189. 'group_name' => $asset->group_name,
  1190. ]
  1191. ];
  1192. } catch (\Exception $e) {
  1193. dLog('generate')->error('创建素材异常: ' . $e->getMessage());
  1194. logDB('generate', 'error', '创建素材异常', ['params' => $params, 'error' => $e->getMessage()]);
  1195. return [
  1196. 'code' => 20003,
  1197. 'msg' => '创建素材失败: ' . $e->getMessage(),
  1198. 'data' => null
  1199. ];
  1200. }
  1201. }
  1202. /**
  1203. * 查询素材
  1204. *
  1205. * @param string $assetId
  1206. * @return array
  1207. */
  1208. public function getAsset(string $assetId): array
  1209. {
  1210. try {
  1211. // 验证参数
  1212. if (empty($assetId)) {
  1213. return [
  1214. 'code' => 20003,
  1215. 'msg' => '素材ID不能为空',
  1216. 'data' => null
  1217. ];
  1218. }
  1219. // 先获取表中是否已有该素材(仅限智帧AI来源的素材)
  1220. $assetInfo = DB::table('mp_assets')->where('asset_id', $assetId)->where('source_type', 'zhizhen')->first();
  1221. if ($assetInfo) {
  1222. $assetInfo = (array)$assetInfo;
  1223. $status = getProp($assetInfo, 'status');
  1224. if (in_array($status, ['Active', 'Failed'])) {
  1225. $assetInfo['id'] = $assetInfo['asset_id'];
  1226. return [
  1227. 'code' => 0,
  1228. 'msg' => '查询成功',
  1229. 'data' => [
  1230. 'asset' => $assetInfo
  1231. ]
  1232. ];
  1233. }
  1234. }
  1235. // 获取API配置
  1236. $apiKey = env('ZHIZHEN_AI_SK');
  1237. $apiUrl = 'https://model-relay.zzengine.net';
  1238. if (empty($apiKey)) {
  1239. return [
  1240. 'code' => 1002,
  1241. 'msg' => '智帧AI API KEY未配置,请检查环境变量ZHIZHEN_AI_SK',
  1242. 'data' => null
  1243. ];
  1244. }
  1245. dLog('generate')->info('智帧AI查询素材: ' . $assetId);
  1246. // 调用智帧AI API
  1247. $response = $this->httpClient->post($apiUrl . '/api/v1/asset/get', [
  1248. 'headers' => [
  1249. 'X-Api-Key' => $apiKey,
  1250. 'Content-Type' => 'application/json',
  1251. ],
  1252. 'json' => [
  1253. 'asset_id' => $assetId,
  1254. ],
  1255. ]);
  1256. $responseData = json_decode($response->getBody()->getContents(), true);
  1257. dLog('generate')->info('智帧AI查询素材响应: ', $responseData);
  1258. logDB('generate', 'info', '智帧AI查询素材', ['asset_id' => $assetId, 'response' => $responseData]);
  1259. if ($responseData['code'] !== 0) {
  1260. return [
  1261. 'code' => $responseData['code'],
  1262. 'msg' => $responseData['msg'] ?? '查询素材失败',
  1263. 'data' => null
  1264. ];
  1265. }
  1266. $assetInfo = $responseData['data']['asset'] ?? [];
  1267. // 更新本地数据库记录
  1268. $localAsset = MpAsset::where('asset_id', $assetId)->first();
  1269. if ($localAsset) {
  1270. $localAsset->update([
  1271. 'status' => $assetInfo['status'] ?? MpAsset::STATUS_PROCESSING,
  1272. 'cover_url' => $assetInfo['url'] ?? null,
  1273. 'extra_params' => $responseData,
  1274. ]);
  1275. }
  1276. return [
  1277. 'code' => 0,
  1278. 'msg' => '查询成功',
  1279. 'data' => [
  1280. 'asset' => $assetInfo
  1281. ]
  1282. ];
  1283. } catch (\Exception $e) {
  1284. dLog('generate')->error('查询素材异常: ' . $e->getMessage());
  1285. logDB('generate', 'error', '查询素材异常', ['asset_id' => $assetId, 'error' => $e->getMessage()]);
  1286. return [
  1287. 'code' => 20003,
  1288. 'msg' => '查询素材失败: ' . $e->getMessage(),
  1289. 'data' => null
  1290. ];
  1291. }
  1292. }
  1293. /**
  1294. * 更新所有待处理任务的状态
  1295. *
  1296. * @return void
  1297. */
  1298. public function updatePendingTasks(): void
  1299. {
  1300. dLog('generate')->info('更新图片状态ing');
  1301. // 1. 处理火山API的pending任务(批量提交到队列)
  1302. $volcPendingTasks = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PENDING)
  1303. ->whereIn('model', BaseConst::VOLC_PIC_MODELS)
  1304. ->orderBy('created_at', 'asc')
  1305. ->limit(self::DISPATCH_CANDIDATE_LIMIT) // 候选窗口需大于每轮派发上限,避免已入队的任务挡住后面的任务
  1306. ->get();
  1307. $volcDispatched = 0;
  1308. foreach ($volcPendingTasks as $task) {
  1309. if ($volcDispatched >= self::DISPATCH_BATCH_LIMIT) {
  1310. break;
  1311. }
  1312. // 同一任务已在队列中或正在执行时跳过,避免重复入队
  1313. if (!$this->dispatchTaskOnce(self::DISPATCH_GROUP_VOLC, (int)$task->id, \App\Jobs\ProcessVolcImageGenerationJob::class)) {
  1314. continue;
  1315. }
  1316. dLog('generate')->info('开始分发火山API任务到队列: ' . $task->task_id);
  1317. $volcDispatched++;
  1318. }
  1319. // 2. 处理GPT-Image2的pending任务(批量提交到队列)
  1320. $gptImage2PendingTasks = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PENDING)
  1321. ->whereIn('model', BaseConst::GPT_IMAGE2_MODELS)
  1322. ->orderBy('created_at', 'asc')
  1323. ->limit(self::DISPATCH_CANDIDATE_LIMIT) // 候选窗口需大于每轮派发上限,避免已入队的任务挡住后面的任务
  1324. ->get();
  1325. $gptImage2Dispatched = 0;
  1326. foreach ($gptImage2PendingTasks as $task) {
  1327. if ($gptImage2Dispatched >= self::DISPATCH_BATCH_LIMIT) {
  1328. break;
  1329. }
  1330. // 同一任务已在队列中或正在执行时跳过,避免重复入队
  1331. if (!$this->dispatchTaskOnce(self::DISPATCH_GROUP_GPT_IMAGE2, (int)$task->id, \App\Jobs\ProcessGptImage2GenerationJob::class)) {
  1332. continue;
  1333. }
  1334. dLog('generate')->info('开始分发GPT-Image2任务到队列: ' . $task->task_id);
  1335. $gptImage2Dispatched++;
  1336. }
  1337. // 3. 获取所有处理中的任务(即梦AI和NanoBanana系列需要查询状态)
  1338. $processingTasks = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PROCESSING)
  1339. ->whereIn('model', array_merge([self::MODEL_JIMENG_4], BaseConst::NANO_BANANA_MODELS))
  1340. ->get();
  1341. foreach ($processingTasks as $task) {
  1342. $statusInfo = $this->queryTaskStatus($task);
  1343. if ($statusInfo['status'] === 'success') {
  1344. logDB('generate', 'info', '即梦AI图片生成任务成功', ['task_id' => $task->task_id, 'result_url' => $statusInfo['result_url']]);
  1345. // 图片生成成功统一处理(事务:状态更新 + 计费 + 分镜同步)
  1346. $this->handleImageTaskSuccess($task, $statusInfo['result_url'], $statusInfo['result_json'] ?? []);
  1347. } elseif ($statusInfo['status'] === 'failed') {
  1348. logDB('generate', 'error', '即梦AI图片生成任务失败', ['task_id' => $task->task_id, 'error' => $statusInfo['error_message']]);
  1349. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  1350. 'error_message' => $statusInfo['error_message'],
  1351. 'result_json' => $statusInfo['result_json'] ?? []
  1352. ]);
  1353. if (getProp($task, 'alias_segment_id')) {
  1354. DB::table('mp_episode_segments')->where('segment_id', getProp($task, 'alias_segment_id'))->update([
  1355. 'pic_task_status' => '失败',
  1356. ]);
  1357. }
  1358. }
  1359. // 如果仍然是处理中状态,不做任何操作
  1360. //处理:如果任务处理超过12小时,标记为失败
  1361. $processingTime = now()->diffInHours($task->created_at);
  1362. if ($processingTime > 12) {
  1363. logDB('generate', 'warning', '即梦AI图片生成任务超时', ['task_id' => $task->task_id, 'processing_hours' => $processingTime]);
  1364. $task->updateStatus(MpGeneratePicTask::STATUS_FAILED, [
  1365. 'error_message' => '任务处理超时(超过12小时)'
  1366. ]);
  1367. }
  1368. }
  1369. // 4. 检查是否有任务完成,如果有则处理下一个排队任务(仅即梦AI)
  1370. $this->processNextQueuedTask();
  1371. }
  1372. /**
  1373. * 派发单个任务到队列(带派发去重)
  1374. *
  1375. * 同一任务在「已入队等待执行」或「正在执行」期间不会重复入队,
  1376. * 避免每轮扫描(约3秒一次)把同一条任务重复推送进队列。
  1377. *
  1378. * @param string $group 派发分组,用于区分同一任务ID在不同业务下的锁
  1379. * @param int $taskId 图片任务ID
  1380. * @param string $jobClass 队列任务类
  1381. * @return bool 本次是否成功入队;false 表示已有同任务的 job 在队列中或执行中
  1382. */
  1383. private function dispatchTaskOnce(string $group, int $taskId, string $jobClass): bool
  1384. {
  1385. if (!self::acquireTaskDispatchLock($group, $taskId)) {
  1386. return false;
  1387. }
  1388. try {
  1389. $jobClass::dispatch($taskId)->onConnection('redis');
  1390. } catch (\Throwable $e) {
  1391. // 入队失败必须释放锁,否则该任务在锁过期前无法再次派发
  1392. self::releaseTaskDispatchLock($group, $taskId);
  1393. throw $e;
  1394. }
  1395. return true;
  1396. }
  1397. /**
  1398. * 获取任务派发锁
  1399. *
  1400. * 在队列所在的 Redis 连接上执行原生 `SET key value EX ttl NX` 原子命令,
  1401. * 不依赖 Lua 脚本(部分 Redis 会禁用 EVAL)也不需要额外的缓存 store 配置。
  1402. *
  1403. * @param string $group
  1404. * @param int $taskId
  1405. * @return bool true 表示获取成功(可以入队)
  1406. */
  1407. public static function acquireTaskDispatchLock(string $group, int $taskId): bool
  1408. {
  1409. try {
  1410. $result = Redis::connection(self::DISPATCH_LOCK_CONNECTION)->set(
  1411. self::buildTaskDispatchLockKey($group, $taskId),
  1412. 1,
  1413. 'EX',
  1414. self::DISPATCH_LOCK_TTL,
  1415. 'NX'
  1416. );
  1417. // NX 未命中时:predis 返回 null,phpredis 返回 false
  1418. return $result !== null && $result !== false;
  1419. } catch (\Throwable $e) {
  1420. // 锁不可用时降级为不去重(与旧逻辑一致),避免任务因此完全无法派发
  1421. dLog('generate')->warning('获取图片任务派发锁失败,本次不拦截重复派发', [
  1422. 'group' => $group,
  1423. 'task_id' => $taskId,
  1424. 'error' => $e->getMessage(),
  1425. ]);
  1426. return true;
  1427. }
  1428. }
  1429. /**
  1430. * 释放任务派发锁
  1431. *
  1432. * 在任务执行结束、任务已由其它流程处理、任务最终失败时调用;
  1433. * 未释放的锁会在 DISPATCH_LOCK_TTL 后自动过期,保证任务仍能被重新派发。
  1434. *
  1435. * @param string $group
  1436. * @param int $taskId
  1437. * @return void
  1438. */
  1439. public static function releaseTaskDispatchLock(string $group, int $taskId): void
  1440. {
  1441. try {
  1442. Redis::connection(self::DISPATCH_LOCK_CONNECTION)->del(
  1443. self::buildTaskDispatchLockKey($group, $taskId)
  1444. );
  1445. } catch (\Throwable $e) {
  1446. // 释放失败不影响主流程,锁会由 TTL 自动过期
  1447. dLog('generate')->warning('释放图片任务派发锁失败', [
  1448. 'group' => $group,
  1449. 'task_id' => $taskId,
  1450. 'error' => $e->getMessage(),
  1451. ]);
  1452. }
  1453. }
  1454. /**
  1455. * 构建任务派发锁 key
  1456. *
  1457. * @param string $group
  1458. * @param int $taskId
  1459. * @return string
  1460. */
  1461. public static function buildTaskDispatchLockKey(string $group, int $taskId): string
  1462. {
  1463. return self::DISPATCH_LOCK_PREFIX . $group . ':' . $taskId;
  1464. }
  1465. /**
  1466. * 处理下一个排队的任务
  1467. *
  1468. * @return void
  1469. */
  1470. public function processNextQueuedTask(): void
  1471. {
  1472. // 检查是否还有正在处理的任务(仅即梦AI)
  1473. if ($this->hasProcessingTask()) {
  1474. return; // 如果还有处理中的任务,不处理新任务
  1475. }
  1476. // 获取最早的待处理任务(仅即梦AI)
  1477. $nextTask = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PENDING)
  1478. ->where('model', self::MODEL_JIMENG_4)
  1479. ->orderBy('created_at', 'asc')
  1480. ->first();
  1481. if ($nextTask) {
  1482. dLog('generate')->info('开始处理排队任务: ' . $nextTask->task_id);
  1483. logDB('generate', 'info', '开始处理即梦AI排队任务', ['task_id' => $nextTask->task_id]);
  1484. $this->submitTaskToJimengApi($nextTask);
  1485. }
  1486. }
  1487. /**
  1488. * 获取任务队列状态
  1489. *
  1490. * @return array
  1491. */
  1492. public function getQueueStatus(): array
  1493. {
  1494. $processingCount = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PROCESSING)->count();
  1495. $pendingCount = MpGeneratePicTask::where('status', MpGeneratePicTask::STATUS_PENDING)->count();
  1496. return [
  1497. 'processing_count' => $processingCount,
  1498. 'pending_count' => $pendingCount,
  1499. 'can_submit_new_task' => $processingCount === 0
  1500. ];
  1501. }
  1502. /**
  1503. * 处理智帧AI的参考图片转素材
  1504. *
  1505. * @param array $referenceImages 参考图片URL数组
  1506. * @param string &$fullPrompt 完整提示词(引用传递,用于删除失败图片的标记)
  1507. * @return array 返回素材ID数组
  1508. */
  1509. public function processReferenceImagesToAssets(array $referenceImages, string &$fullPrompt): array
  1510. {
  1511. return $this->processReferenceAssetsToZhizhenAssets($referenceImages, MpAsset::TYPE_IMAGE, '图片', '图片', $fullPrompt);
  1512. }
  1513. /**
  1514. * 处理智帧AI的参考视频转素材
  1515. *
  1516. * @param array $referenceVideos 参考视频URL数组
  1517. * @param string &$fullPrompt 完整提示词(引用传递,用于删除失败视频的标记)
  1518. * @return array 返回素材ID数组
  1519. */
  1520. public function processReferenceVideosToAssets(array $referenceVideos, string &$fullPrompt): array
  1521. {
  1522. return $this->processReferenceAssetsToZhizhenAssets($referenceVideos, MpAsset::TYPE_VIDEO, '视频', '视频', $fullPrompt);
  1523. }
  1524. /**
  1525. * 处理智帧AI的参考音频转素材
  1526. *
  1527. * @param array $referenceAudios 参考音频URL数组
  1528. * @param string &$fullPrompt 完整提示词(引用传递,用于删除失败音频的标记)
  1529. * @return array 返回素材ID数组
  1530. */
  1531. public function processReferenceAudiosToAssets(array $referenceAudios, string &$fullPrompt): array
  1532. {
  1533. return $this->processReferenceAssetsToZhizhenAssets($referenceAudios, MpAsset::TYPE_AUDIO, '音频', '音频', $fullPrompt);
  1534. }
  1535. /**
  1536. * 处理智帧AI的参考素材转素材(图片/视频/音频通用方法)
  1537. *
  1538. * @param array $urls 参考素材URL数组
  1539. * @param string $assetType 素材类型(Image/Video/Audio)
  1540. * @param string $label 素材中文名(图片/视频/音频),用于日志与素材命名
  1541. * @param string $tagName 提示词标记名(图片/视频/音频),用于 <{tagName}N> 标记的清理
  1542. * @param string &$fullPrompt 完整提示词(引用传递,用于删除失败素材的标记)
  1543. * @return array 返回素材ID数组(asset://格式)
  1544. */
  1545. private function processReferenceAssetsToZhizhenAssets(array $urls, string $assetType, string $label, string $tagName, string &$fullPrompt): array
  1546. {
  1547. $assetIds = [];
  1548. $failedIndexes = []; // 记录失败的素材序号
  1549. foreach ($urls as $index => $url) {
  1550. $mediaIndex = $index + 1; // 素材序号从1开始
  1551. try {
  1552. dLog('generate')->info("开始处理参考{$label} {$mediaIndex}: {$url}");
  1553. // 1. 创建素材
  1554. $createResult = $this->createAsset([
  1555. 'url' => $url,
  1556. 'asset_type' => $assetType,
  1557. 'name' => "参考{$label}_{$mediaIndex}_" . time(),
  1558. ]);
  1559. if ($createResult['code'] !== 0) {
  1560. dLog('generate')->warning("{$label} {$mediaIndex} 创建素材失败: " . $createResult['msg']);
  1561. logDB('generate', 'warning', '创建素材失败', [
  1562. 'url' => $url,
  1563. 'asset_type' => $assetType,
  1564. 'index' => $mediaIndex,
  1565. 'error' => $createResult['msg']
  1566. ]);
  1567. $failedIndexes[] = $mediaIndex;
  1568. continue;
  1569. }
  1570. $assetId = $createResult['data']['asset_id'];
  1571. dLog('generate')->info("{$label} {$mediaIndex} 素材创建成功,asset_id: {$assetId},开始轮询查询状态");
  1572. // 2. 轮询查询素材状态(每3秒查询一次,最多查询100次即5分钟)
  1573. $maxAttempts = 100;
  1574. $attempt = 0;
  1575. $assetStatus = null;
  1576. while ($attempt < $maxAttempts) {
  1577. sleep(3); // 等待3秒
  1578. $attempt++;
  1579. $getResult = $this->getAsset($assetId);
  1580. if ($getResult['code'] !== 0) {
  1581. dLog('generate')->warning("{$label} {$mediaIndex} 查询素材状态失败: " . $getResult['msg']);
  1582. continue;
  1583. }
  1584. $assetStatus = $getResult['data']['asset']['status'] ?? null;
  1585. dLog('generate')->info("{$label} {$mediaIndex} 素材状态查询第 {$attempt} 次,状态: {$assetStatus}");
  1586. if ($assetStatus === 'Active') {
  1587. // 素材可用
  1588. $assetIds[] = $assetId;
  1589. dLog('generate')->info("{$label} {$mediaIndex} 素材处理成功,asset_id: {$assetId}");
  1590. logDB('generate', 'info', '素材转换成功', [
  1591. 'url' => $url,
  1592. 'asset_type' => $assetType,
  1593. 'index' => $mediaIndex,
  1594. 'asset_id' => $assetId
  1595. ]);
  1596. break;
  1597. } elseif ($assetStatus === 'Failed') {
  1598. // 素材处理失败
  1599. dLog('generate')->warning("{$label} {$mediaIndex} 素材处理失败");
  1600. logDB('generate', 'warning', '素材处理失败', [
  1601. 'url' => $url,
  1602. 'asset_type' => $assetType,
  1603. 'index' => $mediaIndex,
  1604. 'asset_id' => $assetId
  1605. ]);
  1606. $failedIndexes[] = $mediaIndex;
  1607. break;
  1608. }
  1609. // 其他状态(Processing等)继续等待
  1610. }
  1611. // 如果超时还未成功
  1612. if ($attempt >= $maxAttempts && $assetStatus !== 'Active') {
  1613. dLog('generate')->warning("{$label} {$mediaIndex} 素材处理超时");
  1614. logDB('generate', 'warning', '素材处理超时', [
  1615. 'url' => $url,
  1616. 'asset_type' => $assetType,
  1617. 'index' => $mediaIndex,
  1618. 'asset_id' => $assetId,
  1619. 'last_status' => $assetStatus
  1620. ]);
  1621. $failedIndexes[] = $mediaIndex;
  1622. }
  1623. } catch (\Exception $e) {
  1624. dLog('generate')->error("{$label} {$mediaIndex} 处理异常: " . $e->getMessage());
  1625. logDB('generate', 'error', '素材转换异常', [
  1626. 'url' => $url,
  1627. 'asset_type' => $assetType,
  1628. 'index' => $mediaIndex,
  1629. 'error' => $e->getMessage()
  1630. ]);
  1631. $failedIndexes[] = $mediaIndex;
  1632. }
  1633. }
  1634. // 处理失败的素材:从 fullPrompt 中删除对应的标记并调整序号
  1635. if (!empty($failedIndexes)) {
  1636. dLog('generate')->info("开始处理失败{$label}的提示词调整,失败序号: " . implode(',', $failedIndexes));
  1637. // 按序号从大到小排序,避免删除时影响后续序号
  1638. rsort($failedIndexes);
  1639. foreach ($failedIndexes as $failedIndex) {
  1640. // 删除失败素材的标记,例如 <图片1>、<视频2>、<音频3> 等
  1641. $pattern = "/<{$tagName}{$failedIndex}>/u";
  1642. $fullPrompt = preg_replace($pattern, '', $fullPrompt);
  1643. dLog('generate')->info("已从提示词中删除 <{$tagName}{$failedIndex}>");
  1644. }
  1645. // 重新调整剩余素材的序号
  1646. // 例如:如果素材2失败,素材3、素材4需要改为素材2、素材3
  1647. $needAdjust = false;
  1648. foreach ($failedIndexes as $failedIndex) {
  1649. $needAdjust = true;
  1650. // 将所有大于失败序号的素材标记序号减1
  1651. for ($i = $failedIndex + 1; $i <= count($urls); $i++) {
  1652. $oldPattern = "/<{$tagName}{$i}>/u";
  1653. $newTag = "<{$tagName}" . ($i - 1) . ">";
  1654. $fullPrompt = preg_replace($oldPattern, $newTag, $fullPrompt);
  1655. }
  1656. }
  1657. if ($needAdjust) {
  1658. dLog('generate')->info("提示词{$label}序号调整完成");
  1659. logDB('generate', 'info', '提示词调整', [
  1660. 'failed_indexes' => $failedIndexes,
  1661. 'adjusted_prompt' => $fullPrompt
  1662. ]);
  1663. }
  1664. }
  1665. dLog('generate')->info("参考{$label}转素材完成,成功数量: " . count($assetIds) . ',失败数量: ' . count($failedIndexes));
  1666. // 格式化为可直接在API接口中使用的数组
  1667. foreach ($assetIds as &$assetId) {
  1668. $assetId = "asset://{$assetId}";
  1669. }
  1670. unset($assetId);
  1671. return $assetIds;
  1672. }
  1673. /**
  1674. * 根据宽高反推图片比例
  1675. * @param int $width 宽度
  1676. * @param int $height 高度
  1677. * @return string 比例字符串,如 "9:16",匹配不到则返回 "auto"
  1678. */
  1679. private function getRatioFromSize($width, $height)
  1680. {
  1681. if (!$width || !$height) {
  1682. return 'auto';
  1683. }
  1684. // 遍历所有预定义的比例,查找匹配的
  1685. foreach (BaseConst::IMAGE_RATIOS as $ratio => $size) {
  1686. if ($size['width'] == $width && $size['height'] == $height) {
  1687. return $ratio;
  1688. }
  1689. }
  1690. // 没有精确匹配,尝试通过计算比例匹配
  1691. $gcd = $this->gcd($width, $height);
  1692. $simplifiedRatio = ($width / $gcd) . ':' . ($height / $gcd);
  1693. // 检查简化后的比例是否在预定义列表中
  1694. if (isset(BaseConst::IMAGE_RATIOS[$simplifiedRatio])) {
  1695. return $simplifiedRatio;
  1696. }
  1697. return 'auto';
  1698. }
  1699. /**
  1700. * 压缩图片流(针对PNG格式)
  1701. *
  1702. * @param string $imageStream 图片二进制数据
  1703. * @param int $maxBytes 最大字节数(默认3MB)
  1704. * @return string|null 压缩后的图片二进制数据,失败时返回原数据
  1705. */
  1706. private function compressImageStream(string $imageStream, int $maxBytes = 3 * 1024 * 1024): ?string
  1707. {
  1708. try {
  1709. // 检查图片类型
  1710. $imgInfo = @getimagesizefromstring($imageStream);
  1711. if (!$imgInfo) {
  1712. dLog('generate')->warning('无法识别图片类型,返回原始数据');
  1713. return $imageStream;
  1714. }
  1715. $mime = $imgInfo['mime'] ?? '';
  1716. // 只处理PNG,JPEG不做有损压缩
  1717. if ($mime === 'image/jpeg') {
  1718. return $imageStream;
  1719. }
  1720. // 如果已经小于最大限制,直接返回
  1721. if (strlen($imageStream) <= $maxBytes) {
  1722. return $imageStream;
  1723. }
  1724. // 载入图像对象
  1725. $srcImg = @imagecreatefromstring($imageStream);
  1726. if (!$srcImg) {
  1727. dLog('generate')->warning('无法创建图像资源,返回原始数据');
  1728. return $imageStream;
  1729. }
  1730. $origW = imagesx($srcImg);
  1731. $origH = imagesy($srcImg);
  1732. // 逐步降低质量直到满足大小要求
  1733. $quality = 9; // PNG压缩等级 0-9
  1734. $scaleFactor = 1.0;
  1735. $maxAttempts = 10;
  1736. $attempt = 0;
  1737. $compressedData = null;
  1738. while ($attempt < $maxAttempts) {
  1739. $attempt++;
  1740. // 如果需要缩放
  1741. if ($scaleFactor < 1.0) {
  1742. $newW = (int)($origW * $scaleFactor);
  1743. $newH = (int)($origH * $scaleFactor);
  1744. $resizedImg = imagecreatetruecolor($newW, $newH);
  1745. // 保持透明度
  1746. imagealphablending($resizedImg, false);
  1747. imagesavealpha($resizedImg, true);
  1748. imagecopyresampled($resizedImg, $srcImg, 0, 0, 0, 0, $newW, $newH, $origW, $origH);
  1749. $targetImg = $resizedImg;
  1750. } else {
  1751. $targetImg = $srcImg;
  1752. }
  1753. // 输出到内存
  1754. ob_start();
  1755. imagepng($targetImg, null, $quality);
  1756. $compressedData = ob_get_clean();
  1757. if ($scaleFactor < 1.0) {
  1758. imagedestroy($targetImg);
  1759. }
  1760. // 检查大小
  1761. $size = strlen($compressedData);
  1762. dLog('generate')->info('图片压缩尝试', [
  1763. 'attempt' => $attempt,
  1764. 'quality' => $quality,
  1765. 'scale' => $scaleFactor,
  1766. 'size' => $size,
  1767. 'target' => $maxBytes
  1768. ]);
  1769. if ($size <= $maxBytes) {
  1770. break;
  1771. }
  1772. // 调整参数
  1773. if ($quality > 0) {
  1774. $quality--;
  1775. } else {
  1776. $scaleFactor *= 0.9; // 每次缩小10%
  1777. }
  1778. }
  1779. imagedestroy($srcImg);
  1780. return $compressedData ?: $imageStream;
  1781. } catch (\Exception $e) {
  1782. dLog('generate')->error('图片压缩失败', ['error' => $e->getMessage()]);
  1783. return $imageStream;
  1784. }
  1785. }
  1786. /**
  1787. * 从响应数据中移除 b64_json 字段
  1788. * 避免在日志和数据库中存储大量 Base64 数据
  1789. *
  1790. * @param array $responseData
  1791. * @return array
  1792. */
  1793. private function removeB64JsonFromResponse(array $responseData): array
  1794. {
  1795. $cleanedData = $responseData;
  1796. if (isset($cleanedData['data']) && is_array($cleanedData['data'])) {
  1797. foreach ($cleanedData['data'] as &$imageData) {
  1798. if (isset($imageData['b64_json'])) {
  1799. // 移除 b64_json 字段,可选择性保留数据长度信息
  1800. $b64Length = strlen($imageData['b64_json']);
  1801. unset($imageData['b64_json']);
  1802. $imageData['b64_json_removed'] = true;
  1803. $imageData['b64_json_length'] = $b64Length;
  1804. }
  1805. }
  1806. unset($imageData); // 解除引用
  1807. }
  1808. return $cleanedData;
  1809. }
  1810. }