Update api.py

api.py

@@ -32,14 +32,16 @@ def t5():
 
     
 ##这个比较快不用异步
-@app.post("/getOriginalMangaList")
-async def getOriginalMangaList(images: List[UploadFile] = File(...)):
+@app.post("/getOriginalMangaList") #增加指定保存路径功能,默认是/manga路径
+def getOriginalMangaList(images: List[UploadFile] = File(...), save_path: str = "/manga"):
+    saved_files = []
     for idx, image in enumerate(images):
-        img = await image.read()
-        image = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
-        path_to_image = f"/manga/{idx}.jpg"
-        image.save(path_to_image)
-    return "获取图片保存成功"
+        img = image.file.read()
+        image_data = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
+        path_to_image = os.path.join(save_path, f"{idx}.jpg")
+        image_data.save(path_to_image)
+        saved_files.append(path_to_image)
+    return {"message": "Images saved successfully", "saved_files": saved_files}
 ##这个比较快不用异步
 
 ##这个比较快不用异步