Fix on cloud execution

Author: gabriel-piles

requirements.txt

@@ -9,5 +9,5 @@ httpx==0.27.0
 sentry-sdk==2.8.0
 setuptools==75.4.0
 git+https://github.com/huridocs/queue-processor@fce76dc0c39a8d929128304eb77047985ccd344a
-git+https://github.com/huridocs/ml-cloud-connector@6f8dd2aaa890b9cebbab3254dddec65b6bbce95f
+git+https://github.com/huridocs/ml-cloud-connector@f6ae4b5244b27131e28e2ea1b501ab1bba68fdc7
 

src/tests/test_app.py

@@ -1,5 +1,6 @@
 import unittest
 from pathlib import Path
+from time import time
 from unittest import TestCase
 from fastapi.testclient import TestClient
 
@@ -15,8 +16,10 @@ def test_info(self):
 
         self.assertEqual(200, response.status_code)
 
-    # @unittest.skip("This test requires a running cloud service")
+    @unittest.skip("This test requires a running cloud service")
     def test_cloud(self):
+        start = time()
+
         test_file_path = Path(APP_PATH, "tests", "test_files", "test.pdf")
         with open(test_file_path, "rb") as stream:
             file_content = stream.read()
@@ -28,3 +31,7 @@ def test_cloud(self):
 
             response = client.get("/get_xml/default.xml")
             self.assertEqual(200, response.status_code)
+
+        print("start")
+        print("time", round(time() - start, 2), "s")
+        print("time", round(time() - start, 2), "s")

src/use_cases/extract_segments_use_case.py

@@ -5,7 +5,6 @@
 from ml_cloud_connector.domain.RestCall import RestCall
 from ml_cloud_connector.domain.ServerParameters import ServerParameters
 from ml_cloud_connector.domain.ServerType import ServerType
-from ml_cloud_connector.ports.CloudProviderRepository import CloudProviderRepository
 from ml_cloud_connector.use_cases.ExecuteOnCloudUseCase import ExecuteOnCloudUseCase
 
 from configuration import (
@@ -25,6 +24,11 @@
 
 RETRIES = 3
 
+if not USE_LOCAL_SEGMENTATION:
+    SERVER_PARAMETERS = ServerParameters(namespace="google_v2", server_type=ServerType.DOCUMENT_LAYOUT_ANALYSIS)
+    CLOUD_PROVIDER = GoogleV2Repository(server_parameters=SERVER_PARAMETERS, service_logger=service_logger)
+    EXECUTE_ON_CLOUD = ExecuteOnCloudUseCase(cloud_provider=CLOUD_PROVIDER, service_logger=service_logger)
+
 
 def get_xml_name(task: Task) -> str:
     xml_file_name = f"{task.tenant}__{task.params.filename.lower().replace('.pdf', '.xml')}"
@@ -72,10 +76,6 @@ def extract_segments(task: Task, xml_file_name: str = "") -> ExtractionData:
 
 
 def extract_segments_cloud(pdf_file: PdfFile, task: Task, xml_file_name: str = "") -> (bool, ExtractionData):
-    server_parameters = ServerParameters(namespace="google_v2", server_type=ServerType.DOCUMENT_LAYOUT_ANALYSIS)
-    cloud_provider = GoogleV2Repository(server_parameters=server_parameters, service_logger=service_logger)
-    execute_on_cloud_use_case = ExecuteOnCloudUseCase(cloud_provider=cloud_provider, service_logger=service_logger)
-
     with open(pdf_file.get_path(task.params.filename), "rb") as stream:
         file_content = stream.read()
 
@@ -88,11 +88,11 @@ def extract_segments_cloud(pdf_file: PdfFile, task: Task, xml_file_name: str = "
         files=files,
         data={"fast": "False"},
     )
-    response, success, error = execute_on_cloud_use_case.execute(rest_call)
+    response, success, error = EXECUTE_ON_CLOUD.execute(rest_call)
     if not success:
         return False, None
 
-    if not save_cloud_xml_file(execute_on_cloud_use_case, xml_file_name):
+    if not save_cloud_xml_file(xml_file_name):
         return False, None
 
     segments: list[SegmentBox] = [SegmentBox(**segment_box) for segment_box in response.json()]
@@ -106,14 +106,14 @@ def extract_segments_cloud(pdf_file: PdfFile, task: Task, xml_file_name: str = "
     )
 
 
-def save_cloud_xml_file(execute_on_cloud_use_case: ExecuteOnCloudUseCase, xml_file_name: str) -> bool:
+def save_cloud_xml_file(xml_file_name: str) -> bool:
     try:
         rest_call = RestCall(
             port=DOCUMENT_LAYOUT_ANALYSIS_PORT,
             endpoint=["get_xml", xml_file_name],
             method="GET",
         )
-        response, success, error = execute_on_cloud_use_case.execute(rest_call)
+        response, success, error = EXECUTE_ON_CLOUD.execute(rest_call)
 
         if not success:
             return False