o S6iA @sdZddlZddlmZddlmZddlmZddlm Z m Z m Z dd Z d e eBd efd d Zde eBde de de eBd ef ddZd efddZdS)z4Gemini API client for image analysis and generation.N)Path)genai)types)GOOGLE_API_KEY GEMINI_MODELGEMINI_IMAGE_MODELcCs tjtdS)zGet Gemini client instance.)Zapi_key)rZClientrr r P/var/www/tkim.planitai.co.jp/blog/20251208-make-apng-tool/./src/gemini_client.py get_client s r image_pathreturncCst}t|}t|d }|}Wdn1swYtjj|d|jddd}d}|j j t ||gd}|j t |dS) z Analyze a face image to extract features for lip sync generation. Args: image_path: Path to the image file Returns: dict with face analysis results rbNimage/rdata mime_typea^Analyze this face image for lip sync animation creation. Please identify and describe: 1. Face position and angle (frontal, 3/4 view, etc.) 2. Mouth region location and shape 3. Art style (realistic photo, anime, cartoon, etc.) 4. Skin tone and lip color 5. Any distinctive features around the mouth area Return the analysis in a structured format.modelcontents)Zanalysisr )r ropenreadrPart from_bytessuffixlowermodelsgenerate_contentrtextstr)r clientf image_data image_partpromptresponser r r analyze_face_images"    r&original_image_pathvisemeviseme_description output_pathc Cs0t}t|}t|}t|d }|}Wdn1swYtjj|d|jddd}d|d|d|d }|j j t ||gtj d d gd d } | j djjD]9} | jr| jjdr| jj} |jjdddt|d}|| Wd|S1swY|SqZtd)ag Generate a viseme (mouth shape) variation of the original image. Args: original_image_path: Path to the original face image viseme: Viseme code (A, E, I, O, U, etc.) viseme_description: Description of the mouth shape output_path: Path to save the generated image Returns: Path to the generated image rNrrrzJEdit this face image to change ONLY the mouth shape. Target mouth shape: z - z Instructions: 1. Keep the face, eyes, hair, and all other features exactly the same 2. Only modify the mouth/lips area 3. The mouth should show: z 4. Maintain the same art style and quality 5. The transition should look natural Generate the edited image with the new mouth shape.ZIMAGEZTEXT)Zresponse_modalities)rrconfigrT)Zparentsexist_okwbzNo image generated in response)r rrrrrrrrrrrZGenerateContentConfigZ candidatesZcontentZpartsZ inline_datar startswithrparentmkdirwrite ValueError) r'r(r)r*r r!r"r#r$r%partZ image_bytesr r r generate_viseme_image;sH       r4c Csbzt}|jjtdd}td|jWdSty0}ztd|WYd}~dSd}~ww)zTest the Gemini API connection.z,Say 'API connection successful' in one line.rz API Test: TzAPI Test Failed: NF)r rrrprintr Exception)r r%er r r test_connection{sr8)__doc__base64ZpathlibrZgooglerZ google.genairr+rrrr rdictr&r4boolr8r r r r s(   - @