|
from fastapi import FastAPI, File, UploadFile |
|
from fastapi.staticfiles import StaticFiles |
|
from fastapi.responses import FileResponse |
|
from PIL import Image |
|
import numpy as np |
|
import urllib.request |
|
import io |
|
import os |
|
from moviepyTest import test |
|
from typing import * |
|
|
|
app = FastAPI() |
|
|
|
@app.get("/inference") |
|
def inference(): |
|
return "<p>Hello, World!</p>" |
|
|
|
@app.get("/infer_t5") |
|
def t5(input): |
|
|
|
return {"output": input} |
|
|
|
@app.get("/moviepyTest") |
|
def t5(): |
|
result = test() |
|
|
|
return {"output": result} |
|
|
|
|
|
|
|
|
|
@app.post("/getOriginalMangaList") |
|
async def getOriginalMangaList(images: List[UploadFile] = File(...)): |
|
for image in images: |
|
img = await image.read() |
|
image = Image.open(io.BytesIO(img)).convert("L").convert("RGB") |
|
path_to_image = f"/manga/{images.index(image)}.jpg" |
|
image.save(path_to_image) |
|
return "获取图片保存成功" |
|
|
