"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "ace922132c3f7d0428ad25794224fc2412098f35"
Unverified Commit 11bf98d0 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2411 from CharlesKeeling65/patch-1

Update app.py: Fix parameter parsing in /file_parse endpoint
parents f0b66d3a 862891e2
...@@ -21,6 +21,7 @@ from magic_pdf.libs.config_reader import get_bucket_name, get_s3_config ...@@ -21,6 +21,7 @@ from magic_pdf.libs.config_reader import get_bucket_name, get_s3_config
from magic_pdf.model.doc_analyze_by_custom_model import doc_analyze from magic_pdf.model.doc_analyze_by_custom_model import doc_analyze
from magic_pdf.operators.models import InferenceResult from magic_pdf.operators.models import InferenceResult
from magic_pdf.operators.pipes import PipeResult from magic_pdf.operators.pipes import PipeResult
from fastapi import Form
model_config.__use_inside_model__ = True model_config.__use_inside_model__ = True
...@@ -102,6 +103,7 @@ def init_writers( ...@@ -102,6 +103,7 @@ def init_writers(
# 处理上传的文件 # 处理上传的文件
file_bytes = file.file.read() file_bytes = file.file.read()
file_extension = os.path.splitext(file.filename)[1] file_extension = os.path.splitext(file.filename)[1]
writer = FileBasedDataWriter(output_path) writer = FileBasedDataWriter(output_path)
image_writer = FileBasedDataWriter(output_image_path) image_writer = FileBasedDataWriter(output_image_path)
os.makedirs(output_image_path, exist_ok=True) os.makedirs(output_image_path, exist_ok=True)
...@@ -176,14 +178,14 @@ def encode_image(image_path: str) -> str: ...@@ -176,14 +178,14 @@ def encode_image(image_path: str) -> str:
) )
async def file_parse( async def file_parse(
file: UploadFile = None, file: UploadFile = None,
file_path: str = None, file_path: str = Form(None),
parse_method: str = "auto", parse_method: str = Form("auto"),
is_json_md_dump: bool = False, is_json_md_dump: bool = Form(False),
output_dir: str = "output", output_dir: str = Form("output"),
return_layout: bool = False, return_layout: bool = Form(False),
return_info: bool = False, return_info: bool = Form(False),
return_content_list: bool = False, return_content_list: bool = Form(False),
return_images: bool = False, return_images: bool = Form(False),
): ):
""" """
Execute the process of converting PDF to JSON and MD, outputting MD and JSON files Execute the process of converting PDF to JSON and MD, outputting MD and JSON files
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment