✨feat: Add async handling for file upload

- Refactor the `create_upload_file` function to use `async` and `await`
- Read the contents of the uploaded file asynchronously using `await file.read()`
- Open the image using `PIL.Image.open` and convert it to RGB
- Generate the caption using `BlipInterrogator.generate_caption`
- Return the caption as a JSON response

This commit adds async handling for file uploads in the `app.py` file.

Author: sudoskys

app.py

@@ -3,15 +3,14 @@
 # @FileName: main.py
 # @Software: PyCharm
 # @Github    ：sudoskys
-import shutil
-import tempfile
-from typing import Optional
+import io
 
 import rtoml
 import uvicorn
 from PIL import Image
-from fastapi import FastAPI, UploadFile
+from fastapi import FastAPI, UploadFile, File
 from loguru import logger
+from starlette.responses import JSONResponse
 
 from utils import Blip
 
@@ -41,17 +40,14 @@
 
 
 @app.post("/upload/")
-def create_upload_file(file: Optional[UploadFile] = None):
-    if not file.file:
-        return {"code": 0, "message": "No upload file sent"}
-    else:
-        with tempfile.NamedTemporaryFile(suffix=".png") as buffer:
-            shutil.copyfileobj(file.file, buffer)
-            image_pil = Image.open(buffer.name).convert('RGB')
-        if image_pil:
-            BlipInterrogatorText = BlipInterrogator.generate_caption(
-                pil_image=image_pil)
-            return {"code": 1, "message": BlipInterrogatorText}
+async def create_upload_file(file: UploadFile = File(...)):
+    contents = await file.read()
+    image_pil = Image.open(io.BytesIO(contents)).convert('RGB')
+    if image_pil:
+        blip_interrogator_text = BlipInterrogator.generate_caption(pil_image=image_pil)
+        return JSONResponse(
+            content={"text": blip_interrogator_text}
+        )
 
 
 if __name__ == '__main__':

config.toml

@@ -1,6 +1,6 @@
 [blip]
 low_vram = true
-device = "cpu" # cpu
+device = "cpu" # cuda or cpu
 
 [server]
 host = '127.0.0.1'