Update app.py

app.py

@@ -187,7 +187,10 @@ def get_start_info(image_path, b_lora_name, instance_prompt):
     
     return gr.update(visible=True, value=f"https://hf.co/{your_username}/{b_lora_name}"), gr.update(visible=True)
 
-def main(image_path, b_lora_trained_folder, instance_prompt, training_type, training_steps):
+def main(started_info, image_path, b_lora_trained_folder, instance_prompt, training_type, training_steps):
+
+    if started_info == None or started_info == "":
+        raise gr.Error("Training did not start.") 
 
     local_dir = "image_to_train"
     # Check if the directory exists and create it if necessary
@@ -400,7 +403,7 @@ with gr.Blocks(css=css) as demo:
         show_api = False
     ).then(
         fn = main,
-        inputs = [image, b_lora_name, instance_prompt, training_type, training_steps],
+        inputs = [started_info, image, b_lora_name, instance_prompt, training_type, training_steps],
         outputs = [status],
         show_api = False
     )