o
    )e	                     @   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	Z	d dl
Z
d dlmZ d dlZeedZeed ZW d   n1 sEw   Y  G d	d
 d
eZdS )    )render)View)JsonResponse)KAKAOBRAIN_KEY_PATHN)Imagerkeyc                   @   s$   e Zd Zdd Zdd Zdd ZdS )	karloViewc                 C   s0   t jdd|idt ddd}t|j}|S )Nz8https://api.kakaobrain.com/v2/inference/karlo/variationsimagezKakaoAK zapplication/json)AuthorizationzContent-Type)jsonheaders)requestspostREST_API_KEYr   loadscontent)selfr
   r   response r   -/home/api_server/kakao_chatbot/karlo/views.py
variations   s   
zkarloView.variationsc                 C   s0   t  }|j||jd t| d}|S )N)formatascii)ioBytesIOsaver   base64encodebytesgetvaluedecode)r   imgimg_byte_arrmy_encoded_imgr   r   r   imageToString   s   zkarloView.imageToStringc                 C   s   t |j}|d d d }t |}|dd}|dr)|dr)|dd	 }|s1td
diS t|}|jdkrlt	
t|j}t|j | |}| |}|d d d }	dddd|	 iigid}
t|
S td
diS )Nactionparamskarlo
secureUrls zList()   messagezCANNOT FIND IMG URL   imagesr   r
   z2.0outputssimpleImageimageUrl)versiontemplatezIMGURL CONNECTION ERROR)r   r   bodyget
startswithendswithr   r   status_coder   openr   r   r   printr   r$   r   )r   requestdatasecure_urls_strsecure_urls_dictsecure_urlsr   r!   
img_base64	image_urlresponseBodyr   r   r   r   "   s4   





zkarloView.postN)__name__
__module____qualname__r   r$   r   r   r   r   r   r	      s    r	   )django.shortcutsr   django.viewsr   django.httpr   kakao_chatbot.settingsr   r   r   r   PILr   r   r:   floadr   r	   r   r   r   r   <module>   s    8