| import gradio as gr | |
| from livro import processar_e_mostrar, verificar_contraste, gerar_preview_kdp | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| desenho = gr.ImageEditor(label="Desenhe aqui sua imagem com caneta", type="numpy", interactive=True) | |
| with gr.Row(): | |
| idade = gr.Slider(minimum=4, maximum=12, step=1, value=6, label="Idade da criança (ajusta número de pontos)") | |
| posicao = gr.Radio(["Topo da página", "Centro da página", "Base da página"], value="Centro da página", label="Posicionamento do desenho") | |
| contraste_alerta = gr.Textbox(label="Aviso de Contraste", interactive=False) | |
| with gr.Row(): | |
| imagem_resultado = gr.Image(label="Prévia com Pontos Numerados") | |
| sobreposicao = gr.Image(label="Preview com Sobreposição") | |
| with gr.Row(): | |
| gerar_btn = gr.Button("Gerar Pontos") | |
| salvar_btn = gr.Button("Salvar como PDF") | |
| salvar_png = gr.Button("Salvar como PNG") | |
| capa_btn = gr.Button("Gerar Preview da Capa") | |
| saida_pdf = gr.File(label="Download PDF") | |
| saida_png = gr.File(label="Download PNG") | |
| saida_capa = gr.Image(label="Preview da Capa") | |
| resultado = {} | |
| def gerar(desenho_array, idade, posicao): | |
| global resultado | |
| if desenho_array is None or desenho_array.sum() < 1000: | |
| return "⚠️ Por favor, desenhe uma imagem para gerar os pontos.", None, None | |
| aviso = verificar_contraste(desenho_array) | |
| try: | |
| resultado = processar_e_mostrar(desenho_array, None, idade, posicao) | |
| return aviso, resultado["preview"], resultado["overlay"] | |
| except Exception as e: | |
| return f"Erro: {str(e)}", None, None | |
| def salvar_pdf(): | |
| return resultado.get("pdf") if resultado else None | |
| def salvar_imagem(): | |
| return resultado.get("png") if resultado else None | |
| def gerar_capa(): | |
| return gerar_preview_kdp() | |
| gerar_btn.click(gerar, inputs=[desenho, idade, posicao], outputs=[contraste_alerta, imagem_resultado, sobreposicao]) | |
| salvar_btn.click(salvar_pdf, outputs=saida_pdf) | |
| salvar_png.click(salvar_imagem, outputs=saida_png) | |
| capa_btn.click(gerar_capa, outputs=saida_capa) | |
| demo.launch() | |