refresh files

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import Response
+import uvicorn
 import shutil
 import uuid
 import os
@@ -13,26 +14,27 @@ modnet_path = model_manager.get_model_path("modnet")
 processor = create_background_remover(model_type="modnet", model_path=modnet_path)
 
 @app.post("/remove-bg/")
-async def remove_background_api(
-    file: UploadFile = File(...),
-    background: str = "transparent"
-):
-    if not file.content_type.startswith('image/'):
-        raise HTTPException(400, "File must be an image")
-    
+async def remove_bg(file: UploadFile = File(...)):
+    # Sauvegarde temporaire
+    input_path = f"/tmp/{uuid.uuid4()}.png"
+    output_path = f"/tmp/{uuid.uuid4()}_output.png"
+
+    with open(input_path, "wb") as f:
+        shutil.copyfileobj(file.file, f)
+
+    # Traitement de l'image
     try:
-        image = Image.open(file.file)
-        result = model.remove_background(image, background)
-        
-        # Sauvegarder en mémoire
-        img_byte_arr = io.BytesIO()
-        result.save(img_byte_arr, format='PNG')
-        img_byte_arr.seek(0)
-        
-        return Response(content=img_byte_arr.getvalue(), media_type="image/png")
+        processor.remove_background(image_path=input_path, output_path=output_path)
+        with open(output_path, "rb") as out_file:
+            return Response(content=out_file.read(), media_type="image/png")
     except Exception as e:
-        raise HTTPException(500, f"Error processing image: {str(e)}")
+        return {"error": str(e)}
+    finally:
+        # Nettoyage
+        if os.path.exists(input_path):
+            os.remove(input_path)
+        if os.path.exists(output_path):
+            os.remove(output_path)
 
-# Ajouter cette partie à la fin de votre app.py existant
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)