feat: AI自动生成小说封面图片

novel-front/src/main/java/com/java2nb/novel/service/impl/BookServiceImpl.java

@@ -7,6 +7,7 @@ import com.java2nb.novel.core.cache.CacheService;
 import com.java2nb.novel.core.config.BookPriceProperties;
 import com.java2nb.novel.core.enums.ResponseStatus;
 import com.java2nb.novel.core.utils.Constants;
+import com.java2nb.novel.core.utils.FileUtil;
 import com.java2nb.novel.core.utils.StringUtil;
 import com.java2nb.novel.entity.Book;
 import com.java2nb.novel.entity.*;
@@ -27,10 +28,16 @@ import lombok.RequiredArgsConstructor;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
+import org.apache.http.client.utils.DateUtils;
 import org.mybatis.dynamic.sql.SortSpecification;
 import org.mybatis.dynamic.sql.render.RenderingStrategies;
 import org.mybatis.dynamic.sql.select.QueryExpressionDSL;
 import org.mybatis.dynamic.sql.select.render.SelectStatementProvider;
+import org.springframework.ai.image.Image;
+import org.springframework.ai.image.ImagePrompt;
+import org.springframework.ai.image.ImageResponse;
+import org.springframework.ai.openai.OpenAiImageModel;
+import org.springframework.ai.openai.OpenAiImageOptions;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
@@ -39,13 +46,16 @@ import org.springframework.transaction.annotation.Transactional;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
 import java.util.*;
+import java.util.concurrent.ThreadPoolExecutor;
 import java.util.stream.Collectors;
 
 import static com.java2nb.novel.mapper.BookCategoryDynamicSqlSupport.bookCategory;
+import static com.java2nb.novel.mapper.BookCategoryDynamicSqlSupport.sort;
 import static com.java2nb.novel.mapper.BookCommentDynamicSqlSupport.bookComment;
 import static com.java2nb.novel.mapper.BookContentDynamicSqlSupport.bookContent;
 import static com.java2nb.novel.mapper.BookContentDynamicSqlSupport.content;
 import static com.java2nb.novel.mapper.BookDynamicSqlSupport.*;
+import static com.java2nb.novel.mapper.BookDynamicSqlSupport.book;
 import static com.java2nb.novel.mapper.BookIndexDynamicSqlSupport.bookIndex;
 import static com.java2nb.novel.mapper.BookSettingDynamicSqlSupport.bookSetting;
 import static org.mybatis.dynamic.sql.SqlBuilder.*;
@@ -87,6 +97,10 @@ public class BookServiceImpl implements BookService {
 
     private final BookPriceProperties bookPriceConfig;
 
+    private final OpenAiImageModel openAiImageModel;
+
+    private final ThreadPoolExecutor threadPoolExecutor;
+
     private final IdWorker idWorker = IdWorker.INSTANCE;
 
 
@@ -235,7 +249,7 @@ public class BookServiceImpl implements BookService {
             BookIndexDynamicSqlSupport.isVip)
             .from(bookIndex)
             .where(BookIndexDynamicSqlSupport.bookId, isEqualTo(bookId));
-        if("index_num desc".equals(orderBy)){
+        if ("index_num desc".equals(orderBy)) {
             where.orderBy(BookIndexDynamicSqlSupport.indexNum.descending());
         }
         return bookIndexMapper.selectMany(where
@@ -502,6 +516,7 @@ public class BookServiceImpl implements BookService {
 
     @Override
     public void addBook(Book book, Long authorId, String penName) {
+        book.setId(IdWorker.INSTANCE.nextId());
         //判断小说名是否存在
         if (queryIdByNameAndAuthor(book.getBookName(), penName) != null) {
             //该作者发布过此书名的小说
@@ -516,7 +531,36 @@ public class BookServiceImpl implements BookService {
         book.setCreateTime(new Date());
         book.setUpdateTime(book.getCreateTime());
         bookMapper.insertSelective(book);
-
+        if (Objects.isNull(book.getPicUrl()) || !book.getPicUrl().startsWith(Constants.LOCAL_PIC_PREFIX)) {
+            // 用户没有上传封面图片，AI自动生成封面图片
+            threadPoolExecutor.execute(() -> {
+                String prompt = String.format("生成一本小说的封面图片，图片中间显示书名《%s》，书名下方显示作者“%s 著”。",
+                    book.getBookName(), book.getAuthorName());
+                log.debug("prompt:{}", prompt);
+                ImageResponse response = openAiImageModel.call(
+                    new ImagePrompt(prompt,
+                        OpenAiImageOptions.builder()
+                            .quality("hd")
+                            .height(800)
+                            .width(600).build())
+                );
+                Image output = response.getResult().getOutput();
+                Date currentDate = new Date();
+                String picUrl = Constants.LOCAL_PIC_PREFIX +
+                    "aiGen/" + DateUtils.formatDate(currentDate, "yyyy") + "/" +
+                    DateUtils.formatDate(currentDate, "MM") + "/" +
+                    DateUtils.formatDate(currentDate, "dd") + "/" + book.getId() + ".png";
+                FileUtil.downloadFile(output.getUrl(), picSavePath + picUrl);
+                bookMapper.update(update(BookDynamicSqlSupport.book)
+                    .set(BookDynamicSqlSupport.picUrl)
+                    .equalTo(picUrl)
+                    .set(updateTime)
+                    .equalTo(currentDate)
+                    .where(id, isEqualTo(book.getId()))
+                    .build()
+                    .render(RenderingStrategies.MYBATIS3));
+            });
+        }
     }
 
     @Override

novel-front/src/main/resources/application.yml

@@ -48,13 +48,17 @@ book:
 spring:
   ai:
     openai:
-      api-key: sk-nnhjmxuljagcuubbovjztbhkiawqaabzziazeurppinxtgva
+      image:
+        enabled: true
+        base-url: https://api.siliconflow.cn
+        api-key: sk-jjtixmivxaccndqgkqfkbgkzvmbctdxogcrfbjzfttbouitt
+        options:
+          model: Kwai-Kolors/Kolors
+          response_format: URL
+      api-key: sk-jjtixmivxaccndqgkqfkbgkzvmbctdxogcrfbjzfttbouitt
       base-url: https://api.siliconflow.cn
       chat:
         options:
           model: deepseek-ai/DeepSeek-R1-Distill-Llama-8B
 
 
-
-
-

novel-front/src/main/resources/templates/author/index.html

@@ -51,7 +51,8 @@
         </div>
         <div class="my_r">
             <div id="noContentDiv">
-                <div class="tc" style="margin-top: 200px"><a href="/author/book_add.html" class="btn_red">创建作品</a></div>
+                <div class="tc" style="margin-top: 200px"><a href="/author/book_add.html" class="btn_red">创建作品</a>
+                </div>
 
             </div>
             <div class="my_bookshelf" id="hasContentDiv" style="display: none">
@@ -142,10 +143,13 @@
 <script src="/javascript/common.js" type="text/javascript"></script>
 
 <script language="javascript" type="text/javascript">
+    var searchCount = 0;
+    var timeout;
     search(1, 5);
 
     function search(curr, limit) {
-
+        searchCount++;
+        clearTimeout(timeout);
         $.ajax({
             type: "get",
             url: "/author/listBookByPage",
@@ -155,6 +159,7 @@
                 if (data.code == 200) {
                     var bookList = data.data.list;
                     if (bookList.length > 0) {
+                        var aiPicGenerating = bookList[0].picUrl == '/images/default.gif'
                         $("#hasContentDiv").css("display", "block");
                         $("#noContentDiv").css("display", "none");
                         var bookListHtml = "";
@@ -226,6 +231,12 @@
                             });
                         });
 
+                        if (curr === 1 && aiPicGenerating && searchCount < 10) {
+                            timeout = setTimeout(function () {
+                                search(curr, limit);
+                            }, 3000);
+                        }
+
                     }