mes-ai-server-api/application/job/TextToTextJob.php

<?php
namespace app\job;
use app\service\AIGatewayService;
use app\service\ImageService;
use think\Db;
use think\queue\Job;
use think\Queue;
/**
 * 文生文
 */
class TextToTextJob
{
    /**
     * 队列任务入口
     */
    public function fire(Job $job, $data)
    {
        if (isset($data['status_val']) && $data['status_val'] == '文生文') {
            $taskId = $data['task_id'];
            $redis = getTaskRedis();
            $existing = $redis->get("txt_to_txt_task:{$taskId}");
            if ($existing) {
                $info = json_decode($existing, true);
                if (isset($info['status']) && ImageService::isTaskCompleted($info['status'])) {
                    echo "任务 {$taskId} 已完成，跳过重复执行\n";
                    $job->delete();
                    return;
                }
            }
            try {
                echo " 开始处理文生文".date('Y-m-d H:i:s')."\n";

                $result = $this->get_txt_to_txt($data);
                if (is_array($result) && isset($result['code']) && $result['code'] !== 0) {
                    throw new \Exception($result['msg'] ?? '文生文失败');
                }

                echo "🎉 任务 {$taskId} 执行完成，文生文生成成功！\n";
                echo "结束时间：" . date('Y-m-d H:i:s') . "\n";
                $job->delete();

            } catch (\Exception $e) {
                echo "文生文失败: " . $e->getMessage() . "\n";
                $job->delete();
            }
            $job->delete();
        }else{

            $logId = $data['log_id'] ?? null;
            echo " 开始处理文生文".date('Y-m-d H:i:s')."\n";

            try {
                if ($logId) {
                    Db::name('image_task_log')->where('id', $logId)->update([
                        'status' => 1,
                        'log' => '文生文处理中',
                        'update_time' => date('Y-m-d H:i:s')
                    ]);
                }

                $fullPath = rtrim($data['sourceDir'], '/') . '/' . ltrim($data['file_name'], '/');
                $list = Db::name("text_to_image")
                    ->where('old_image_url', $fullPath)
                    ->where('img_name', '<>', '')
                    ->select();

                if (!empty($list)) {
                    foreach ($list as $index => $row) {
                        $currentTime = date('Y-m-d H:i:s');
                        echo "处理时间：{$currentTime}\n";
                        echo "👉 正在处理第 " . ($index + 1) . " 条，ID: {$row['id']}\n";

                        $result = $this->get_txt_to_txt($data,$row['id']);
                        echo $result;
                        echo "✅ 处理结果：完成\n";
                        echo "完成时间：" . date('Y-m-d H:i:s') . "\n";

                        echo "Processed: " . static::class . "\n\n";
                    }

                    // 更新日志为成功
                    if ($logId) {
                        Db::name('image_task_log')->where('id', $logId)->update([
                            'status' => 2,
                            'log' => '文生文处理成功',
                            'update_time' => date('Y-m-d H:i:s')
                        ]);
                    }

                    echo "处理完成\n";
                } else {
                    echo "⚠ 未找到可处理的数据，跳过执行\n";

                    if ($logId) {
                        Db::name('image_task_log')->where('id', $logId)->update([
                            'status' => 2,
                            'log' => '无数据可处理，已跳过',
                            'update_time' => date('Y-m-d H:i:s')
                        ]);
                    }
                }

                // 链式执行：检查是否还有下一个任务
                if (!empty($data['chain_next'])) {
                    $nextType = array_shift($data['chain_next']); // 获取下一个任务类型
                    $data['type'] = $nextType;

                    Queue::push('app\job\ImageArrJob', [
                        'task_id' => $data['task_id'],
                        'data' => [$data]  // 继续传一个任务
                    ], 'arrimage');
                }
                $job->delete();
            } catch (\Exception $e) {
                echo "❌ 错误信息: " . $e->getMessage() . "\n";
                echo "📄 文件: " . $e->getFile() . "，第 " . $e->getLine() . " 行\n";

                if ($logId) {
                    Db::name('image_task_log')->where('id', $logId)->update([
                        'status' => -1,
                        'log' => '文生文失败：' . $e->getMessage(),
                        'update_time' => date('Y-m-d H:i:s')
                    ]);
                }
            }
            $job->delete();
        }

    }


    /**
     * 文生文核心处理逻辑（调用 GPT 接口）
     * @return string
     */
    public function get_txt_to_txt($data,$id)
    {
        $ai = new AIGatewayService();
        if ($data['status_type'] == 'ProductImageGeneration') {
            $content = "你是专业的产品文案优化师，仅返回优化后的文案，无任何多余内容：
                        优化规则：
                        1. 语气：亲切自然，符合电商产品描述风格；
                        2. 结构：突出产品卖点，逻辑清晰；
                        3. 禁忌：不添加无关形容词，不修改产品核心信息；
                        需要优化的文案：";
            $prompt = $content. $data['prompt'];
            $gptRes = $ai->buildRequestData($data['model'],$data['status_val'],$prompt);
            $gptText = trim($gptRes['choices'][0]['message']['content']);
            return $gptText;
        }else{
            $template = Db::name('template')
                ->field('id,english_content,content')
                ->where('path', $data['sourceDir'])
                ->where('ids', 1)
                ->find();

            $record = Db::name('text_to_image')
                ->field('id,english_description,chinese_description')
                ->where('id', $id)
                ->order('id desc')
                ->find();

            if (!$record) {
                return '没有找到匹配的图像记录';
            }

            // 拼接提示词调用文生文接口
            $prompt = $template['english_content'] . $record['chinese_description'];
            $gptRes = $ai->buildRequestData($data['model'],$data['status_val'],$prompt);
            $gptText = trim($gptRes['choices'][0]['message']['content'] ?? '');

            // 更新数据库记录
            Db::name('text_to_image')->where('id', $record['id'])->update([
                'english_description' => $gptText,
                'status_name' => "文生文"
            ]);
            return 0;
        }

    }

}