Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
zhougaofeng
magic_pdf
Commits
f80949a5
"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "d88bd51e3c446383b37a65ad2119ea640f7df913"
Commit
f80949a5
authored
Oct 24, 2024
by
zhougaofeng
Browse files
Update pdf_client.py
parent
0a43c18c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
7 deletions
+2
-7
magic_pdf/parse/pdf_client.py
magic_pdf/parse/pdf_client.py
+2
-7
No files found.
magic_pdf/parse/pdf_client.py
View file @
f80949a5
...
...
@@ -18,21 +18,16 @@ class ocrPdfClient:
"path"
:
str
(
path
),
"output_dir"
:
str
(
output_dir
),
}
file_name_without_extension
,
_
=
os
.
path
.
splitext
(
os
.
path
.
basename
(
path
))
logger
.
info
(
f
'pdf_server:
{
self
.
api_url
}
, pdf path:
{
path
}
'
)
logger
.
info
(
f
'reading:
{
path
}
'
)
try
:
response
=
requests
.
post
(
f
"
{
self
.
api_url
}
/pdf_ocr"
,
json
=
payload
)
#logger.info(f'response:{response.json()}')
output_dir
=
response
.
json
()[
'output_path'
]
response
.
raise_for_status
()
#logger.info(f'output_dir:{output_dir}')
return
output_dir
if
response
.
json
()[
'status_code'
]
==
200
else
None
except
requests
.
exceptions
.
RequestException
as
e
:
logger
.
error
(
f
"OCR PDF API request failed:
{
e
}
"
)
return
None
def
parse_args
():
parser
=
argparse
.
ArgumentParser
()
parser
.
add_argument
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment