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

<?php
namespace app\job;
use app\service\AIGatewayService;
use think\Db;
use think\queue\Job;
use think\Queue;
/**
 * 图生文任务处理类
 * 功能：识别图片内容并调用大模型生成描述，支持链式任务衔接
 */
class ImageJob{
    /**
     * 队列主入口
     * @param Job $job 当前队列任务对象
     * @param array $data 包含图片路径、提示词、任务ID、链式信息等
     */
    public function fire(Job $job, $data)
    {
        if (empty($data['status_val']) || $data['status_val'] == '图生文') {
            // 获取任务ID
            $taskId = $data['id'] ?? null;
            if (empty($taskId)) {
                $job->delete();
                return;
            }

            // 创建Redis连接（配置见 application/extra/queue.php）
            $redis = getTaskRedis();

            // 设置锁键名
            $lockKey = "image_to_text_lock:{$taskId}";
            $lockExpire = 180; // 锁的过期时间，单位：秒（2分钟）

            // 尝试获取锁（使用SET命令的NX和EX选项进行原子操作）
            // NX: 只在键不存在时设置
            // EX: 设置过期时间
            $lockAcquired = $redis->set($lockKey, time(), ['NX', 'EX' => $lockExpire]);

            if (!$lockAcquired) {
                // 无法获取锁，任务正在执行
                echo "❌ 检测到相同ID({$taskId})的任务正在执行，当前任务跳过\n";
                $job->delete();
                return;
            } else {
                echo "🔒 成功获取任务锁，ID: {$taskId}\n";
            }

            echo "━━━━━━━━━━ ▶ 图生文任务开始处理━━━━━━━━━━\n";
            $result = $this->get_img_to_txt($data);
            // 标准化结果文本
            if ($result === true || $result === 1 || $result === '成功') {
                $resultText = '成功';
            } else {
                $resultText = (string) $result ?: '失败或无返回';
            }
            echo "✅ 处理结果：{$resultText}\n";
            echo "完成时间：" . date('Y-m-d H:i:s') . "\n";
            echo "图生文已处理完成\n";

            // 释放锁 - 使用del()替代被弃用的delete()方法
            $redis->del($lockKey);
            echo "🔓 释放任务锁，ID: {$taskId}\n";
            $job->delete();
            die;
        } else {
            $logId = $data['log_id'];
            try {
                echo date('Y-m-d H:i:s')."图生文开始\n";
                // 更新日志状态：处理中
                if ($logId) {
                    Db::name('image_task_log')->where('id', $logId)->update([
                        'status' => 1,
                        'log' => '图生文处理中',
                        'update_time' => date('Y-m-d H:i:s')
                    ]);
                }
                //执行逻辑
                $result = $this->processImage($data);
                echo $result;

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

                echo date('Y-m-d H:i:s')."图生文结束\n";

                //链式任务：图生文成功后继续推送文生文
                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() . "\n";
                echo "行号: " . $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();
            }
        }
        $job->delete();
    }

    /**
     * 失败回调（可用于后续通知或重试机制）
     */
    public function failed($data)
    {
        echo "ImageJob failed: " . json_encode($data);
    }



    public function processImage($data)
    {
        // 根据传入的数据处理图片
        $res = $this->imageToText($data["sourceDir"],$data["file_name"],$data["prompt"],$data["sys_id"],$data["imgtotxt_selectedOption"],$data);
        echo $res;
    }

    public function get_img_to_txt($params){

        $prompt = $params['prompt'];//提示词
        $old_path = $params['path'] ?? '';//原图路径
        $model = $params['model'];//模型

        if(empty($model)) {
            return json(['code' => 1, 'msg' => '模型请求失败']);
        }
        $aiGateway = new AIGatewayService();
        // 获取图片的base64数据和MIME类型
        $imageData = AIGatewayService::file_get_contents($old_path);
        $base64Data = $imageData['base64Data'];
        $mimeType = $imageData['mimeType'];

        $formattedPrompt = "1. 输出语言：所有内容必须为纯简体中文，禁止出现任何英文、拼音、数字、注释、解释性文字、引导语、示例、标点外的特殊符号；
            2. 第一步（提取原图产品）：
               - 用1句完整中文描述原图产品，字数控制在50字以内；
               - 必须包含「主体细节、产品名称、商标、类型、颜色、风格」核心；
            3. 第二步（生成新图提示词）：
               - 仅替换【模板提示词】中「产品主体」相关内容为第一步的产品描述；
               - 严格保留模板中「设计风格、光影、背景、比例、排版」等非产品相关信息；
               - 完全排除模板中的标题类信息，仅保留提示词核心内容；
               - 替换后必须完全保留原图产品的核心特征，禁止修改模板非产品核心信息；
            4. 输出格式：不允许添加任何解释、引导、说明、示例、备注,仅返回「产品描述 + 替换后提示词」，直接输出纯文本；
            【模板提示词】{$prompt}";

        $result = $aiGateway->callGptApi($model, $formattedPrompt, $mimeType, $base64Data);

        // Gemini模型响应格式处理
        $imgToTxtContent = $result['candidates'][0]['content']['parts'][0]['text'];

        // // 图生文成功后，调用文生文功能(gemini-2.0-flash)
        $txtToTxtPrompt = "转换成中文格式，去掉其他特殊符号,不允许添加任何解释、引导、说明、示例等文字：\n\n{$imgToTxtContent}";
        $txtToTxtResult = $aiGateway->txtGptApi($txtToTxtPrompt, 'gemini-2.0-flash');
        $finalContent = $txtToTxtResult['candidates'][0]['content']['parts'][0]['text'];

        // 处理调试输出内容
        $debugContent = json_encode($finalContent, JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT);
        // 去掉特殊符号（只保留字母、数字、中文、常用标点和换行）
        $debugContent = preg_replace('/[^\p{Han}\w\s\n\r\.,，。！!？?\-\:：;；]/u', '', $debugContent);
        // 去掉第一个冒号前的文字
        $debugContent = preg_replace('/^[^:]+:/', '', $debugContent);

        Db::name('product')->where('id',  $params['id'])->update(['content' => $debugContent]);
        return  '成功';
    }

    /**
     * 执行图生文逻辑（图像转文本）
     */
    public function imageToText($sourceDirRaw, $fileName, $prompt,$sys_id,$imgtotxt_selectedOption,$call_data)
    {
        // 自动拆分文件名
        if (!$fileName && preg_match('/([^\/]+\.(jpg|jpeg|png))$/i', $sourceDirRaw, $matches)) {
            $fileName = $matches[1];
            $sourceDirRaw = preg_replace('/\/' . preg_quote($fileName, '/') . '$/', '', $sourceDirRaw);
        }

        if ($sourceDirRaw === '' || $fileName === '') {
            return '参数错误：原图路径 或 图片名称 不能为空';
        }

        // 构建路径
        $rootPath = str_replace('\\', '/', ROOT_PATH);
        $sourceDir = rtrim($rootPath . 'public/' . $sourceDirRaw, '/') . '/';
        $filePath = $sourceDir . $fileName;
        $relativePath = $sourceDirRaw . '/' . $fileName;

        // 文件夹是否存在（绝对路径检查）
        if (!is_dir($sourceDir)) {
            return '源目录不存在：' . $sourceDir;
        }

        // 文件是否存在
        if (!is_file($filePath)) {
            return '文件不存在：' . $filePath;
        }

        // 获取图片信息
        $ext = strtolower(pathinfo($filePath, PATHINFO_EXTENSION));
        $mime = ($ext === 'jpg' || $ext === 'jpeg') ? 'jpeg' : $ext;

        list($width, $height) = getimagesize($filePath);
        $imageData = base64_encode(file_get_contents($filePath));
        if (!$imageData || strlen($imageData) < 1000) {
            throw new \Exception('图片内容读取失败');
        }
        $imageUrl = "data:image/{$mime};base64,{$imageData}";

        // 记录提示词日志
        $logDir = $rootPath . 'runtime/logs/';
        if (!is_dir($logDir)) mkdir($logDir, 0755, true);

        // 调用 图生文 接口
        $ai = new AIGatewayService();
        $gptRes = $ai->callGptApi($imgtotxt_selectedOption,$prompt,$imageUrl,'');

        $gptText = trim($gptRes['choices'][0]['message']['content'] ?? '');
    
        //图生文返回内容日志runtime/logs/img_to_txt.txt
        // file_put_contents(
        //     $logDir . 'img_to_txt.txt',
        //     "\n======== " . date('Y-m-d H:i:s') . " ========\n" .
        //   $gptText. "\n\n",
        //     FILE_APPEND
        // );die;

        // 验证 GPT 返回格式
        if (strpos($gptText, '---json json---') === false) {
            return  'GPT 返回格式不正确，缺少分隔符';
        }

        // 以 ---json json--- 分割
        $parts = array_map('trim', explode('---json json---', $gptText));

        // 清理“第一段”、“第二段”等标签前缀
        $cleanPrefix = function ($text) {
            return preg_replace('/^第[一二三四五六七八九十]+段[:：]?\s*/u', '', $text);
        };
        // 防止越界，逐个安全提取
        $chineseDesc = isset($parts[0]) ? $cleanPrefix(trim($parts[0])) : '';
        $englishDesc = isset($parts[1]) ? $cleanPrefix(trim($parts[1])) : '';
        $part2 = isset($parts[2]) ? $cleanPrefix(trim($parts[2])) : '';

        // 只保留中英文、数字、下划线、短横线、空格
        $img_name = preg_replace('/[^\x{4e00}-\x{9fa5}A-Za-z0-9_\- ]/u', '', $part2);

        //图生文返回日志拆分后内容runtime/logs/img_to_txt.txt
        // file_put_contents(
        //     $logDir . 'img_to_txt.txt',
        //     "\n======== " . date('Y-m-d H:i:s') . " ========\n" .
        //     $englishDesc . "\n--- 中文 ---\n" .
        //     $chineseDesc . "\n---英文 ---\n" .
        //     $img_name . "\n--- 命名 ---\n" ,
        //     FILE_APPEND
        // );

        $now = date('Y-m-d H:i:s');
        $record = [
            'chinese_description' => $chineseDesc,
            'english_description' => $englishDesc,
            'img_name' => $img_name,
            'sys_id' => $sys_id,
            'status' => 0,
            'status_name' => "图生文",
            'model' => "",
            'size' => "",
            'error_msg' => '',
            'update_time' => $now
        ];

        // 查询是否存在相同的原图记录
        $exists = Db::name('text_to_image')->where('old_image_url', $relativePath)->find();

        if ($exists) {
            // 如果已存在，执行更新
            Db::name('text_to_image')->where('id', $exists['id'])->update($record);
        } else {
            // 不存在则添加必要字段并插入
            $record['old_image_url'] = $relativePath;
            $record['new_image_url'] = '';
            $record['custom_image_url'] = '';
            $record['created_time'] = $now;
            Db::name('text_to_image')->insert($record);
        }
        return ;
    }
}