
    Ti3                         d dl mZmZmZ d dlZd dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZ dd	lmZ dd
eeeef      dededeeef   fdZedk(  r ej*                   eg              yy)    )DictAnyListN)generate)save_image_to_canvas)get_image_info_and_save)	FILES_DIR)DEFAULT_PORT   )JaazServicemessages
session_id	canvas_idreturnc           	        K   	 | d   }d}t        |j                  d      t              rB|d   D ]:  }|j                  d      dk(  s|j                  di       j                  dd      } n |s	ddd	d
gdS 	 t               }|j                  |       d{   }|s	dddd
gdS |j                  d      r|d   }	t        d|	        ddd|	 d
gdS |j                  d      s	dddd
gdS d}
|d   }|}|r|r	 t        d      }t        j                  j                  t        |      }t        ||d       d{   \  }}}}t        dt        |dz              }t        dt        |dz              }| d| }
t        |||
|||       d{   }t        d|
        dd| d|
 d t"         | d!dS # t        $ r!}t        d|        dddd
gdcY d}~S d}~ww xY w7 R7 7 Y# t         $ r}t        d|        Y d}~ed}~ww xY w# t$        j&                  t         f$ r\}t)        |      j+                         }	d"|	v sd#|	v rddd$d
gdcY d}~S t        d%|        dddt)        |       d
gdcY d}~S d}~ww xY ww)&ua   
    基于云端服务的图像生成响应函数
    实现和 magic_agent 相同的功能
     contenttype	image_urlurl	assistanttextu   ✨ not found input image)r   r   )roler   u&   ❌ Jaaz service configuration error: u    ✨ Cloud API Key not configuredNu   ✨ Magic generation failederroru   ❌ Magic generation error: u   ✨ Magic Generation Error: 
result_urlu*   ✨ Magic generation failed: No result URL
   )sizeF)is_b64   r   .u   ✨ 图片已保存到画布: u!   ❌ 保存图片到画布失败: u"   ✨ Magic Success!!!

Result url: z

![image_id: z](http://localhost:)timeoutz	timed outu   ✨ time outu!   ❌ 创建魔法回复时出错: )
isinstancegetlistr   
ValueErrorprintgenerate_magic_imager   ospathjoinr	   r   maxintr   	Exceptionr
   asyncioTimeoutErrorstrlower)r   r   r   user_messageimage_contentcontent_itemjaaz_serviceeresult	error_msgfilenamer   r   file_idfile_path_without_extension	mime_typewidthheight	extensions                      h/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/services/OpenAIAgents_service/jaaz_magic_agent.pycreate_jaaz_responserC      s    
F'/|l&&y148 ,Y 7 ##F+{:$0$4$4#R%)),UB "	 # !' ; 	&=L $88GG# !' =  ::gwI0<=# !'">yk J  zz,'# !' L  L)
	 )?"+.0ggll9g.N+ =T:5= 73	5&) As519~.QFQJ0 &Ya	{3 #7z9hXachjp"qq	6xjAB
  =j\IYZbYccv  xD  wE  FO  EP  PQ  R
 	
a  
	:1#>?# !' B 
	 Hf7 r ?9!=>>?   ), FLLN		![I%=# !' .  5aS9:# !'">s1vh G s	  JAH /H 6J8
F; H G(H %J&/H JH /J0H >AG/ ?G+ AG/ G-G/ &H :J;	G%G G%H J G%%H +G/ -G/ /	H8HH HH J,)JJJ"J=J>JJJ__main__)r   r   )typingr   r   r   r0   r*   nanoidr   tools.utils.image_canvas_utilsr   tools.utils.image_utilsr   services.config_servicer	   commonr
   r7   r   r2   rC   __name__run     rB   <module>rO      s    # "  	  ? ; -  &Kd38n)= K3 K`c Kmqruwzrzm{ KZ zGKK$R() rN   