Update routers/textclas.py

routers/textclas.py

@@ -1,19 +1,40 @@
 from fastapi import APIRouter, HTTPException, Query
 from fastapi.responses import JSONResponse
-from PIL import Image
+import cv2
+import numpy as np
 import requests
-import base64
+from PIL import Image
 from io import BytesIO
+import base64
 
 # Criação do roteador
 router = APIRouter()
 
-@router.get("/image/base64/")
-def get_image_as_base64(
-    image_url: str = Query(..., description="URL da imagem para ser processada")
-):
+
+def remove_handwritten_content(image: np.ndarray) -> np.ndarray:
     """
-    Recebe uma URL de imagem, converte para base64 e retorna no navegador.
+    Remove conteúdo manuscrito de uma imagem usando técnicas do OpenCV.
+    """
+    # Converte a imagem para escala de cinza
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+
+    # Aplica threshold para detectar manuscritos
+    _, thresh = cv2.threshold(gray, 180, 255, cv2.THRESH_BINARY_INV)
+
+    # Aplica dilatação para conectar as partes manuscritas
+    kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (3, 3))
+    dilated = cv2.dilate(thresh, kernel, iterations=2)
+
+    # Aplica inpainting para remover manuscritos
+    result = cv2.inpaint(image, dilated, inpaintRadius=3, flags=cv2.INPAINT_TELEA)
+    
+    return result
+
+
+@router.get("/image/remove-handwriting/")
+def process_image(image_url: str = Query(..., description="URL da imagem para ser processada")):
+    """
+    Recebe uma URL de imagem, remove conteúdo manuscrito e retorna em base64.
     """
     try:
         # Baixar a imagem da URL
@@ -22,21 +43,25 @@ def get_image_as_base64(
             raise HTTPException(
                 status_code=400, detail="Não foi possível baixar a imagem da URL fornecida."
             )
-        
-        # Abrir a imagem com Pillow
-        image = Image.open(BytesIO(response.content))
-        
-        # Converter a imagem para base64
-        buffered = BytesIO()
-        image.save(buffered, format=image.format if image.format else "PNG")
-        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        
-        # Retornar a imagem em base64
+
+        # Carregar imagem com OpenCV
+        image_array = np.asarray(bytearray(response.content), dtype=np.uint8)
+        image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
+        if image is None:
+            raise HTTPException(status_code=400, detail="A URL não contém uma imagem válida.")
+
+        # Remover manuscrito
+        processed_image = remove_handwritten_content(image)
+
+        # Converter imagem processada para base64
+        _, buffer = cv2.imencode('.png', processed_image)
+        img_base64 = base64.b64encode(buffer).decode('utf-8')
+
         return JSONResponse(content={
             "image_url": image_url,
-            "image_base64": f"data:image/{image.format.lower()};base64,{img_base64}"
+            "processed_image_base64": f"data:image/png;base64,{img_base64}"
         })
-    
+
     except requests.exceptions.RequestException as e:
         raise HTTPException(status_code=400, detail=f"Erro ao acessar a URL: {str(e)}")
     except Exception as e: