Update status_checker.py

status_checker.py

@@ -5,9 +5,10 @@ import gradio as gr
 from huggingface_hub import upload_folder, HfFolder, delete_repo
 import sys
 import time
+import subprocess
 from huggingface_hub.utils import logging
 
-
+logging.set_verbosity_info()
 logger = logging.get_logger(__name__)
 
 HfFolder().save_token(os.getenv("HF_TOKEN"))
@@ -16,8 +17,17 @@ output_dataset_id = "nateraw/asdf123"
 # Where user will write outputs from their script
 outputs_dir = Path('outputs')
 
-print(f"USER SCRIPT PID? - {os.getenv('USER_SCRIPT_PID', None)}")
+process_pid = os.getenv('USER_SCRIPT_PID', None)
+logger.info(f"USER SCRIPT PID? - {process_pid} - type {type(process_pid)}")
+
+
+def process_is_complete():
+    # Attempt at solving the todo shown below
+    p = subprocess.Popen(['ps', '-p', process_pid], stdout=subprocess.PIPE)
+    out = p.communicate()[0].decode('utf-8').strip().split('\n')
+    return len(out) == 1
 
+    
 # A file that marks that the script has finished running
 # TODO - maybe check process like this instead of with done.txt: https://stackoverflow.com/a/2944076
 output_file = Path('done.txt')
@@ -25,6 +35,7 @@ output_file = Path('done.txt')
 def status_checker():
     logger.info("Waiting to find output_file to check if script is done running")
     while True:
+        logger.info(f"Process is complete? {process_is_complete()}")
         if output_file.exists():
             logger.info("Found the output file - Uploading the outputs from the script")
             upload_folder(repo_id=output_dataset_id, folder_path=str(outputs_dir), path_in_repo='.', repo_type='dataset')