From ec63c41648e96aafbf0ff3f7657fc3a4879e13c3 Mon Sep 17 00:00:00 2001
From: zoujing <kidevelop@163.com>
Date: Mon, 6 Apr 2026 19:19:32 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=89=A9=E4=BD=99=E6=AC=A1=E6=95=B0?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3=E5=AF=B9=E6=8E=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/AigcGenerationTimesApi.ts | 11 +++++++++++
 src/api/index.ts                  |  1 +
 src/views/history/index.vue       |  3 ++-
 src/views/home/index.vue          | 24 +++++++++++++++++++++---
 4 files changed, 35 insertions(+), 4 deletions(-)
 create mode 100644 src/api/AigcGenerationTimesApi.ts

diff --git a/src/api/AigcGenerationTimesApi.ts b/src/api/AigcGenerationTimesApi.ts
new file mode 100644
index 0000000..54d73fe
--- /dev/null
+++ b/src/api/AigcGenerationTimesApi.ts
@@ -0,0 +1,11 @@
+/* eslint-disable */
+// @ts-ignore
+import request from '@common/ajax';
+import type { Response } from './types';
+
+export const getAigcGenerationTimes = () => {
+  return request<Response<number>>({
+    url: '/aigc/remainingNumberOfGenerationTimes',
+    method: 'get'
+  });
+};
\ No newline at end of file
diff --git a/src/api/index.ts b/src/api/index.ts
index 2bd08a2..ba5bc10 100644
--- a/src/api/index.ts
+++ b/src/api/index.ts
@@ -3,6 +3,7 @@
 export * from './types';
 
 export * from './AigcSceneListApi';
+export * from './AigcGenerationTimesApi';
 export * from './AigcGeneratorPhotoTaskListApi';
 export * from './AigcCreateGeneratorPhotoTaskApi';
 export * from './UploadFile';
diff --git a/src/views/history/index.vue b/src/views/history/index.vue
index a900bad..2f845a9 100644
--- a/src/views/history/index.vue
+++ b/src/views/history/index.vue
@@ -112,10 +112,11 @@ const lookPicture = (item: TaskItem) => {
 const getTaskList = async () => {
   showLoadingToast('加载中...')
   try {
+    /// 任务状态 0-排队中 1-生成中 2-已完成 3-失败
     const res = await generatorPhotoTaskList({
       pageNum: 1,
       pageSize: 20,
-      taskStatus: '1'
+      taskStatus: ''
     })
     console.log('🚀 ~ generatorPhotoTaskList res:', res)
     closeToast()
diff --git a/src/views/home/index.vue b/src/views/home/index.vue
index 11def4a..b1329f6 100644
--- a/src/views/home/index.vue
+++ b/src/views/home/index.vue
@@ -44,7 +44,7 @@
 
       <div class="active-title">{{ activeScene.sceneName }}</div>
 
-      <div class="quota-text">✨ 今日还可生成 <span>2</span> 张</div>
+      <div class="quota-text">✨ 今日还可生成 <span>{{ aigcGenerationTimes }}</span> 张</div>
 
       <div class="footer-section">
         <button class="generate-btn" @click="generateAction">立即生成</button>
@@ -89,10 +89,10 @@ import PhotoGuide from '../components/PhotoGuide.vue'
 // @ts-ignore
 import AgreementTip from '../components/AgreementTip.vue'
 // @ts-ignore
-import { getAigcSceneList, AigcSceneListItem, AigcSceneStyleItem } from '@api'
+import {getAigcGenerationTimes, getAigcSceneList, AigcSceneListItem, AigcSceneStyleItem } from '@api'
 // @ts-ignore
 import { createGeneratorPhotoTask } from '@api'
-import { showFailToast, showSuccessToast } from 'vant'
+import { showFailToast, showSuccessToast, showToast } from 'vant'
 import 'vant/lib/toast/style'
 
 // --- 测试数据 ---
@@ -116,6 +116,8 @@ const showGuide = ref(false)
 
 const showAgree = ref(!Session.get('hasAgreedPrivacy'))
 
+const aigcGenerationTimes = ref(0)
+
 const cancelAgree = () => {
   console.log('用户拒绝了协议')
   showAgree.value = false
@@ -146,6 +148,10 @@ const onHistory = () => {
 }
 
 const generateAction = () => {
+  if (aigcGenerationTimes.value <= 0) {
+    showToast('今日生成次数已用完，请明天再试！')
+    return
+  }
   showGuide.value = true
 }
 
@@ -255,8 +261,20 @@ const fetchSceneList = async () => {
   console.log('处理后的场景列表：', JSON.stringify(mockData))
 }
 
+/// 获取生成次数
+const getGenerationTimes = async () => {
+  try {
+    const res = await getAigcGenerationTimes()
+    aigcGenerationTimes.value = res > 0 ? res : 0
+    console.log('🚀 ~ getAigcGenerationTimes res:', res)
+  } catch (error) {
+    console.error('🚀 ~ getAigcGenerationTimes error details:', error)
+  }
+}
+
 onMounted(() => {
   fetchSceneList()
+  getGenerationTimes()
 })
 </script>