Update app.py

app.py

@@ -6,11 +6,31 @@ from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
 import requests
 from io import BytesIO
-from PIL import Image
+from PIL import Image, ImageFilter, ImageEnhance
 import rembg
+import onnxruntime as ort
 
 app = FastAPI()
 
+# Limita o número de threads do onnxruntime
+options = ort.SessionOptions()
+options.intra_op_num_threads = 2
+
+def resize_image(image, max_size=1024):
+    width, height = image.size
+    if width > max_size or height > max_size:
+        ratio = min(max_size / width, max_size / height)
+        new_size = (int(width * ratio), int(height * ratio))
+        image = image.resize(new_size, Image.ANTIALIAS)
+    return image
+
+def adjust_brightness_contrast(image, brightness=1.2, contrast=1.2):
+    enhancer = ImageEnhance.Brightness(image)
+    image = enhancer.enhance(brightness)
+    enhancer = ImageEnhance.Contrast(image)
+    image = enhancer.enhance(contrast)
+    return image
+
 @app.get("/remove-background")
 async def remove_background(image_url: str):
     try:
@@ -21,8 +41,15 @@ async def remove_background(image_url: str):
         # Abre a imagem usando Pillow
         image = Image.open(BytesIO(response.content))
         
+        # Pré-processamento: redimensiona e ajusta brilho/contraste
+        image = resize_image(image, max_size=1024)
+        image = adjust_brightness_contrast(image)
+        
         # Remove o fundo da imagem usando rembg
-        output = rembg.remove(image)
+        output = rembg.remove(image, model="u2net", session_options=options)
+        
+        # Pós-processamento: suaviza as bordas
+        output = output.filter(ImageFilter.SMOOTH_MORE)
         
         # Converte a imagem de volta para bytes
         img_byte_arr = BytesIO()