o
    ˆT6i­  ã                   @   s|   d Z ddlZej dd¡ ddlmZ ddlmZmZm	Z	 defdd„Z
d	d
„ Zedkr<eƒ Zeder5dnd› ƒ dS dS )z+Test script for Gemini image analysis - v2.é    NÚ.)ÚPath)ÚImageÚ	ImageDrawÚ	ImageFontÚoutput_pathc                 C   s  t jdddd}t |¡}|jg d¢dddd	 |jg d
¢ddd |jg d¢ddd |jg d¢dd |jg d¢dd |jg d¢ddddd |jg d¢ddddd |jg d¢ddd |jg d¢ddddd |jg d¢dddd d | jjd!d!d" | 	| ¡ t
d#| › ƒ | S )$z)Create a simple cartoon face for testing.ÚRGB)é  r	   z#FFE4C4)Úcolor)é2   r   é^  r   z#FFDAB9z#8B4513é   )ÚfillÚoutlineÚwidth)éx   éŒ   éª   é¾   Zwhitez#333)r   r   )éæ   r   i  r   )é‡   é›   r   é¯   )r   )éõ   r   i	  r   )én   és   é´   r   r   r   )r   r   )éÜ   r   i"  r   ))éÈ   r   )é¹   éð   )é×   r    z#CD853FN)r   r   )r   r    i  i6  z#CD5C5Cé   )r   é   r   é–   ih  é   T)ÚparentsÚexist_okzCreated test image: )r   Únewr   ZDrawZellipseZarcZpolygonÚparentÚmkdirÚsaveÚprint)r   ÚimgZdraw© r.   úR/var/www/tkim.planitai.co.jp/blog/20251208-make-apng-tool/./test_image_analysis.pyÚcreate_test_face_image
   s    

r0   c               
   C   s®   ddl m}  tdƒ}t|ƒ tdƒ tdƒ tdƒ z"| |ƒ}tdƒ tdƒ t|d	 ƒ tdƒ td
|d › ƒ W dS  tyV } ztd|› ƒ W Y d}~dS d}~ww )z#Test the face analysis with Gemini.r   )Úanalyze_face_imagezuploads/test_cartoon_face.pngz3
==================================================zTesting Gemini Face Analysisz2==================================================z
Analysis Result:z(----------------------------------------Zanalysisz
Image path: Z
image_pathTzError during analysis: NF)Zsrc.gemini_clientr1   r   r0   r,   Ú	Exception)r1   Ztest_image_pathÚresultÚer.   r.   r/   Útest_face_analysis+   s&   €þr5   Ú__main__Ú
u   âœ“ Test passed!u   âœ— Test failed!)Ú__doc__ÚsysÚpathÚinsertÚpathlibr   ÚPILr   r   r   r0   r5   Ú__name__Úsuccessr,   r.   r.   r.   r/   Ú<module>   s   !þ