diff --git a/server/main.py b/server/main.py index 6d3db1a..3473746 100644 --- a/server/main.py +++ b/server/main.py @@ -265,6 +265,7 @@ async def _generate_podcast_task( "task_results": task_results[auth_id][task_id], "timestamp": int(time.time()), "status": task_results[auth_id][task_id]["status"], + "usetime": usetime, } MAX_RETRIES = 3 # 定义最大重试次数 @@ -326,6 +327,7 @@ async def generate_podcast_submission( "timestamp": time.time(), "callback_url": callback_url, # 存储回调地址 "auth_id": auth_id, # 存储 auth_id + "usetime": usetime, } background_tasks.add_task( diff --git a/web/src/app/api/points/route.ts b/web/src/app/api/points/route.ts index 1ae7c65..cb0da8a 100644 --- a/web/src/app/api/points/route.ts +++ b/web/src/app/api/points/route.ts @@ -30,7 +30,7 @@ export async function GET(request: NextRequest) { // GET 函数接收 request } export async function PUT(request: NextRequest) { - const { task_id, auth_id, timestamp, status } = await request.json(); + const { task_id, auth_id, timestamp, status, usetime } = await request.json(); const lang = getLanguageFromRequest(request); // 获取语言 const { t } = await getTranslation(lang, 'errors'); // 初始化翻译 try { @@ -63,7 +63,11 @@ export async function PUT(request: NextRequest) { const userId = auth_id; // 这里假设 auth_id 就是 userId // 5. 扣减积分 - const pointsToDeduct = parseInt(process.env.POINTS_PER_PODCAST || '10', 10); // 从环境变量获取,默认10 + let pointsToDeduct = parseInt(process.env.POINTS_PER_PODCAST || '10', 10); // 从环境变量获取,默认10 + if(usetime === '8-15 minutes') { + pointsToDeduct = pointsToDeduct * 2; + } + const reasonCode = "podcast_generation"; const description = `${t("podcast_generation_task")}: ${task_id}`; // 多语言实现