图片模型图片

lib/pages/ai-draw-image-result/controller.dart

@@ -41,6 +41,7 @@ class AIDrawImageResultController extends GetxController {
   var showProgressView = false.obs;
   var loseTime = "0.0s".obs;
   var messageId = "";
+  var didCollected = false.obs;
 
   Rx<Uint8List?> imageData = Rx<Uint8List?>(null);
   Uint8List? placehoderImageData;
@@ -110,17 +111,35 @@ class AIDrawImageResultController extends GetxController {
   }
 
   void doCollect() async {
-    EasyLoading.show(status: "收藏中...");
-    final res = await NewsAPI.updateCollectImg(resultImageUrl, 1);
-    EasyLoading.dismiss();
-    if (res.status == 401) {
-      EasyLoading.showInfo('您还未登录，请登录后体验功能。');
-      Get.toNamed(AppRoutes.SIGN_IN);
+    if (didCollected.value) {
+      EasyLoading.show(status: "取消收藏中...");
+      final res = await NewsAPI.updateCollectImg(resultImageUrl, 0);
+      EasyLoading.dismiss();
+      if (res.status == 401) {
+        EasyLoading.showInfo('您还未登录，请登录后体验功能。');
+        Get.toNamed(AppRoutes.SIGN_IN);
+      } else {
+        if (res.status == 200) {
+          EasyLoading.showSuccess("取消收藏成功");
+          didCollected.value = false;
+        } else {
+          EasyLoading.showError(res.message ?? "");
+        }
+      }
     } else {
-      if (res.status == 200) {
-        EasyLoading.showSuccess("收藏成功");
+      EasyLoading.show(status: "收藏中...");
+      final res = await NewsAPI.updateCollectImg(resultImageUrl, 1);
+      EasyLoading.dismiss();
+      if (res.status == 401) {
+        EasyLoading.showInfo('您还未登录，请登录后体验功能。');
+        Get.toNamed(AppRoutes.SIGN_IN);
       } else {
-        EasyLoading.showError(res.message ?? "");
+        if (res.status == 200) {
+          EasyLoading.showSuccess("收藏成功");
+          didCollected.value = true;
+        } else {
+          EasyLoading.showError(res.message ?? "");
+        }
       }
     }
   }

lib/pages/ai-draw-image-result/view.dart

@@ -147,7 +147,9 @@ class AIDrawImageResultPage extends GetView<AIDrawImageResultController> {
                         color: AppColor.primary,
                         borderRadius: BorderRadius.circular(20),
                       ),
-                      child: Text("收藏")
+                      child: Image(image: AssetImage("assets/images/${controller.didCollected.value ? 'collect_selected_icon.png' : 'collect_normal_icon.png'}"), 
+                          width: 20,
+                          height: 20,)
                     ),
                   ),
                 ),
@@ -184,9 +186,9 @@ class AIDrawImageResultPage extends GetView<AIDrawImageResultController> {
                       crossAxisAlignment: CrossAxisAlignment.start,
                       children: [
                         Container(
-                          padding: EdgeInsets.fromLTRB(20, 0, 20, 0),
+                          padding: const EdgeInsets.fromLTRB(20, 0, 20, 0),
                           child: Container(
-                            padding: EdgeInsets.fromLTRB(10, 5, 10, 5),
+                            padding: const EdgeInsets.fromLTRB(10, 0, 10, 0),
                             decoration: BoxDecoration(
                               color: AppColor.gray6,
                               borderRadius: BorderRadius.circular(4),
@@ -195,19 +197,20 @@ class AIDrawImageResultPage extends GetView<AIDrawImageResultController> {
                                 width: 1,
                               )
                             ),
-                            child: Text("风格：${controller.styleName.value}", 
-                              style: TextStyle(
+                            child: Text("模型：${controller.styleName.value}", 
+                              style: const TextStyle(
                                 color: AppColor.primary,
-                                fontSize: 12
+                                fontSize: 12,
+                                fontWeight: FontWeight.bold
                               ),
                             ),
                           ),
                         ),
-                        SizedBox(height: 20,),
+                        const SizedBox(height: 20,),
                         Container(
-                          padding: EdgeInsets.fromLTRB(20, 0, 20, 0),
+                          padding: const EdgeInsets.fromLTRB(20, 0, 20, 0),
                           child: Text(controller.text.value, 
-                            style: TextStyle(
+                            style: const TextStyle(
                               color: Colors.white70,
                               fontSize: 14
                             ),

lib/pages/ai-draw-image/controller.dart

@@ -46,14 +46,15 @@ class AIDrawImageController extends GetxController {
   ];
 
   var styles = [
-    AIDrawImageStyle(s: " --v 5.2", imageUrl: "assets/images/全能风格.jpg", name: "MJV5.2", vip: "VIP"),
-    AIDrawImageStyle(s: " --niji 5", imageUrl: "assets/images/通用风格.jpg", name: "NIJI5", vip: ""),
-    AIDrawImageStyle(s: " --niji 4", imageUrl: "assets/images/2.5D动漫.jpg", name: "NIJI4", vip: "VIP"),
-    AIDrawImageStyle(s: " --v 5", imageUrl: "assets/images/真实照片.jpg", name: "MJV5", vip: "VIP"),
-    AIDrawImageStyle(s: " --v 4", imageUrl: "assets/images/日系女生.jpg", name: "MJV4", vip: "VIP"),
-    AIDrawImageStyle(s: " --v 3", imageUrl: "assets/images/韩系写真.jpg", name: "MJV3", vip: "VIP"),
-    AIDrawImageStyle(s: " --v 2", imageUrl: "assets/images/光影风格.jpg", name: "MJV2", vip: "VIP"),
-    AIDrawImageStyle(s: " --v 1", imageUrl: "assets/images/全能增强.jpg", name: "MJV1", vip: "VIP"),
+    AIDrawImageStyle(s: " --v 5.2", imageUrl: "assets/images/全能风格.jpg", name: "mj-v5.2", vip: ""),
+    AIDrawImageStyle(s: " --v 5.1", imageUrl: "assets/images/MJ51.jpg", name: "mjv-v5.1", vip: ""),
+    AIDrawImageStyle(s: " --niji 5", imageUrl: "assets/images/通用风格.jpg", name: "niji-v5", vip: ""),
+    AIDrawImageStyle(s: " --niji 4", imageUrl: "assets/images/2.5D动漫.jpg", name: "niji-v4", vip: ""),
+    AIDrawImageStyle(s: " --v 5", imageUrl: "assets/images/真实照片.jpg", name: "mj-v5", vip: ""),
+    AIDrawImageStyle(s: " --v 4", imageUrl: "assets/images/日系女生.jpg", name: "mj-v4", vip: ""),
+    AIDrawImageStyle(s: " --v 3", imageUrl: "assets/images/韩系写真.jpg", name: "mj-v3", vip: ""),
+    AIDrawImageStyle(s: " --v 2", imageUrl: "assets/images/光影风格.jpg", name: "mj-v2", vip: ""),
+    AIDrawImageStyle(s: " --v 1", imageUrl: "assets/images/全能增强.jpg", name: "mj-v1", vip: ""),
     /*
     AIDrawImageStyle(s: " --s 450", imageUrl: "assets/images/韩系女生.jpg", name: "韩系女生", vip: "VIP"),
     AIDrawImageStyle(s: " --s 500", imageUrl: "assets/images/吉卜力.jpg", name: "吉卜力", vip: "VIP"),

lib/pages/ai-draw-image/text_to_image_view.dart

@@ -459,7 +459,12 @@ Widget makeSectionTitleView(String index, String name) {
               ],
             ),
             const SizedBox(height: 5,),
-            Text(item.name)
+            Text(item.name, 
+              style: const TextStyle(
+                fontWeight: FontWeight.bold,
+                fontSize: 12
+              ),
+            )
           ],
         );
        },
@@ -527,13 +532,16 @@ Widget makeSectionTitleView(String index, String name) {
       child: Column(
           children: [
             const SizedBox(height: 10,),
-            ClipRRect(
+            Container(
+              padding: item.ratio == "1:1" ? const EdgeInsets.fromLTRB(5, 5, 5, 5) : EdgeInsets.zero,
+              child: ClipRRect(
               borderRadius: BorderRadius.circular(5),
               child: Image(image: AssetImage(item.imageUrl), 
                 width: item.ratio == "1:1" ? 20 : 30,
                 height: item.ratio == "1:1" ? 20 : 30,
               ),
             ),
+            ),
             const SizedBox(height: 3,),
             Text(item.name,
               style: const TextStyle(