filter

core/src/main/java/com/tzld/supply/job/ContentScreenJob.java

@@ -145,7 +145,7 @@ public class ContentScreenJob {
         if (Objects.isNull(obj.getIsSuitable()) || !obj.getIsSuitable()) {
             return SpiderContentScreenStatusEnum.ABANDONED.getCode();
         }
-        if (obj.getPropagationScore() < 0.7 || obj.getSuitabilityScore() < 0.8) {
+        if (obj.getPropagationScore() < 0.8 || obj.getSuitabilityScore() < 0.9) {
             return SpiderContentScreenStatusEnum.ABANDONED.getCode();
         }
         return SpiderContentScreenStatusEnum.PASSED.getCode();

server/src/test/java/ProduceTest.java

@@ -2,12 +2,10 @@ import cn.hutool.core.collection.CollectionUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.tzld.supply.Application;
 import com.tzld.supply.api.FFmpegApiService;
-import com.tzld.supply.dao.mapper.supply.spider.ProduceVideoAudioMapper;
-import com.tzld.supply.dao.mapper.supply.spider.ProduceVideoMapper;
-import com.tzld.supply.dao.mapper.supply.spider.SubtitleStyleMapper;
-import com.tzld.supply.dao.mapper.supply.spider.ToolsAudioTransRecordMapper;
+import com.tzld.supply.dao.mapper.supply.spider.*;
 import com.tzld.supply.job.ContentMediaSearchJob;
 import com.tzld.supply.job.VideoGenerateJob;
+import com.tzld.supply.model.entity.PrecisionScreenEntity;
 import com.tzld.supply.model.entity.ali.AliVoiceResultSentenceData;
 import com.tzld.supply.model.param.FFmpeg.VideoAddAssSubtitleParam;
 import com.tzld.supply.model.param.FFmpeg.VideoInfoParam;
@@ -44,6 +42,8 @@ public class ProduceTest {
     private SubtitleStyleMapper subtitleStyleMapper;
     @Autowired
     private ToolsAudioTransRecordMapper toolsAudioTransRecordMapper;
+    @Autowired
+    private SpiderContentMapper spiderContentMapper;
 
     @Test
     public void test() {
@@ -88,7 +88,7 @@ public class ProduceTest {
         if (CollectionUtil.isEmpty(audioList)) {
             return;
         }
-        ProduceVideoAudio audio =  audioList.get(0);
+        ProduceVideoAudio audio = audioList.get(0);
         toolsAudioTransService.getAudioTransResults(produceVideoId, audio.getUrl(), true);
     }
 
@@ -99,7 +99,7 @@ public class ProduceTest {
         SubtitleStyle subtitleStyle = subtitleStyleMapper.selectByPrimaryKey("subtitle_style_109");
         int lineWidth = 80;
         String mediaInfoStr = ffmpegApiService.videoInfo(new VideoInfoParam(videoUrl));
-        VideoGenerateJob.MediaInfo mediaInfo =  JSONObject.parseObject(mediaInfoStr, VideoGenerateJob.MediaInfo.class);
+        VideoGenerateJob.MediaInfo mediaInfo = JSONObject.parseObject(mediaInfoStr, VideoGenerateJob.MediaInfo.class);
         if (mediaInfo.getWidth() > mediaInfo.getHeight()) {
             lineWidth = 220;
         }
@@ -131,4 +131,18 @@ public class ProduceTest {
         String addAssSubtitleVideoUrl = ffmpegApiService.videoAddAssSubtitle(addAssSubtitleParam);
         System.out.println(addAssSubtitleVideoUrl);
     }
+
+    @Test
+    public void getSpiderContentScore() {
+        SpiderContentExample example = new SpiderContentExample();
+        example.createCriteria().andStatusEqualTo(2).andAiPrecisionStatusEqualTo(1).andCreateTimeGreaterThan(1764086400000L);
+        List<SpiderContent> contentList = spiderContentMapper.selectByExampleWithBLOBs(example);
+        for (SpiderContent spiderContent : contentList) {
+            PrecisionScreenEntity aiPrecisionResult = JSONObject.parseObject(spiderContent.getAiPrecisionResult(), PrecisionScreenEntity.class);
+            System.out.println("title: " + spiderContent.getTitle()
+                    + ", suitabilityScore: " + aiPrecisionResult.getSuitabilityScore()
+                    + ", propagationScore: " + aiPrecisionResult.getPropagationScore()
+                    + ", keywords: " + aiPrecisionResult.getKeywords());
+        }
+    }
 }