Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import subprocess
 from typing import Union
 from huggingface_hub import whoami
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, HfHubHTTPError
 is_spaces = True if os.environ.get("SPACE_ID") else False
 
 if is_spaces:
@@ -253,6 +253,7 @@ def start_training(
         shutil.copy(script_location + "/script.py", dataset_folder)
         # copy requirements.autotrain to dataset_folder as requirements.txt
         shutil.copy(script_location + "/requirements.autotrain", dataset_folder + "/requirements.txt")
+        
         # command to run autotrain spacerunner
         # --- Pre-create subspace and inject token ---
         hf_token = oauth_token.token
@@ -263,8 +264,15 @@ def start_training(
             # Try to create the space (if it doesn't exist)
             api.create_repo(repo_id=new_space_id, repo_type="space", space_sdk="docker", private=True)
             print(f"Created space {new_space_id}")
+        except HfHubHTTPError as e:
+            if e.response is not None and e.response.status_code == 409:
+                print(f"Space {new_space_id} already exists — continuing.")
+            else:
+                print(f"Error creating space: {e}")
+                raise e
         except Exception as e:
-            print(f"Space creation may have failed or already exists: {e}")
+            print(f"General space creation error: {e}")
+            raise e
 
         try:
             # Inject HF_TOKEN into the new space
@@ -276,6 +284,7 @@ def start_training(
         cmd = f"autotrain spacerunner --project-name {slugged_lora_name} --script-path {dataset_folder}"
         cmd += f" --username {profile.username} --token {hf_token} --backend spaces-l4x1"
         outcome = subprocess.run(cmd.split())
+        
         if outcome.returncode == 0:
             return f"""# Your training has started. 
     ## - Training Status: <a href='https://huggingface.co/spaces/{profile.username}/autotrain-{slugged_lora_name}?logs=container'>{profile.username}/autotrain-{slugged_lora_name}</a> <small>(in the logs tab)</small>