update callback

gradio_pyxelate.py

@@ -7,13 +7,22 @@
 '''
 
 import os
-import io
 from glob import glob
+from loguru import logger
 
 import gradio as gr
 from skimage import io as sio
 from pyxelate import Pyx, Pal
 
+
+# ----------- Settings -----------
+examples = sorted(glob(os.path.join('examples', '*.jpg')))
+
+# ----------- Logging -----------
+logger.add('app.log', mode='a')
+logger.info('===== APP RESTARTED =====')
+
+# ----------- Params -----------
 DOWNSAMPLE_MIN = 1
 DOWNSAMPLE_MAX = 10
 COLOR_MIN = 1
@@ -36,18 +45,30 @@ DITHER_CHOICES = ['none', 'naive', 'bayer', 'floyd', 'atkinson']
 
 
 def predict(image_obj, sampling_param, color_param, palette_param, dither_param):
-    
-    return None
+    img = sio.imread(image_obj.name)
+    logger.info('--- image loaded')
+
+    if palette_param != 'none':
+        color_param = 'none'
+        palette = Pal[palette_param]
+    else:
+        palette = color_param
+
+    pyx = Pyx(factor=sampling_param, palette=palette, dither=dither_param)
+    pyx.fit(img)
+    img_out = pyx.transform(img)
+    logger.info('--- output generated')
+    return img_out
 
 iface = gr.Interface(
     predict,
     title='이미지를 픽셀화 시키는 데모입니다.',
     description='이미지가 주어졌을 때 이미지를 스케일링하고 색감을 바꾸어 픽셀화 시킬 수 있습니다.',
     inputs=[
-        gr.inputs.Image(label='인풋 이미지를 준비해주세요'),
+        gr.inputs.Image(label='인풋 이미지를 준비해주세요', type='file'),
         gr.inputs.Slider(label='다운샘플할 크기를 입력해주세요 (클수록 픽셀크기가 커짐 = 저화질)',
                          minimum=DOWNSAMPLE_MIN, maximum=DOWNSAMPLE_MAX, step=1, default=5),
-        gr.inputs.Slider(label='사용할 색깔의 개수를 입력해주세요 (선택 시 아래 팔레트 설정은 무시됩니다)',
+        gr.inputs.Slider(label='사용할 색깔의 개수를 입력해주세요 (선택 시 아래 팔레트 설정은 "none"으로 해주세요)',
                          minimum=DOWNSAMPLE_MIN, maximum=DOWNSAMPLE_MAX, step=1, default=5),
         gr.inputs.Dropdown(label='색깔 팔레트를 선택하세요 (선택 시 위의 색깔 개수는 무시됩니다)',
                            choices=PALETTE_CHOICES, default='none', type='value'),
@@ -57,7 +78,7 @@ iface = gr.Interface(
     outputs=[
         gr.outputs.Image(label='결과 이미지 입니다')
     ],
-    # examples=examples,
+    examples=examples,
     enable_queue=True,
     article='<p style="text-align:center">Credits to <a href="https://github.com/sedthh/pyxelate">GitHub</a></p>',
 )

requirements.txt

@@ -2,4 +2,5 @@ gradio==2.4.6
 scikit-learn>=0.24.1
 scikit-image>=0.18.1
 numba>=0.53.1
+loguru==0.5.3
 git+https://github.com/sedthh/pyxelate.git