Update app.py

app.py

@@ -91,6 +91,24 @@ def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instan
     except subprocess.CalledProcessError as e:
         print(f"An error occurred: {e}")
 
+
+def clear_directory(directory_path):
+    # Check if the directory exists
+    if os.path.exists(directory_path):
+        # Iterate over all the files and directories inside the specified directory
+        for filename in os.listdir(directory_path):
+            file_path = os.path.join(directory_path, filename)
+            try:
+                # Check if it is a file or a directory and remove accordingly
+                if os.path.isfile(file_path) or os.path.islink(file_path):
+                    os.unlink(file_path)  # Remove the file
+                elif os.path.isdir(file_path):
+                    shutil.rmtree(file_path)  # Remove the directory
+            except Exception as e:
+                print(f'Failed to delete {file_path}. Reason: {e}')
+    else:
+        print(f'The directory {directory_path} does not exist.')
+
 def main(image_path, b_lora_trained_folder, instance_prompt, class_prompt, training_type, training_steps):
 
     if is_shared_ui:
@@ -116,6 +134,9 @@ def main(image_path, b_lora_trained_folder, instance_prompt, class_prompt, train
     # Check if the directory exists and create it if necessary
     if not os.path.exists(local_dir):
         os.makedirs(local_dir)
+    else :
+        directory_to_clear = local_dir
+        clear_directory(directory_to_clear)
 
     shutil.copy(image_path, local_dir)
     print(f"source image has been copied in {local_dir} directory")