Update app.py

app.py

@@ -1,27 +1,32 @@
 from fastapi import FastAPI, HTTPException
-import httpx
-from rembg import remove
-from io import BytesIO
 from fastapi.responses import StreamingResponse
+import requests
+from io import BytesIO
+from PIL import Image
+import rembg
 
 app = FastAPI()
 
-@app.get("/remove-background/")
+@app.get("/remove-background")
 async def remove_background(image_url: str):
     try:
         # Baixa a imagem da URL fornecida
-        async with httpx.AsyncClient() as client:
-            response = await client.get(image_url)
-            response.raise_for_status()  # Verifica se a URL foi bem-sucedida
-            image_data = response.content
+        response = requests.get(image_url)
+        response.raise_for_status()
         
-        # Remover o fundo da imagem usando o rembg
-        output_image = remove(image_data)
-
-        # Retorna a imagem com o fundo removido como resposta
-        return StreamingResponse(BytesIO(output_image), media_type="image/png")
+        # Abre a imagem usando Pillow
+        image = Image.open(BytesIO(response.content))
+        
+        # Remove o fundo da imagem usando rembg
+        output = rembg.remove(image)
+        
+        # Converte a imagem de volta para bytes
+        img_byte_arr = BytesIO()
+        output.save(img_byte_arr, format='PNG')
+        img_byte_arr.seek(0)
+        
+        # Retorna a imagem processada diretamente no navegador
+        return StreamingResponse(img_byte_arr, media_type="image/png")
     
-    except httpx.RequestError as e:
-        raise HTTPException(status_code=400, detail="Erro ao baixar a imagem: " + str(e))
     except Exception as e:
-        raise HTTPException(status_code=500, detail="Erro ao processar a imagem: " + str(e))
+        raise HTTPException(status_code=400, detail=str(e))