Update app.py

app.py

@@ -83,7 +83,7 @@ def check_sleep_time():
     sleep_time = get_sleep_time()
     if sleep_time is None :
         sleep_time_value = "Don't sleep"
-        return sleep_time_value, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+        return sleep_time_value, gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
     else :
         if sleep_time == 300:
             sleep_time_value = "5 minutes"
@@ -103,7 +103,7 @@ def check_sleep_time():
             sleep_time_value = "72 hours"
         elif sleep_time == 604800:
             sleep_time_value = "1 week"
-        return sleep_time_value, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return sleep_time_value, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
     
 def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instance_prompt, class_prompt, max_train_steps, checkpoint_steps):
     
@@ -262,6 +262,11 @@ div#warning-setgpu p.actions > a {
     font-weight: 600;
     font-size: 1.2em;
 }
+div#warning-setsleeptime {
+    background-color: #fff4eb;
+    padding: 0 10px 5px;
+    margin: 20px 0;
+}
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
@@ -311,14 +316,22 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Column():
                 
-                    with gr.Row():
-                        current_sleep_time = gr.Dropdown(
-                            label="current space sleep time",
-                            choices = [
-                                "Don't sleep", "5 minutes", "15 minutes", "30 minutes", "1 hour", "10 hours", "24 hours", "48 hours", "72 hours", "1 week"
-                            ]
-                        )
-                        #check_sleep_time_btn = gr.Button("check", scale=1)
+                    
+                    current_sleep_time = gr.Dropdown(
+                        label="current space sleep time",
+                        choices = [
+                            "Don't sleep", "5 minutes", "15 minutes", "30 minutes", "1 hour", "10 hours", "24 hours", "48 hours", "72 hours", "1 week"
+                        ]
+                    )
+                    sleep_time_message = gr.HTML('''
+                    <div class="gr-prose">
+                        <p>
+                        First of all, please make sure your space's sleep time value is set on long enough, so it do not fall asleep during training. <br />
+                        Set it to <strong>"Don't sleep"</strong> or <strong>more than 1 hour</strong> to be safe.
+                        Don't worry, after training is finished, sleep time will be back to 5 minutes.
+                        </p>
+                    </div>
+                    ''', elem_id="warning-setsleeptime")
                     training_type = gr.Radio(label="Training type", choices=["style", "concept"], value="style", visible=False)
                     b_lora_name = gr.Textbox(label="Name your B-LoRa model", placeholder="b_lora_trained_folder", visible=False)
                     with gr.Row():
@@ -342,7 +355,7 @@ with gr.Blocks(css=css) as demo:
     demo.load(
         fn = check_sleep_time,
         inputs = None, 
-        outputs = [current_sleep_time, b_lora_name, instance_prompt, class_prompt, training_type, training_steps, train_btn]
+        outputs = [current_sleep_time, sleep_time_message, b_lora_name, instance_prompt, class_prompt, training_type, training_steps, train_btn]
     )
     
     training_type.change(