Upload 3 files

app.py

@@ -1,10 +1,9 @@
-
 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")
+        imagem_input = gr.Image(label="Imagem Original", type="numpy")
         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)")
@@ -20,10 +19,10 @@ with gr.Blocks() as demo:
 
     resultado = {}
 
-    def gerar(imagem_path, mask, idade):
+    def gerar(imagem_array, mask_array, idade):
         global resultado
-        aviso = verificar_contraste(imagem_path)
-        resultado = processar_e_mostrar(imagem_path, mask, idade)
+        aviso = verificar_contraste(imagem_array)
+        resultado = processar_e_mostrar(imagem_array, mask_array, idade)
         return aviso, resultado["preview"]
 
     def salvar_pdf():

livro.py

@@ -1,37 +1,36 @@
-
 import cv2
 import numpy as np
 from reportlab.pdfgen import canvas
 from reportlab.lib.units import inch
 import tempfile
-import os
 from PIL import Image, ImageDraw, ImageFont
 
 PAGE_WIDTH = 8.67 * inch
 PAGE_HEIGHT = 11.5 * inch
 MARGIN = 0.5 * inch
 
-def verificar_contraste(imagem_path):
-    img = cv2.imread(imagem_path)
-    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+def verificar_contraste(imagem_array):
+    gray = cv2.cvtColor(imagem_array, cv2.COLOR_BGR2GRAY)
     contrast = gray.std()
     if contrast < 30:
-        return "⚠️ Baixo contraste detectado! Use uma imagem mais nítida ou com fundo branco puro."
+        return "⚠️ Baixo contraste detectado! Use uma imagem com contorno escuro e fundo branco puro."
     return "✅ Contraste adequado."
 
-def detectar_pontos(imagem_path, mascara_path=None, idade=6):
-    img = cv2.imread(imagem_path)
-    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+def detectar_pontos(imagem_array, mascara_array=None, idade=6):
+    gray = cv2.cvtColor(imagem_array, cv2.COLOR_BGR2GRAY)
     _, thresh = cv2.threshold(gray, 200, 255, cv2.THRESH_BINARY_INV)
-    if mascara_path and os.path.exists(mascara_path):
-        mask = cv2.imread(mascara_path, 0)
-        thresh = cv2.bitwise_and(thresh, mask)
+
+    if mascara_array is not None:
+        mask = cv2.cvtColor(mascara_array, cv2.COLOR_BGR2GRAY)
+        _, mask_thresh = cv2.threshold(mask, 10, 255, cv2.THRESH_BINARY)
+        thresh = cv2.bitwise_and(thresh, mask_thresh)
+
     contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
     if not contours:
         return []
 
     main_contour = max(contours, key=cv2.contourArea)
-    max_points = min(30 + (idade - 4) * 10, 150)  # Ex: idade 4 -> 30 pts, 12 -> 110 pts
+    max_points = min(30 + (idade - 4) * 10, 150)
     approx = [tuple(pt[0]) for pt in main_contour[::max(1, len(main_contour)//max_points)]]
     return sorted(approx, key=lambda p: (p[1], p[0]))
 
@@ -54,8 +53,8 @@ def gerar_pdf(pontos):
     c.save()
     return temp_pdf.name
 
-def processar_e_mostrar(imagem_path, mascara_path, idade):
-    pontos = detectar_pontos(imagem_path, mascara_path, idade)
+def processar_e_mostrar(imagem_array, mascara_array, idade):
+    pontos = detectar_pontos(imagem_array, mascara_array, idade)
     preview = gerar_preview_com_pontos(pontos)
     pdf = gerar_pdf(pontos)
     return {"preview": preview, "pdf": pdf}
@@ -65,11 +64,10 @@ def gerar_preview_kdp():
     draw = ImageDraw.Draw(img)
     font = ImageFont.load_default()
     draw.rectangle([50, 50, 1450, 950], outline="black", width=4)
-    draw.text((600, 100), "Livro de Conectar Pontos", fill="black", font=font)
-    draw.text((580, 160), "Para crianças de 4 a 12 anos", fill="black", font=font)
+    draw.text((600, 100), "Dot-to-Dot Book", fill="black", font=font)
+    draw.text((580, 160), "For Kids Ages 4 to 12", fill="black", font=font)
     draw.ellipse((1100, 700, 1300, 900), outline="gray", width=3)
-    draw.text((1120, 780), "Sua Arte Aqui", fill="gray", font=font)
-
+    draw.text((1120, 780), "Your Art Here", fill="gray", font=font)
     capa_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
     img.save(capa_path)
     return capa_path

requirements.txt

@@ -1,5 +1,5 @@
 
-gradio
+gradio==4.21.0
 opencv-python
 numpy
 reportlab