Progress bar

main.py

@@ -53,19 +53,24 @@ generate_tree_images stores the cutout tree images in a separate folder
 """
 
 
-def greet(image_path: str):
+def greet(image_path: str, progress=gr.Progress()):
     current_directory = os.getcwd()
 
+    progress(0, desc="Starting...")
+
     output_directory = os.path.join(current_directory, "outputs")
     if not os.path.exists(output_directory):
         os.makedirs(output_directory)
 
+    progress(0.1, desc="Running detectree2...")
     run_detectree2(image_path, store_path=output_directory)
 
+    progress(0.2, desc="Postprocessing...")
     processed_output_df = postprocess(output_directory + '/detectree2_delin.geojson', output_directory + '/processed_delin')
 
     processed_geojson = output_directory + '/processed_delin.geojson'
 
+    progress(0.3, desc="Generating tree images...")
     generate_tree_images(processed_geojson, image_path)
 
     output_folder = './tree_images'
@@ -85,6 +90,8 @@ def greet(image_path: str):
     processed_output_df['species'] = all_top_3_list
 
     final_output_path = 'result'
+
+    progress(0.4, desc="Exporting GeoJSON...")
     export_geojson(processed_output_df, final_output_path)
 
     with rasterio.open(image_path) as src: