worker-automate-hub 0.4.418__tar.gz → 0.4.419__tar.gz

{worker_automate_hub-0.4.418 → worker_automate_hub-0.4.419}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: worker-automate-hub
-Version: 0.4.418
+Version: 0.4.419
 Summary: Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta.
 Author: Joel Paim
 Requires-Python: >=3.12,<4.0

{worker_automate_hub-0.4.418 → worker_automate_hub-0.4.419}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "worker-automate-hub"
-version = "0.4.418"
+version = "0.4.419"
 description = "Worker Automate HUB é uma aplicação para automatizar rotinas de RPA nos ambientes Argenta."
 authors = ["Joel Paim"]
 readme = "README.md"

{worker_automate_hub-0.4.418 → worker_automate_hub-0.4.419}/worker_automate_hub/tasks/jobs/entrada_de_notas_9.py

@@ -463,6 +463,7 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
 
                         max_attempts = 3
                         attempts = 0
+                        captured_text = None
                         resultado = None
                         while attempts < max_attempts:
                             main_window.set_focus()
@@ -493,9 +494,7 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                             console.print(f"Realizando OCR...\n")
                             await worker_sleep(1)
                             captured_text = pytesseract.image_to_string(Image.open(path_to_png))
-                            console.print(
-                                f"Texto Full capturado {captured_text}...\n"
-                            )
+                            console.print(f"Texto Full capturado {captured_text}...\n")
                             os.remove(path_to_png)
 
                             pattern_qtd_restante = r"Quantidade restante:\s([\d,]+)"
@@ -503,6 +502,24 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                             if resultado:
                                 break
                             else:
+                                await worker_sleep(2)
+                                console.print(f"Não conseguiu encontrar a quantidade restante, tentando cortar a imagem...\n")
+                                width, height = image.size
+                                box = (0, height - 100, width, height)
+                                cropped_image = image.crop(box)
+                                
+                                cropped_image = cropped_image.convert("L")
+                                cropped_image = cropped_image.filter(ImageFilter.SHARPEN)
+                                enhancer = ImageEnhance.Contrast(cropped_image)
+                                cropped_image = enhancer.enhance(2.0)
+                                
+                                captured_text = pytesseract.image_to_string(cropped_image)
+                                console.print(f"Texto Full capturado {captured_text}...\n")
+
+                                resultado = re.search(pattern_qtd_restante, captured_text)
+                                if resultado:
+                                    break
+
                                 await worker_sleep(2)
                                 attempts += 1
 
@@ -521,13 +538,13 @@ async def entrada_de_notas_9(task: RpaProcessoEntradaDTO) -> RpaRetornoProcessoD
                             else:
                                 return RpaRetornoProcessoDTO(
                                     sucesso=False,
-                                    retorno=f"Não foi possivel obter o resultado da quantidade restante, texto extraido: {resultado}",
+                                    retorno=f"Não foi possivel obter o resultado da quantidade restante, texto extraido: {captured_text}",
                                     status=RpaHistoricoStatusEnum.Falha,
                                 )
                         except Exception as e:
                             return RpaRetornoProcessoDTO(
                                 sucesso=False,
-                                retorno=f"Erro {e}, Não foi possivel obter o resultado da quantidade restante, texto extraido: {resultado}",
+                                retorno=f"Erro {e}, Não foi possivel obter o resultado da quantidade restante",
                                 status=RpaHistoricoStatusEnum.Falha,
                             )