github/ocr_api_server
5
0
Fork 0
mirror of https://github.com/sml2h3/ocr_api_server.git synced 2025-05-04 08:29:24 +08:00
Code Issues Projects Releases Wiki Activity

Merge 29145a689e into a40a6b96d7

Browse Source
This commit is contained in:
Ruan HongChao 2024-11-16 09:18:04 +08:00 committed by GitHub
commit bfbab3821c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -14,4 +14,4 @@ RUN pip install --no-cache-dir -r requirements.txt
EXPOSE 8000 EXPOSE 8000
# 运行应用 # 运行应用
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"] CMD ["python", "app/main.py"]

8
app/main.py
View File

@ -15,7 +15,11 @@ async def decode_image(image: Union[UploadFile, StarletteUploadFile, str, None])
raise HTTPException(status_code=400, detail="No image provided") raise HTTPException(status_code=400, detail="No image provided")
if isinstance(image, (UploadFile, StarletteUploadFile)): if isinstance(image, (UploadFile, StarletteUploadFile)):
return await image.read() content = await image.read()
if not content:
raise HTTPException(status_code=400, detail="Uploaded file is empty")
return content
elif isinstance(image, str): elif isinstance(image, str):
try: try:
# 检查是否是 base64 编码的图片 # 检查是否是 base64 编码的图片
@ -57,7 +61,7 @@ async def slide_match_endpoint(
simple_target: bool = Form(False) simple_target: bool = Form(False)
): ):
try: try:
if (background is None and target is None) or (background_file.size == 0 and target_file.size == 0): if (target_file is None and target is None) or (background_file is None and background is None):
return APIResponse(code=400, message="Both target and background must be provided") return APIResponse(code=400, message="Both target and background must be provided")
target_bytes = await decode_image(target_file or target) target_bytes = await decode_image(target_file or target)