AIImageGenerationService.php 84 KB

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