updated logging

app.py

@@ -1,5 +1,13 @@
 import os
 import sys
+import logging
+logger = logging.getLogger(__file__)
+handler = logging.StreamHandler(sys.stdout)
+handler.setLevel(logging.INFO)
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+
 import json
 import uuid
 import traceback
@@ -33,7 +41,9 @@ example_nifti_path = "img.nii.gz"
 
 def main_func(input_file_list):
     with tempfile.TemporaryDirectory() as tempdir:
-        print('start time:',datetime.datetime.now(timezone.utc).strftime('%Y-%m-%d-%H-%M-%S-%Z'))
+        tstamp = datetime.datetime.now(timezone.utc).strftime('%Y-%m-%d-%H-%M-%S-%Z')
+        print(f'start time: {tstamp}')
+        print('image reading...')
         image_file = os.path.join(tempdir,f"image.nii.gz")
         first_file = input_file_list[0].name
         if first_file.endswith(".dcm"):
@@ -43,15 +53,20 @@ def main_func(input_file_list):
             img_obj = sitk.ReadImage(first_file)
         else:
             raise ValueError("only accept .nii.gz file or .dcm files!")
-        print('img_obj.GetDirection()',img_obj.GetDirection())
+
+        print(f'size {img_obj.GetSize()}')
+        print(f'spacing {img_obj.GetSpacing()}')
+        print(f'direction {img_obj.GetDirection()}',)
         
         sitk.WriteImage(img_obj,image_file)
 
+        print('inference starting...')
         inference(image_file,tempdir,weight_file)
 
         mask_file = os.path.join(tempdir,"lungqia-0.nii.gz")
         csv_file = os.path.join(tempdir,"export-0.csv")
         orientation_file = os.path.join(tempdir,"orientation.json")
+        orientation_file = None
         my_uid = uuid.uuid4().hex
         output_pdf_file = os.path.join('/tmp',f'{my_uid}.pdf')
         output_csv_file = os.path.join('/tmp',f'{my_uid}.csv')
@@ -59,11 +74,12 @@ def main_func(input_file_list):
         try:
             assert(os.path.exists(mask_file))
             assert(os.path.exists(csv_file))
-            assert(os.path.exists(orientation_file))
+            #assert(os.path.exists(orientation_file))
         except:
             traceback.print_exc()
             raise ValueError("inference failed to generate output files")
 
+        print('report generating...')
         inst = NiftiVisualizer("NA",image_file,mask_file,csv_file,tempdir,orientation_file=orientation_file)
         tmp_pdf_file = inst.gen_pdf()
         assert(os.path.exists(tmp_pdf_file))
@@ -73,8 +89,9 @@ def main_func(input_file_list):
         shutil.copy(tmp_pdf_file,output_pdf_file)
         shutil.copy(csv_file,output_csv_file)
 
-        print('uid:',os.path.basename(output_csv_file))
-        print('end time:',datetime.datetime.now(timezone.utc).strftime('%Y-%m-%d-%H-%M-%S-%Z'))
+        print(f'uid: {my_uid}')
+        tstamp = datetime.datetime.now(timezone.utc).strftime('%Y-%m-%d-%H-%M-%S-%Z')
+        print(f'end time: {tstamp}')
         return output_pdf_file, output_df, output_pdf_file, output_csv_file
 
 with gr.Blocks() as demo: