refactor: ♻️ changing outdated prints to logger

Author: marcvergees

app/services/file_manipulator.py

@@ -1,6 +1,9 @@
 import os
 from app.services.filler import Filler
 from app.services.llm import LLM
+from app.core.logging import get_logger
+
+logger = get_logger(__name__)
 
 
 class FileManipulator:
@@ -39,25 +42,23 @@ def fill_form(self, user_input: str, fields: list, pdf_form_path: str, model: st
         It receives the raw data, runs the PDF filling logic,
         and returns the path to the newly created file.
         """
-        print("[1] Received request from frontend.")
-        print(f"[2] PDF template path: {pdf_form_path}")
+        logger.info("[1] Received request from frontend.")
+        logger.info("[2] PDF template path: %s", pdf_form_path)
 
         if not os.path.exists(pdf_form_path):
             raise FileNotFoundError(f"PDF template not found at {pdf_form_path}")
 
-        print("[3] Starting extraction and PDF filling process...")
+        logger.info("[3] Starting extraction and PDF filling process...")
         try:
             self.llm._target_fields = fields
             self.llm._transcript_text = user_input
             self.llm._model = model
             output_name = self.filler.fill_form(pdf_form=pdf_form_path, llm=self.llm)
 
-            print("\n----------------------------------")
-            print("✅ Process Complete.")
-            print(f"Output saved to: {output_name}")
+            logger.info("Process complete. Output saved to: %s", output_name)
 
             return output_name
 
         except Exception as e:
-            print(f"An error occurred during PDF generation: {e}")
+            logger.error("An error occurred during PDF generation: %s", e)
             raise e

app/services/llm.py

@@ -4,6 +4,9 @@
 from requests.exceptions import Timeout, RequestException
 
 from app.core.config import OLLAMA_HOST, OLLAMA_MODEL
+from app.core.logging import get_logger
+
+logger = get_logger(__name__)
 
 
 class LLM:
@@ -51,9 +54,9 @@ def main_loop(self):
                         json_data = response.json()
                         break
                     except Timeout:
-                        print(f"[LOG]: Ollama request timed out (attempt {attempt+1}) for field '{field}'. Retrying...")
+                        logger.warning("Ollama request timed out (attempt %d) for field '%s'. Retrying...", attempt + 1, field)
                     except RequestException as e:
-                        print(f"[LOG]: Ollama request failed: {e}")
+                        logger.error("Ollama request failed: %s", e)
             except requests.exceptions.ConnectionError:
                 raise ConnectionError(
                     f"Could not connect to Ollama at {ollama_url}. "
@@ -67,12 +70,9 @@ def main_loop(self):
             else:
                 parsed_response = json_data["response"]
                 self.add_response_to_json(field, parsed_response)
-                print(f"[{i}/{total_fields}] Extracted data for field '{field}' successfully.")
+                logger.info("[%d/%d] Extracted data for field '%s' successfully.", i, total_fields, field)
 
-        print("----------------------------------")
-        print("\t[LOG] Resulting JSON created from the input text:")
-        print(json.dumps(self._json, indent=2))
-        print("--------- extracted data ---------")
+        logger.info("Resulting JSON created from the input text:\n%s", json.dumps(self._json, indent=2))
 
         return self