Update app.py

app.py

@@ -9,6 +9,8 @@ from io import BytesIO
 from PIL import Image, ImageFilter
 import rembg
 import onnxruntime as ort
+from concurrent.futures import ThreadPoolExecutor
+import asyncio
 
 app = FastAPI()
 
@@ -17,6 +19,9 @@ options = ort.SessionOptions()
 options.intra_op_num_threads = 2  # Limita o número de threads para evitar sobrecarga
 options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL  # Execução sequencial para melhor desempenho em CPU
 
+# Pool de threads para executar tarefas bloqueantes
+executor = ThreadPoolExecutor(max_workers=4)
+
 def resize_image(image, max_size=512):
     """Redimensiona a imagem para uma largura máxima de 512px, mantendo a proporção."""
     width, height = image.size
@@ -26,29 +31,36 @@ def resize_image(image, max_size=512):
         image = image.resize(new_size, Image.Resampling.LANCZOS)
     return image
 
+def process_image(image_url):
+    # Baixa a imagem da URL fornecida
+    response = requests.get(image_url)
+    response.raise_for_status()
+    
+    # Abre a imagem usando Pillow
+    image = Image.open(BytesIO(response.content))
+    
+    # Pré-processamento: apenas redimensiona para 512px
+    image = resize_image(image, max_size=512)
+    
+    # Remove o fundo da imagem usando rembg
+    output = rembg.remove(image, 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()
+    output.save(img_byte_arr, format='PNG')
+    img_byte_arr.seek(0)
+    
+    return img_byte_arr
+
 @app.get("/remove-background")
 async def remove_background(image_url: str):
     try:
-        # Baixa a imagem da URL fornecida
-        response = requests.get(image_url)
-        response.raise_for_status()
-        
-        # Abre a imagem usando Pillow
-        image = Image.open(BytesIO(response.content))
-        
-        # Pré-processamento: apenas redimensiona para 512px
-        image = resize_image(image, max_size=512)
-        
-        # Remove o fundo da imagem usando rembg
-        output = rembg.remove(image, 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()
-        output.save(img_byte_arr, format='PNG')
-        img_byte_arr.seek(0)
+        # Executa o processamento da imagem em um thread separado
+        loop = asyncio.get_event_loop()
+        img_byte_arr = await loop.run_in_executor(executor, process_image, image_url)
         
         # Retorna a imagem processada diretamente no navegador
         return StreamingResponse(img_byte_arr, media_type="image/png")