o
    bf0                     @   s`   d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dl	m
Z
 d dlZG dd deZdS )	    )render)JsonResponse)ViewSet)GeneratorFromStringsN)BASE_DIRc                   @   s   e Zd Zdd Zdd ZdS )FontGeneratorViewSetc              
   C   s   z!t |j t|j}|d |d }}| || tddiW S  ty@ } zt d| tdt|iW  Y d }~S d }~ww )NtextfontstatussuccesszERROR:ERROR)printbodyjsonloadsfont_generatorr   	Exceptionstr)selfrequestZrequest_datar   r	   e r   C/Users/leeseokho/Desktop/Study/Cat_ch/server/fontGenerator/views.pycreate
   s   

zFontGeneratorViewSet.createc                 C   s.  t jtd|d }t jtd|}td| t j|s$td| t j|dd t|D ]e\}}t	|g|gd}z.t
|\}}	|d u rMtd	| | d
| d}
t j||
}|| td|  W q/ tyy   td|  Y q/ ty } ztd| d|  W Y d }~q/d }~ww d S )NzfontGenerator/fonts/ko/z.ttfzfontGenerator/train_images/z
Font path:zFont file not found: T)exist_ok)stringsZfontsz'Generated image is None for character: Z_test_font_img_z.pngzImage saved: z(Failed to generate image for character: z&Error generating image for character 'z': )ospathjoinr   r   isfileFileNotFoundErrormakedirs	enumerater   next
ValueErrorsaveStopIterationr   )r   r   r	   Z	font_pathZtrain_img_pathindexchar	generatorZgenerated_image_Zoutput_file_nameZoutput_file_pathr   r   r   r   r      s4   

 z#FontGeneratorViewSet.font_generatorN)__name__
__module____qualname__r   r   r   r   r   r   r   	   s    r   )django.shortcutsr   django.httpr   rest_framework.viewsetsr   Ztrdg.generatorsr   r   Zcat_ch.settingsr   r   r   r   r   r   r   <module>   s    