add _is_space flag, block inference during training, understand clone fix

app.py

@@ -47,6 +47,10 @@ MODELS_DIR = os.environ.get("ACE_MODELS_DIR", "/app/models")
 
 ACE_SERVER_BIN = "/app/ace-server"
 
+# Detect if running on HF Space (ace-server available) vs locally (PyTorch only)
+_is_space = os.path.isfile(ACE_SERVER_BIN) or os.environ.get("SPACE_ID") is not None
+_training_in_progress = False
+
 # HF repo for on-demand GGUF downloads
 GGUF_HF_REPO = "Serveurperso/ACE-Step-1.5-GGUF"
 
@@ -445,6 +449,8 @@ def gradio_main():
     def generate_music(caption, lyrics, instrumental, bpm, duration, seed,
                        steps, lora_select, lm_model_select,
                        progress=gr.Progress(track_tqdm=True)):
+        if _training_in_progress:
+            return None, "Training in progress. Inference unavailable until training completes. Press Cancel to stop training."
         if not _server_ok():
             return None, "ace-server not running. Check logs."
 
@@ -625,6 +631,8 @@ def gradio_main():
             yield _log_text(), gr.Button(visible=False), gr.Button(visible=True), gr.File()
 
         # Stop ace-server before training (frees memory)
+        global _training_in_progress
+        _training_in_progress = True
         _log("[INFO] Stopping ace-server for training...")
         yield _log_text(), gr.Button(visible=False), gr.Button(visible=True), gr.File()
         _stop_ace_server()
@@ -712,6 +720,7 @@ def gradio_main():
             yield _log_text(), gr.Button(visible=True), gr.Button(visible=False), gr.File()
 
         finally:
+            _training_in_progress = False
             # Always restart ace-server
             _log("[INFO] Restarting ace-server...")
             yield _log_text(), gr.Button(visible=False), gr.Button(visible=True), gr.File()