import gradio as gr
from kumikolib import Kumiko
import json

k = Kumiko({
	'debug': False,
	'progress': True,
	'rtl': False,
	'min_panel_size_ratio': False
})

def panel_extract(input_img, margin_height):
    info = k.parse_image(input_img)
    print(info)
    # info = json.loads(info)
    cut_images = k.cut_rect(input_img, info, int(margin_height))
    rect_images = k.draw_rect(input_img, info)
     
    return rect_images, cut_images

css_code = """
    .output-image {
        width: 800px;
        height: 1000px;
    }
    .input-image-example {
        width: 200px;
        height: 300px;
    }

"""

demo = gr.Interface(
    fn=panel_extract, 
    inputs=["image", gr.inputs.Number(default=10)],
    outputs=["image", "image"],
    css=css_code,
    examples=[
        ["examples/01.jpg", 10],
        ["examples/02.jpg", 10],
        ["examples/03.jpg", 10],
        ["examples/04.jpg", 10],
        ["examples/05.jpg", 10],
    ]
)

if __name__ == "__main__":    
    demo.close()
    demo.launch(share=True, show_error=True)