Commit 3b30e749 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update pdf_client.py

parent 46988c37
...@@ -25,9 +25,11 @@ class ocrPdfClient: ...@@ -25,9 +25,11 @@ class ocrPdfClient:
) )
try: try:
response = requests.post(f"{self.api_url}/pdf_ocr", json=payload) 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() response.raise_for_status()
output_dir = os.path.join(output_dir, file_name_without_extension) #logger.info(f'output_dir:{output_dir}')
return output_dir if response.status_code == 200 else None return output_dir if response.json()['status_code'] == 200 else None
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
logger.error(f"OCR PDF API request failed: {e}") logger.error(f"OCR PDF API request failed: {e}")
return None return None
...@@ -87,3 +89,4 @@ def main(): ...@@ -87,3 +89,4 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
main() main()
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