Add progress bars

main.py

@@ -64,28 +64,35 @@ def delete_files_in_directory(directory_path: str) -> None:
         else:
             print(f"Skipped non-file: {file_path}")
 
-def process_image(image_path: str):
+def process_image(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="Deleting files")
     delete_files_in_directory("tiles")
     delete_files_in_directory("tiles/predictions")
     delete_files_in_directory("tiles/predictions_geo")
     delete_files_in_directory("tree_images")
     delete_files_in_directory("detected_trees")
+    progress(0.01, desc="Running detectree2")
 
 
 
     run_detectree2(image_path, store_path=output_directory)
+    progress(0.2, desc="Ran detectree2")
+
 
     processed_output_df = postprocess(output_directory + '/detectree2_delin.geojson', output_directory + '/processed_delin.geojson')
+    progress(0.4, desc="Post processed")
 
     processed_geojson = output_directory + '/processed_delin.geojson'
 
+    progress(0.6, desc="Generating tree images")
     generate_tree_images(processed_geojson, image_path)
+    progress(0.7, desc="Generated tree images")
 
     output_folder = './tree_images'
 
@@ -111,7 +118,9 @@ def process_image(image_path: str):
 
     final_output_path = 'result'
 
+    progress(0.8, desc="Exporting geojson")
     export_geojson(processed_output_df, final_output_path)
+    progress(0.9, desc="Exported geojson")
 
     return final_output_path, image_path
 
@@ -168,9 +177,11 @@ def plot_results(geojson_path, image_path):
 
 
 
-def greet(image_path: str):
-    geojson_path, image_path = process_image(image_path)
+def greet(image_path: str, progress=gr.Progress()):
+    geojson_path, image_path = process_image(image_path, progress)
+    progress(0.0, desc="Plotting results")
     fig = plot_results (geojson_path, image_path)
+    progress(1, desc="Plotted results")
     
     return fig
     
@@ -191,7 +202,7 @@ def main():
         outputs=gr.Plot(label="Tree Crowns"),
     )
 
-    demo.launch(server_name='0.0.0.0', debug=True)
+    demo.launch(server_name='0.0.0.0', debug=True, show_error=True)
 
 if __name__ == "__main__":
     main()