| import gradio as gr | |
| from livro import processar_e_mostrar, verificar_contraste, gerar_preview_kdp | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| imagem_input = gr.Image(label="Imagem Original", type="filepath") | |
| mascara = gr.ImageEditor(label="Marcar áreas para preservar (opcional)") | |
| with gr.Row(): | |
| idade = gr.Slider(minimum=4, maximum=12, step=1, value=6, label="Idade da criança (ajusta número de pontos)") | |
| contraste_alerta = gr.Textbox(label="Aviso de Contraste", interactive=False) | |
| with gr.Row(): | |
| imagem_resultado = gr.Image(label="Prévia com Pontos Numerados") | |
| with gr.Row(): | |
| gerar_btn = gr.Button("Gerar Pontos") | |
| salvar_btn = gr.Button("Salvar como PDF") | |
| capa_btn = gr.Button("Gerar Preview da Capa") | |
| saida_pdf = gr.File(label="Download do PDF") | |
| saida_capa = gr.Image(label="Preview da Capa") | |
| resultado = {} | |
| def gerar(imagem_path, mask, idade): | |
| global resultado | |
| aviso = verificar_contraste(imagem_path) | |
| resultado = processar_e_mostrar(imagem_path, mask, idade) | |
| return aviso, resultado["preview"] | |
| def salvar_pdf(): | |
| return resultado.get("pdf") | |
| def gerar_capa(): | |
| return gerar_preview_kdp() | |
| gerar_btn.click(gerar, inputs=[imagem_input, mascara, idade], outputs=[contraste_alerta, imagem_resultado]) | |
| salvar_btn.click(salvar_pdf, outputs=saida_pdf) | |
| capa_btn.click(gerar_capa, outputs=saida_capa) | |
| demo.launch() | |