mp_audio_manage/app/Services/Anime/AnimeService.php

<?php

namespace App\Services\Anime;

use App\Consts\ErrorConst;
use App\Facade\Site;
use App\Libs\Utils;
use App\Models\MpGeneratePicTask;
use App\Services\AIGeneration\AIImageGenerationService;
use Dflydev\DotAccessData\Util;
use GuzzleHttp\Client;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\Log;
use Illuminate\Support\Facades\Redis;
use OSS\Core\OssException;
use OSS\OssClient;

class AnimeService
{
    protected $aiImageGenerationService;
    private $url;
    private $api_key;
    private $headers;

    public function __construct(AIImageGenerationService $aiImageGenerationService) {
        $this->aiImageGenerationService = $aiImageGenerationService;
        $this->url = 'https://api.deepseek.com/chat/completions';
        $this->api_key = env('DEEPSEEK_API_KEY');
        $this->headers = [
            'Authorization' => 'Bearer '.$this->api_key,
            'Content-Type'  => 'application/json; charset=UTF-8'
        ];
    }

    public function batchSetRoleImg($data) {
        $anime_id = getProp($data, 'anime_id');
        $episode_number = getProp($data, 'episode_number');
        if (!$anime_id) Utils::throwError('1002:请选择对话');
        $anime = DB::table('mp_animes')->where('id', $anime_id)->where('is_deleted', 0)->first();
        $roles = json_decode(getProp($anime, 'roles'), true);
        $art_style = getProp($anime, 'art_style');
        
        foreach ($roles as &$role) {
            $role_name = getProp($role, 'role');
            $description = getProp($role, 'description');
            if ($art_style) $description = "美术风格：\n$art_style\n主体描述：$description";
            // 参考图地址
            $ref_img_url = getProp($role, 'url');
            
            // 此处调用AIImageGenerationService的生成图片任务接口: createImageGenerationTask(注意使用默认参数,prompt使用角色描述),并返回任务id
            try {
                $params = [
                    'prompt' => $description,
                    'ref_img_urls' => !empty($ref_img_url) ? [$ref_img_url] : []
                ];
                
                $task = $this->aiImageGenerationService->createImageGenerationTask($params);
                $task_id = $task->id;
                
                if (!$task_id) {
                    Utils::throwError("20003: 角色({$role_name})生成图片失败");
                }
                
                $role['task_id'] = $task_id;
                $role['task_status'] = 'processing';
            } catch (\Exception $e) {
                Utils::throwError("20003:" . $e->getMessage());
            }
        }
        
        // 保存角色信息
        // $boolen = DB::table('mp_animes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->where('is_default', 1)->update([
        $boolen = DB::table('mp_animes')->where('id', $anime_id)->update([
            'roles'             => json_encode($roles, 256),
            'generate_status'   => '执行中',
            'generate_at'       => date('Y-m-d H:i:s'),
            'updated_at'        => date('Y-m-d H:i:s')
        ]);
        if (!$boolen) {
            Utils::throwError('20003:保存角色信息失败');
        }

        return $boolen;
    }


    public function batchSetSceneImg($data) {
        $anime_id = getProp($data, 'anime_id');
        if (!$anime_id) Utils::throwError('1002:请选择对话');
        $episode_number = getProp($data, 'episode_number');
        $anime = DB::table('mp_animes')->where('id', $anime_id)->where('is_deleted', 0)->first();
        $scenes = json_decode(getProp($anime, 'scenes'), true);
        $art_style = getProp($anime, 'art_style');
        
        foreach ($scenes as &$scene) {
            $scene_name = getProp($scene, 'scene');
            $description = getProp($scene, 'description');
            if ($art_style) $description = "美术风格：\n$art_style\n\n场景描述：$description";
            // 参考图地址
            $ref_img_url = getProp($scene, 'url');
            
            // 此处调用AIImageGenerationService的生成图片任务接口: createImageGenerationTask(注意使用默认参数,prompt使用角色描述),并返回任务id
            try {
                $params = [
                    'prompt' => $description,
                    'ref_img_urls' => !empty($ref_img_url) ? [$ref_img_url] : []
                ];
                
                $task = $this->aiImageGenerationService->createImageGenerationTask($params);
                $task_id = $task->id;
                
                if (!$task_id) {
                    Utils::throwError("20003: 角色({$scene_name})生成图片失败");
                }
                
                $scene['task_id'] = $task_id;
                $scene['task_status'] = 'processing';
            } catch (\Exception $e) {
                Utils::throwError("20003:" . $e->getMessage());
            }
        }
        
        // 保存角色信息
        // $boolen = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->where('is_default', 1)->update([
        $boolen = DB::table('mp_animes')->where('id', $anime_id)->update([
            'scenes'             => json_encode($scenes, 256),
            'generate_status'   => '执行中',
            'generate_at'       => date('Y-m-d H:i:s'),
            'updated_at'        => date('Y-m-d H:i:s')
        ]);
        if (!$boolen) {
            Utils::throwError('20003:保存角色信息失败');
        }

        return $boolen;
    }

    public function editAnime($data) {
        $script_arr =getProp($data, 'script', []);
        if (!$script_arr) Utils::throwError('20003:剧本内容不能为空');
        $start_episode_sequence = getProp($data, 'start_episode_sequence', 1);
        $anime_id = getProp($data, 'anime_id');
        try {   
            DB::beginTransaction();
            $table_data = [
                'intro' => isset($script_arr['intro']) ? $script_arr['intro'] : '',
                'highlights' => isset($script_arr['highlights']) ? $script_arr['highlights'] : '',
                'role_relationship' => isset($script_arr['role_relationship']) ? $script_arr['role_relationship'] : '',
                'core_contradiction' => isset($script_arr['core_contradiction']) ? $script_arr['core_contradiction'] : '',
                'roles' => isset($script_arr['roles']) ? json_encode($script_arr['roles'], 256) : '',
                'art_style' => isset($script_arr['art_style']) ? $script_arr['art_style'] : '',
                'scenes' => isset($script_arr['scenes']) ? json_encode($script_arr['scenes'], 256) : '',
                'status' => 3,
                'start_episode_sequence' => $start_episode_sequence,
                'end_episode_sequence' => $start_episode_sequence + count($script_arr['episodes']) - 1,
                'episode_num' => count($script_arr['episodes']),
                'updated_at' => date('Y-m-d H:i:s')
            ];

            // 保存剧本内容
            if (!empty($anime_id)) {
                $boolen = DB::table('mp_animes')->where('id', $anime_id)->update($table_data);
                if (!$boolen) {
                    dLog('deepseek')->info('对话保存失败', $table_data);
                    Utils::throwError('20003:对话保存失败');
                }
            }else {
                $table['created_at'] = $table_data['updated_at'];
                $anime_id = DB::table('mp_animes')->insertGetId($table_data);
                if (!$anime_id) {
                    dLog('deepseek')->info('对话保存失败', $table_data);
                    Utils::throwError('20003:对话保存失败');
                }
            }
            
            
            // 保存分集内容
            // 删除历史分集内容
            $boolen1 = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->delete();
            
            $episodes  = [];
            $sequence = 1;
            foreach ($script_arr['episodes'] as $episode) {
                $segment_number = 1;
                foreach($episode['segments'] as $segment) {
                    $episodes[] = [
                        'anime_id' => $anime_id,
                        'episode_number' => $episode['episode_number'],
                        'title' => $episode['title'],
                        // 'content' => $episode['content'],
                        'content' => '',
                        'segment_number' => $segment_number,
                        'segment_content' => $segment['segment_content'],
                        'sequence' => $sequence,
                        'created_at' => date('Y-m-d H:i:s'),
                        'updated_at' => date('Y-m-d H:i:s')
                    ];
                    $sequence++;
                    $segment_number++;
                }
                
            }

            $boolen2 = DB::table('mp_anime_episodes')->insert($episodes);
            if (!$boolen2) {
                dLog('deepseek')->info('分集内容保存失败', $episodes);
                Utils::throwError('20003:分集内容保存失败');
            }

        } catch (\Exception $e) {
            DB::rollBack();
            Utils::throwError('20003:'.$e->getMessage());
        }
        DB::commit();
        return true;
    }

    public function animeDetail($data) {
        $anime_id = getProp($data, 'anime_id');
        $anime = DB::table('mp_animes')->where('id', $anime_id)->where('is_deleted', 0)
        ->selectRaw('id as anime_id, anime_name, intro, highlights, roles, role_relationship, core_contradiction, art_style, scenes, status, remark, generate_status, created_at')->first();
        if (!$anime) Utils::throwError('20003:该剧本不存在!');
        $anime = (array)$anime;
        $anime['roles'] = json_decode($anime['roles']);
        $anime['scenes'] = json_decode($anime['scenes']);

        // 获取分集信息
        $episodes = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('is_default', 1)
        ->select('id as episode_id', 'episode_number', 'title')
        ->orderBy('episode_number')->get()->map(function ($value) {
            return (array)$value;
        })->toArray();

        $anime['episodes'] = $episodes;

        return $anime;
    }

    public function episodeInfo($data) {
        $anime_id = getProp($data, 'anime_id');
        $episode_number = getProp($data, 'episode_number');
        if (!$anime_id || !$episode_number) {
            Utils::throwError('1002:请选择剧集');
        }

        $episode = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->where('is_default', 1)->select('id as episode_id', 'anime_id', 'episode_number', 'title', 'intro', 'art_style', 'roles', 'scenes')->first();
        if (!$episode) Utils::throwError('20003: 该剧集不存在!');
        $episode = (array)$episode;
        $episode['roles'] = json_decode($episode['roles'], true);
        $episode['scenes'] = json_decode($episode['scenes'], true);
        
        // 获取分镜信息
        $segments = DB::table('mp_episode_segments')->where('anime_id', $anime_id)->where('episode_id', getProp($episode, 'episode_id'))->orderBy('segment_number')->select('act_number', 'act_title', 'segment_id', 'segment_number', 'segment_content')->get()->map(function ($value) {
            return (array)$value;
        })->toArray();
        
        $acts = [];
        foreach($segments as $segment) {
            if (isset($acts[getProp($segment, 'act_number')])) {
                $acts[getProp($segment, 'act_number')]['segments'][] = [
                    'segment_id'    => getProp($segment, 'segment_id'),
                    'segment_number'    => getProp($segment, 'segment_number'),
                    'segment_content'    => getProp($segment, 'segment_content'),
                ];
            }else {
                $acts[getProp($segment, 'act_number')]= [
                    'act_number' => getProp($segment, 'act_number'),
                    'act_title' => getProp($segment, 'act_title'),
                    'segments' => [
                        [
                            'segment_id'    => getProp($segment, 'segment_id'),
                            'segment_number'    => getProp($segment, 'segment_number'),
                            'segment_content'    => getProp($segment, 'segment_content'),
                        ]
                    ]
                ];
            }
        }
        $episode['acts'] = array_values($acts);

        return $episode;
    }

    public function copyEpisodeVersion($data) {
        $episode_id = getProp($data, 'episode_id');

        $episode = DB::table('mp_anime_episodes')->where('id', $episode_id)->first();
        if (!$episode) Utils::throwError('20003:该剧集不存在');

        $episode = (array)$episode;
        $count = DB::table('mp_anime_episodes')->where('anime_id', $episode['anime_id'])->where('episode_number', $episode['episode_number'])->count('id');
        $sequence = $count + 1;
        unset($episode['id']);
        $episode['sequence'] = $sequence;
        $episode['created_at'] = date('Y-m-d H:i:s');
        $episode['updated_at'] = date('Y-m-d H:i:s');
        $episode['is_default'] = 1;

        try {
            DB::beginTransaction();
            // 移除is_default标志
            $boolen = DB::table('mp_anime_episodes')->where('anime_id', $episode['anime_id'])->where('episode_number', $episode['episode_number'])->update(['is_default' => 0, 'updated_at'=>date('Y-m-d H:i:s')]);
            if (!$boolen) {
                Utils::throwError('20003:更新失败!');
            }

            // 新增副本
            $new_episode_id = DB::table('mp_anime_episodes')->insertGetId($episode);
            if (!$new_episode_id) {
                Utils::throwError('20003:创建副本失败!');
            }
            $episode['episode_id'] = $new_episode_id;

            // 获取分镜数据
            $segments = DB::table('mp_episode_segments')->where('episode_id', $episode_id)->get()->map(function($item) use ($new_episode_id) {
                $item = (array)$item;
                unset($item['id']);
                $item['episode_id'] = $new_episode_id;
                $item['segment_id'] = date('YmdHis').mt_rand(1000,9999).str_pad($item['segment_number'], 3, "0", STR_PAD_LEFT);
                $item['created_at'] = $item['updated_at'] = date('Y-m-d H:i:s');
                return $item;
            })->toArray();

            // 写入分镜数据
            $boolen2 = DB::table('mp_episode_segments')->insert($segments);
            if (!$boolen2) {
                Utils::throwError('20003:写入分镜数据失败!');
            }
        }catch (\Exception $e) {
            DB::rollBack();
            Utils::throwError('20003:'.$e->getMessage());
        }

        DB::commit();

        return $episode;
    }

    public function episodeVersions($data) {
        $anime_id = getProp($data, 'anime_id');
        $episode_number = getProp($data, 'episode_number');
        if (!$anime_id || !$episode_number) {
            Utils::throwError('1002:请选择剧集');
        }

        $episodes = DB::table('mp_anime_episodes')->where('anime_id', $anime_id)->where('episode_number', $episode_number)->select('anime_id', 'episode_number', 'sequence', 'title', 'intro', 'art_style', 'roles', 'scenes')->get()->map(function($value) {
            return (array)$value;
        })->toArray();
        if (!$episodes) Utils::throwError('20003: 该剧集不存在!');
        foreach($episodes as &$episode) {
            $episode['version'] = 'V'.$episode['sequence'];
            $episode['roles'] = json_decode($episode['roles'], true);
            $episode['scenes'] = json_decode($episode['scenes'], true);
        }

        return $episodes;
    }

    public function bindEpisodeVersion($data) {
        $episode_id = getProp($data, 'episode_id');

        $episode = DB::table('mp_anime_episodes')->where('id', $episode_id)->first();
        if (!$episode) Utils::throwError('20003:该剧集不存在');
        $episode = (array)$episode;

        if ((int)$episode['is_default'] === 1) return true;

        try {
            DB::beginTransaction();
            // 移除is_default标志
            $boolen = DB::table('mp_anime_episodes')->where('anime_id', $episode['anime_id'])->where('episode_number', $episode['episode_number'])->update(['is_default' => 0, 'updated_at'=>date('Y-m-d H:i:s')]);
            if (!$boolen) {
                Utils::throwError('20003:更新失败!');
            }

            // 绑定副本
            $boolen2 = DB::table('mp_anime_episodes')->where('id', $episode_id)->update(['is_default' => 1, 'updated_at'=>date('Y-m-d H:i:s')]);
            if (!$boolen2) {
                Utils::throwError('20003:绑定失败!');
            }
        }catch (\Exception $e) {
            DB::rollBack();
            Utils::throwError('20003:'.$e->getMessage());
        }

        DB::commit();

        return true;
    }

    public function chatChangeImg($data) {
        $segment_id = getProp($data, 'segment_id');
        $prompt = getProp($data, 'prompt');
        if (!$prompt || !$segment_id) {
            Utils::throwError('1002:请输入提示词,并且选择需修改的图片');
        }
        $segment = DB::table("mp_episode_segments")->where('segment_id', $segment_id)->first();
        if (!$segment) Utils::throwError('20003:该分镜不存在!');
        $ref_img_url = getProp($segment, 'img_url');
        if (empty($ref_img_url)) {
            $prompt = getProp($segment, 'segment_content');
        }

        try {
            $params = [
                'alias_segment_id' => $segment_id,
                'prompt' => $prompt,
                'ref_img_urls' => !empty($ref_img_url) ? [$ref_img_url] : []
            ];
            
            $task = $this->aiImageGenerationService->createImageGenerationTask($params);
            $task_id = $task->id;
            
            if (!$task_id) {
                Utils::throwError("20003: 生成图片失败");
            }
            
            // 更新任务ID
            $boolen = DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update(['pic_task_id' => $task_id, 'pic_task_status' => '生成中', 'updated_at' => date('Y-m-d H:i:s')]);
            if (!$boolen) Utils::throwError('20003:更新任务ID失败');
        } catch (\Exception $e) {
            Utils::throwError("20003:" . $e->getMessage());
        }

        $start_count = 0;
        $img_url = '';
        while ($start_count <= 30) {
            sleep(3);

            $task = MpGeneratePicTask::where('id', $task_id)->first();
            $statusInfo = $this->aiImageGenerationService->queryTaskStatus($task);

            if ($statusInfo['status'] === 'success') {
                $task->updateStatus('success', [
                    'result_url'    => $statusInfo['result_url'],
                    'result_json'   => $statusInfo['result_json'] ?? []
                ]);

                // 同步调整分镜图片状态和结果
                if (getProp($task, 'alias_segment_id')) {
                    DB::table('mp_episode_segments')->where('segment_id', getProp($task, 'alias_segment_id'))->update([
                        'img_url'   => $statusInfo['result_url'][0],
                        'pic_task_status'   => '已完成',
                    ]);
                }
                $img_url = $statusInfo['result_url'][0];
                break;
            } elseif ($statusInfo['status'] === 'failed') {
                $task->updateStatus('failed', [
                    'error_message' => $statusInfo['error_message'],
                    'result_json'   => $statusInfo['result_json'] ?? []
                ]);

                if (getProp($task, 'alias_segment_id')) {
                    DB::table('mp_episode_segments')->where('segment_id', getProp($task, 'alias_segment_id'))->update([
                        'pic_task_status'   => '失败',
                    ]);
                }
                break;
            }
            $start_count+=3;
        }

        if (!$img_url) Utils::throwError('20003:图片生成中,请稍后刷新');
        return $img_url;
        
    }

    public function batchSetSegmentTasks($data) {
        $anime_id = getProp($data, 'anime_id');
        $episode_number = getProp($data, 'episode_number');
        
        if (!$anime_id || !$episode_number) {
            Utils::throwError('1002:请选择剧集');
        }

        // 获取剧集信息
        $episode = DB::table('mp_anime_episodes')
            ->where('anime_id', $anime_id)
            ->where('episode_number', $episode_number)
            ->where('is_default', 1)
            ->select('id as episode_id', 'anime_id', 'episode_number')
            ->first();
            
        if (!$episode) {
            Utils::throwError('20003:该剧集不存在');
        }

        // 获取动漫的角色信息（包含参考图）
        $anime = DB::table('mp_animes')
            ->where('id', $anime_id)
            ->select('roles', 'scenes')
            ->first();
            
        if (!$anime || !$anime->roles) {
            Utils::throwError('20003:未找到角色信息');
        }

        $roles = json_decode($anime->roles, true);
        if (!$roles) {
            Utils::throwError('20003:角色信息格式错误');
        }
        // 构建角色名称到参考图的映射
        $role_map = [];
        foreach ($roles as $role) {
            $role_name = getProp($role, 'role');
            $role_url = getProp($role, 'url');
            if (!empty($role_name) && !empty($role_url)) {
                // // 处理角色名称，去除装扮后缀（如：许芸-校服装 -> 许芸）- 修复中文乱码问题(弃用,容易产生重复图片)
                // $base_name = preg_replace('/[-－].*$/u', '', $role_name);
                // $role_map[$base_name] = $role_url;
                // 同时保存完整名称的映射
                $role_map[$role_name] = $role_url;
            }
        }

        $scenes = json_decode($anime->scenes, true);
        if (!$scenes) {
            Utils::throwError('20003:角色信息格式错误');
        }
        // 构建角色名称到参考图的映射
        $scene_map = [];
        foreach ($scenes as $scene) {
            $scene_name = getProp($scene, 'scene');
            $scene_url = getProp($scene, 'url');
            if (!empty($scene_name) && !empty($scene_url)) {
                $scene_map[$scene_name] = $scene_url;
            }
        }

        // 获取所有分镜信息
        $segments = DB::table('mp_episode_segments')
            ->where('anime_id', $anime_id)
            ->where('episode_id', $episode->episode_id)
            ->where('pic_task_status', '')
            ->orderBy('segment_number')
            ->select('episode_number', 'segment_id', 'segment_number', 'segment_content', 'scene', 'characters')
            ->limit(3)
            ->get()
            ->toArray();

        if (empty($segments)) {
            Utils::throwError('20003:未找到分镜信息');
        }

        $updated_segments = [];
        $failed_segments = [];

        // 批量为每个分镜生成图片任务
        foreach ($segments as $segment) {
            $segment_id = $segment->segment_id;
            $segment_content = $segment->segment_content;
            
            if (empty($segment_content)) {
                $failed_segments[] = [
                    'segment_id' => $segment_id,
                    'error' => '分镜内容为空'
                ];
                continue;
            }

            try {
                // 解析分镜内容，提取画面描述作为主要提示词
                $prompt = $segment_content;
                $ref_img_urls = [];
                // 分镜场景
                $scene = getProp($segment, 'scene');
                $matched_scene = '';
                if (isset($scene_map[$scene])) {
                    $img_index = count($ref_img_urls) + 1;
                    $prompt = "场景：$scene(图$img_index)\n$prompt";
                    $ref_img_urls[] = $scene_map[$scene];
                    $matched_scene = $scene;
                }

                // 分镜角色
                $characters = getProp($segment, 'characters');
                $characters = str_replace(['、', ',', ';'], ',', $characters);
                $roles = explode(',', $characters);
                
                // 从分镜内容中提取涉及的角色
                $segment_characters = $this->extractCharactersFromSegment($segment_content, array_keys($role_map), $roles);
                
                // 获取匹配角色的参考图
                foreach ($segment_characters as $character) {
                    if (isset($role_map[$character])) {
                        // 将$prompt中出现的角色替换成角色(图1、2、3,序号是$ref_img_urls的个数加1)
                        $img_index = count($ref_img_urls) + 1;
                        $prompt = str_replace($character, $character . '(图' . $img_index . ')', $prompt);
                        $ref_img_urls[] = $role_map[$character];
                    }
                }
                
                // 如果没有找到匹配的角色参考图，不使用参考图
                if (empty($ref_img_urls)) {
                    $ref_img_urls = [];
                }

                // 准备生成任务参数
                $params = [
                    'alias_segment_id' => $segment_id,
                    'prompt' => $prompt,
                    'ref_img_urls' => $ref_img_urls
                ];
                
                // $task_id = mt_rand(100000, 999999);
                // 调用AI图片生成服务
                $task = $this->aiImageGenerationService->createImageGenerationTask($params);
                $task_id = $task->id;
                
                if (!$task_id) {
                    $failed_segments[] = [
                        'segment_id' => $segment_id,
                        'error' => '创建生成任务失败'
                    ];
                    continue;
                }
                
                // 更新分镜的任务信息
                $update_result = DB::table('mp_episode_segments')
                    ->where('segment_id', $segment_id)
                    ->update([
                        'pic_task_id' => $task_id,
                        'pic_task_status' => '生成中',
                        'updated_at' => date('Y-m-d H:i:s')
                    ]);

                // 如果是第一集的首帧图片,则存入待更新数组
                if ((int)getProp($segment, 'episode_number') === 1 && (int)getProp($segment, 'segment_number') === 1) {
                    Redis::sadd('anime_first_frame_urls', $segment_id);
                }
                
                if ($update_result) {
                    $updated_segments[] = [
                        'segment_id' => $segment_id,
                        'task_id' => $task_id,
                        'status' => '生成中',
                        'matched_scenes' => $matched_scene,
                        'matched_roles' => $segment_characters,
                        'ref_urls_count' => count($ref_img_urls)
                    ];
                } else {
                    $failed_segments[] = [
                        'segment_id' => $segment_id,
                        'error' => '更新分镜任务状态失败'
                    ];
                }
                
            } catch (\Exception $e) {
                $failed_segments[] = [
                    'segment_id' => $segment_id,
                    'error' => $e->getMessage()
                ];
            }
        }

        // // 更新剧集生成状态
        // if (!empty($updated_segments)) {
        //     DB::table('mp_anime_episodes')
        //         ->where('id', $episode->episode_id)
        //         ->update([
        //             'generate_status' => '执行中',
        //             'updated_at' => date('Y-m-d H:i:s')
        //         ]);
        // }

        return [
            'success_count' => count($updated_segments),
            'failed_count' => count($failed_segments),
            'success_segments' => $updated_segments,
            'failed_segments' => $failed_segments,
            'total_segments' => count($segments)
        ];
    }

    /**
     * 从分镜内容中提取涉及的角色
     */
    private function extractCharactersFromSegment($segment_content, $available_characters, $roles) {
        $found_characters = [];
        
        foreach ($available_characters as $character) {
            // 检查角色名称是否在分镜内容中出现
            if (strpos($segment_content, $character) !== false) {
                $found_characters[] = $character;
            }
        }

        if (!$found_characters) {
            foreach ($roles as $character) {
                // 检查角色名称是否在分镜内容中出现
                if (strpos($segment_content, $character) !== false) {
                    $found_characters[] = $character;
                }
            }
        }
        
        return array_unique($found_characters);
    }
}