Update app.py

app.py

@@ -169,6 +169,10 @@ def clear_directory(directory_path):
     else:
         print(f'The directory {directory_path} does not exist.')
 
+def get_start_info(b_lora_name):
+    your_username = api.whoami(token=hf_token)["name"]
+    return f"https://hf.co/{your_username}/{b_lora_trained_folder}"
+
 def main(image_path, b_lora_trained_folder, instance_prompt, training_type, training_steps):
 
     if is_shared_ui:
@@ -186,12 +190,6 @@ def main(image_path, b_lora_trained_folder, instance_prompt, training_type, trai
     if instance_prompt == "":
         raise gr.Error("You forgot to specify an instance prompt")
 
-    # URL to be opened
-    log_url = f"https://huggingface.co/spaces/{os.environ['SPACE_ID']}?logs=container"
-    
-    # Open URL in the current browser window
-    webbrowser.get().open(log_url)
-
     
     local_dir = "image_to_train"
     # Check if the directory exists and create it if necessary
@@ -214,12 +212,6 @@ def main(image_path, b_lora_trained_folder, instance_prompt, training_type, trai
     train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)
     
     your_username = api.whoami(token=hf_token)["name"]
-
-    # URL to be opened
-    model_url = f"https://huggingface.co/{your_username}/{b_lora_trained_folder}"
-    
-    # Open URL in the current browser window
-    webbrowser.get().open_new_tab(model_url)
     
     #swap_hardware(hardware="cpu-basic")
     swap_sleep_time(300)
@@ -360,8 +352,13 @@ with gr.Blocks(css=css) as demo:
                     
         
         train_btn = gr.Button("Train B-LoRa", visible=False)
-    
-        status = gr.Textbox(label="status")
+
+        with gr.Row():
+            started_info = gr.Textbox(
+                label="Training has started", 
+                info="You can open this space's logs to monitor logs training; once training is finished, your model will be available here:"
+            )
+            status = gr.Textbox(label="status")
 
     current_sleep_time.change(
         fn = swap_sleep_time,
@@ -382,6 +379,10 @@ with gr.Blocks(css=css) as demo:
     )
     
     train_btn.click(
+        fn = get_start_info,
+        inputs = [b_lora_name],
+        outputs = [started_info]
+    ).then(
         fn = main,
         inputs = [image, b_lora_name, instance_prompt, training_type, training_steps],
         outputs = [status]