Update pdf_client.py

magic_pdf/parse/pdf_client.py

@@ -13,6 +13,20 @@ class ocrPdfClient:
     def __init__(self, api_url):
         self.api_url = api_url
 
+    def check_health(self):
+        health_check_url = f'{self.api_url}/health'
+        try:
+            response = requests.get(health_check_url)
+            if response.status_code == 200:
+                logger.info("Server is healthy and ready to process requests.")
+                return True
+            else:
+                logger.error(f'Server health check failed with status code:{response.status_code}')
+                return False
+        except requests.exceptions.RequestException as e:
+            logger.error(f'Health check request failed:{e}')
+            return False
+
     def ocr_pdf_client(self, path, output_dir):
         payload = {
             "path": str(path),