Update handler.py

handler.py

@@ -23,9 +23,9 @@ git_clone_command = "git clone https://github.com/OpenPecha/tibetan-aligner"
 # Run the command using subprocess
 try:
     subprocess.run(git_clone_command, shell=True, check=True)
-    print("Git clone successful!")
+    logging.info("Git clone successful!")
 except subprocess.CalledProcessError as e:
-    print(f"Error while running Git clone command: {e}")
+    logging.info(f"Error while running Git clone command: {e}")
 
 
 ALIGNER_SCRIPT_DIR = Path("./tibetan-aligner").resolve()
@@ -106,7 +106,7 @@ def convert_raw_align_to_tm(align_fn: Path, tm_path: Path):
 
     def load_alignment(fn: Path):
         content = fn.read_text()
-        print("Content !!! \n\n"+content)
+        logging.info("Content !!! \n\n"+content)
         if not content:
             return []
 
@@ -172,7 +172,7 @@ def add_to_publish_todo_repo(org, repo_name, file_path, access_token):
     response = requests.get(url, headers=headers)
 
     if response.status_code == 200:
-        print(f"[INFO] '{file_path}' already added.")
+        logging.info(f"[INFO] '{file_path}' already added.")
         return
 
     payload = {"message": f"Add {file_path}", "content": ""}
@@ -180,10 +180,10 @@ def add_to_publish_todo_repo(org, repo_name, file_path, access_token):
     response = requests.put(url, headers=headers, json=payload)
 
     if response.status_code == 201:
-        print(f"[INFO] '{file_path}' added to publish todo")
+        logging.info(f"[INFO] '{file_path}' added to publish todo")
     else:
-        print(f"[ERROR] Failed to add '{file_path}'.")
-        print(f"[ERROR] Response: {response.text}")
+        logging.info(f"[ERROR] Failed to add '{file_path}'.")
+        logging.info(f"[ERROR] Response: {response.text}")
 
 
 def create_tm(align_fn: Path, text_pair: Dict[str, str]):
@@ -246,10 +246,10 @@ def align(text_pair):
         output_dir = Path(tmpdir)
         bo_fn = download_file(text_pair["bo_file_url"], output_fn=output_dir / "bo.tx")
         en_fn = download_file(text_pair["en_file_url"], output_fn=output_dir / "en.tx")
-        print("bo_fn: ", bo_fn)
-        print("en_fn: ", en_fn)
+        logging.info("bo_fn: ", bo_fn)
+        logging.info("en_fn: ", en_fn)
         aligned_fn = _run_align_script(bo_fn, en_fn, output_dir)
-        print("aligned_fn: ", aligned_fn)
+        logging.info("aligned_fn: ", aligned_fn)
         repo_url = create_tm(aligned_fn, text_pair=text_pair)
         return {"tm_repo_url": repo_url}
 
@@ -267,3 +267,4 @@ class EndpointHandler():
         """
         data = data.pop("inputs",data)
         return align(data)
+