diff --git a/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/JiMeng/JmText2Video.java b/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/JiMeng/JmText2Video.java index 760ed09b..9074d438 100644 --- a/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/JiMeng/JmText2Video.java +++ b/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/JiMeng/JmText2Video.java @@ -2,6 +2,7 @@ package com.dsideal.aiSupport.Util.JiMeng; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; +import com.jfinal.kit.StrKit; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -30,23 +31,29 @@ public class JmText2Video extends JmCommon { public static void main(String[] args) throws Exception { String prompt = "蓝色毛绒玩具在超市里拖地,结果拖把洒出好多五颜六色的粉末,接着把粉末洒向镜头前,镜头随之穿过粉末"; JSONObject jo = JmText2Video.generateVideo(prompt); + log.info("结果:{}", jo); String taskId = jo.getJSONObject("data").getString("task_id"); //检查任务是不是已经结束 while (true) { JSONObject result = JmCommon.queryTaskResult(taskId); + log.info("查询结果:{}", result); if (result.getInteger("code") != 10000) { log.error("查询失败,错误代码=" + result.getInteger("code")); break; } JSONObject data = result.getJSONObject("data"); - String status = data.getString("status"); - if (status.equals("done")) { + + if (!StrKit.isBlank(data.getString("video_url"))) { String video_url = data.getString("video_url"); log.info("视频地址:{}", video_url); break; } else { - log.info("任务处理中..."); - Thread.sleep(5000); + if (data.getString("status").equals("in_queue")) { + log.info("任务排队中..."); + } else { + log.info("任务已经开始处理"); + } + Thread.sleep(3000); } } }