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),
@@ -27,7 +41,7 @@ class ocrPdfClient:
         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(