Ti@XddlZddlZddlmZmZmZmZddlmZddl m Z GddZ y)N)DictAnyOptionalList) HttpClient)config_serviceceZdZdZdZdefdZdeeeffdZ dedefdZ dd ed ed e ed e e d e ede e ededefdZ ddede e de edeeeffdZdede eeeffdZ dd ed ed e ed e e d e ede e ededeeeffdZ dd ed ed ed e d ede e ededeeeffdZ dd ed ededefdZ dd ed ededeeeffdZdefdZy) JaazServiceuJaaz 云端 API 服务 ctjjdi}t|jddj d|_t|jdd|_|j s td|j s td|j jds|j d|_td |j y ) u初始化 Jaaz 服务jaazurl/api_keyzJaaz API URL is not configuredz Jaaz API token is not configuredz/api/v1u+✅ Jaaz service initialized with API URL: N) r app_configgetstrrstripapi_url api_token ValueErrorendswithprint)selfconfigs O/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/services/jaaz_service.py__init__zJaazService.__init__s**..vr:6::eR0188= VZZ 267||=> >~~?@ @||$$Y/"ll^73DL ;DLL>JKreturncHt|jxr |jS)u检查 Jaaz API 是否已配置)boolrrrs r_is_configuredzJaazService._is_configuredsDLL3T^^44rc&d|jddS)u构建请求头zBearer zapplication/json) Authorizationz Content-Type)rr"s r_build_headerszJaazService._build_headers#s  't~~&67.  r image_contentc K |r|jds tdytj4d{}|j |j d|j d|itjd 4d{}|jd k(r|jd{}|jd d}|r2td ||cdddd{cdddd{Std  dddd{dddd{y|jd{}td|jd| dddd{dddd{y7C77777u7g7P7&7#1d{7swYnxYwdddd{7y#1d{7swYyxYw#t$r}td|Yd}~yd}~wwxYww)u 创建云端魔法图像生成任务 Args: image_content: 图片内容（base64 或 URL） Returns: str: 任务 ID，失败时返回空字符串 z data:image/u ❌ Invalid image content formatrNz /image/magicimageN@totalheadersjsontimeouttask_idu✅ Magic task created: u❌ No task_id in responseu!❌ Failed to create magic task:  - u❌ Error creating magic task: ) startswithrrcreate_aiohttppostrr&aiohttp ClientTimeoutstatusr/rtext Exception)rr'sessionresponsedatar2 error_textes rcreate_magic_taskzJaazService.create_magic_task*s  (@(@(O89!002 " "g"<<||nL1 //1$11= (""#-%-]]_4"&((9b"9"!$?#%!"" " " "&,4==?%: ??PPST^S_`b!+"" " " "" 5" "" "&&;%" """" " " " " "0  3A37 8 sG3GG3GE>GA F: F F:#F1F2'F F:%F&F:* G6F7G;G3< F F:F F: G#F $G(G3)F<F = F F:(F)F:- G8F9G=G3>GF:FF:G F: G FF:GF% FF% !F:( G3F64G9G3:G G G G G3 G G0G+&G3+G00G3Npromptmodel resolutionduration aspect_ratio input_imageskwargsc Ktj4d{}|||||d|} |r|| d<|j|jd|j | t j d4d{} | jdk(ri| jd{} | jd d } | r2td | | cdddd{cdddd{Std | jd{} td | jd| 7777[7L7*#1d{7swYnxYwdddd{7y#1d{7swYyxYww)u 创建云端视频生成任务 Args: prompt: 视频生成提示词 model: 视频生成模型 resolution: 视频分辨率 duration: 视频时长（秒） aspect_ratio: 宽高比 input_images: 输入图片列表（可选） **kwargs: 其他参数 Returns: str: 任务 ID Raises: Exception: 当任务创建失败时抛出异常 NrBrCrDrErFrGz/video/sunra/generations^@r+r-r1r2ru✅ Video task created: No task_id in responsez"Failed to create video task: HTTP r3 rr5r6rr&r7r8r9r/rrr;r:)rrBrCrDrErFrGrHr<payloadr=r>r2r?s rcreate_video_taskzJaazService.create_video_taskUs8,,. k k' ($ ,   G*6'||<<. 89++---E: $ k k ??c)!)0D"hhy"5G 8 BC& k k k k k k4((@AA'/}}!6J#&HHYY\]g\h$ijj; k k1 k k8"7 k k k k k k k ksE2D)E2AE9D,:E=#D6 D.!'D6 ED0E E2%D2&E2+D6 D4 D6)E2,E.D60E2E24D66E <D?=E E E2EE2E/#E& $E/+E2r2 max_attemptsintervalc >K|xsd}|xsd}tj4d{}t|D]}|j|jd||j t jd4d{}|jdk(r |jd{}|jd r|jd ijd r|d d }|jd } | dk(r5td|d|cdddd{ccdddd{S| dk(r |jdd} td| | dk(r td| dk(r0tj|d{ dddd{itd| tdtd|jtd|d77`7;777e7V#1d{7swYxYw#1d{7swYyxYww)uE 等待任务完成并返回结果 Args: task_id: 任务 ID max_attempts: 最大轮询次数 interval: 轮询间隔（秒） Returns: Dict[str, Any]: 任务结果 Raises: Exception: 当任务失败或超时时抛出异常 @Nz/task/g4@r+)r.r0r1successr>foundtaskr9 succeededu ✅ Task z completed successfullyfailederrorz Unknown errorz Task failed: cancelledzTask was cancelled processingzUnknown task status: zTask not foundz Failed to get task status: HTTP zTask polling timeout after z attempts)rr5rangerrr&r7r8r9r/rr;asynciosleep) rr2rPrQr<_r=r>rWr9 error_msgs rpoll_for_task_completionz$JaazService.poll_for_task_completionsM($*s ?s,,.! S! S'<( ^";;||nF7)4 //1#11='^^#-%-]]_488I.488FB3G3K3KG3T#'<#7D%)XXh%7F%4 %&/y8O$P!R'+^^^! S! S! S""(8!3,0HHWo,N &/- {0K&L L!';!6&/0D&E E!'-^^^^! S! S! Ss%HG!HAHG$ H$G2,G' -A$G2 HG* H$ H0G,1H6AG2G. G2 HG0H 1G2H!H$H'G2*H,H.G20H2H 8G;9H HHH HHcK |j|d{}|stdddiS|j|ddd{}|stddd iS|jd s'|jdd }td |dd |iStd|jd |S77n#t$r*}dt |}td|d|icYd}~Sd}~wwxYww)u 生成魔法图像的完整流程 Args: image_content: 图片内容（base64 或 URL） Returns: Dict[str, Any]: 包含 result_url 的任务结果，失败时返回包含 error 信息的字典 Nu❌ Failed to create magic taskrZzFailed to create magic taskxg@rPrQu❌ Magic generation failedzMagic generation failed result_urlzNo result URL foundu❌ Magic generation failed: zMagic generation failed: u(✅ Magic image generated successfully: z!Error in magic image generation: u❌ )rArrbrr;r)rr'r2resultrar@s rgenerate_magic_imagez JaazService.generate_magic_images ( 22=AAG78!>?? 88s]`8aaF34!:;;::l+"JJw0EF 5i[AB#s"B8B4%B8B6A/B86B8c Ktj4d{}|||||d|} |r|| d<|j|jd|j | t j d4d{} | jdk(r7| jd{} | jd d } | s>td | jd{} td | jd | dddd{dddd{td |j| d{}|s td|jdrtd|d|jds tdtd|jd|S772777#1d{7swYxYw7#1d{7swYxYw7w)u 使用 Seedance 模型生成视频的完整流程 Args: prompt: 视频生成提示词 model: 视频生成模型 resolution: 视频分辨率 duration: 视频时长（秒） aspect_ratio: 宽高比 input_images: 输入图片列表（可选） **kwargs: 其他参数 Returns: Dict[str, Any]: 包含 result_url 的任务结果 Raises: Exception: 当视频生成失败时抛出异常 NrJrGz/video/seedance/generationrKr+r-r1r2rrLz+Failed to create Seedance video task: HTTP r3u!✅ Seedance video task created: z Seedance video generation failedrZz"Seedance video generation failed: rfz9No result URL found in Seedance video generation responseu+✅ Seedance video generated successfully: )rr5r6rr&r7r8r9r/rr;r:rrb)rrBrCrDrErFrGrHr<rNr=r>r2r?rgs rgenerate_video_by_seedancez&JaazService.generate_video_by_seedance(s :,,. t t' ($ ,   G*6'||<<. :;++---E: $ t t ??c)!)0D"hhy"5G"'(@AA'/}}!6J#&QRZRaRaQbbefpeq$rss t t t t8 1';<44W==>? ? ::g @@QRS Szz,'WX X 9&**\:R9S T V W t t1 "7 t t t t t t t t>>sG#F(G#AG 9F+:G =#F5 F.!6F5F1F57 G F3G  G#G &G#9G!:A/G#+G .F51F53G 5G ;F><G G  G# GG GG#c Ktj4d{}||d|}|j|jd|j |t j d4d{}|jdk(ri|jd{}|jdd }|r2td ||cdddd{cdddd{Std |jd{} td |jd | 7777[7L7*#1d{7swYnxYwdddd{7y#1d{7swYyxYww)ud 创建云端 Midjourney 图像生成任务 Args: prompt: 图像生成提示词 model: 图像生成模型（默认为 midjourney） **kwargs: 其他参数（如 mode 等） Returns: str: 任务 ID Raises: Exception: 当任务创建失败时抛出异常 NrBrCz/image/midjourney/generationr*r+r-r1r2ru✅ Midjourney task created: rLz'Failed to create Midjourney task: HTTP r3rM) rrBrCrHr<rNr=r>r2r?s rcreate_midjourney_taskz"JaazService.create_midjourney_taskrs(,,. p p' G ||<<. <=++---D9 $ p p ??c)!)0D"hhy"5G =gYGH& p p p p p p(((@AA'/}}!6J#&MhooM^^ablam$noo/ p p1 p p,"7 p p p p p p p psE(DE(AE/D"0E3#D,D$'D,> E D& E E(D(E(!D,?D*D,E("E$D,&E(E(*D,,D> 2D53D> :E E( E E(E%E E%!E(cK|jd||d|d{}|s td|j|ddd{}td||s td|j d rtd |d |j d s td |j d }td ||xsiS77w)u 使用 Midjourney 生成图像的完整流程 Args: prompt: 图像生成提示词 model: 图像生成模型（默认为 midjourney） **kwargs: 其他参数（如 mode 等） Returns: Dict[str, Any]: 包含 result_url 的任务结果 Raises: Exception: 当图像生成失败时抛出异常 roNz Failed to create Midjourney taskrSrTreu🎨 Midjourney task result: z"Midjourney image generation failedrZz$Midjourney image generation failed: rgz7No result found in Midjourney image generation responseu-✅ Midjourney image generated successfully: rj)rpr;rbrr)rrBrCrHr2 task_resultrgs rgenerate_image_by_midjourneyz(JaazService.generate_image_by_midjourneys*433     >? ?!99'PS^a9bb  -k];<@A A ??7 #B;wCWBXYZ Zx(UV V* =fXFG|/ cs"C C(C C BC  C c"|jS)uj 检查服务是否已正确配置 Returns: bool: 配置是否有效 )r#r"s r is_configuredzJaazService.is_configureds""$$r)NNNN)NN)480pz16:9N) midjourney)__name__ __module__ __qualname____doc__rr!r#rrr&rArintrrrOfloatrbrhrkrmrprsrurjrrr r sL"55 S#X )S)S)^%)"&&*,09k9k9kSM 9k 3- 9k sm 9ktCy)9k9k 9k|'+$( 8S8Ssm8S5/ 8S c3h 8St#(#(cSVh@X#(R%)"&&*,0777SM 7 3- 7 sm 7tCy)77 c3h7z!",0HHH H  H  HtCy)HH c3hHZ"+p+p+p +p +p`",,, , c3h ,\%t%rr ) r^r7typingrrrrutils.http_clientrservices.config_servicerr rjrrrs%,,(2J%J%r