Progress bar

main.py

@@ -53,19 +53,22 @@ 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()
 
     output_directory = os.path.join(current_directory, "outputs")
     if not os.path.exists(output_directory):
         os.makedirs(output_directory)
 
+    progress(0, desc="Running detectree2")
     run_detectree2(image_path, store_path=output_directory)
+    progress(0.1, desc="Postprocessing")
 
     processed_output_df = postprocess(output_directory + '/detectree2_delin.geojson', output_directory + '/processed_delin')
 
     processed_geojson = output_directory + '/processed_delin.geojson'
 
+    progress(0.2, desc="Generating tree images")
     generate_tree_images(processed_geojson, image_path)
 
     output_folder = './tree_images'
@@ -85,6 +88,8 @@ def greet(image_path: str):
     processed_output_df['species'] = all_top_3_list
 
     final_output_path = 'result'
+
+    progress(0.3, desc="Exporting geojson")
     export_geojson(processed_output_df, final_output_path)
 
     with rasterio.open(image_path) as src:
@@ -100,6 +105,8 @@ def greet(image_path: str):
     # Enable interactive mode
     # plt.ion()
 
+    progress(0.4, desc="Plotting")
+
     # Plotting
     fig, ax = plt.subplots(figsize=(10, 10))