Update app.py

app.py

@@ -83,15 +83,8 @@ 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=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"
-        elif sleep_time == 900:
-            sleep_time_value = "15 minutes"
-        elif sleep_time == 1800:
-            sleep_time_value = "30 minutes"
-        elif sleep_time == 3600:
+    elif sleep_time >= 36000:
+        if sleep_time == 3600:
             sleep_time_value = "1 hour"
         elif sleep_time == 36000:
             sleep_time_value = "10 hours"
@@ -103,6 +96,15 @@ 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=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"
+        elif sleep_time == 900:
+            sleep_time_value = "15 minutes"
+        elif sleep_time == 1800:
+            sleep_time_value = "30 minutes"
+        
         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):
@@ -264,7 +266,7 @@ div#warning-setgpu p.actions > a {
 }
 div#warning-setsleeptime {
     background-color: #fff4eb;
-    padding: 0 10px 5px;
+    padding: 10px 5px;
     margin: 20px 0;
 }
 """
@@ -325,11 +327,9 @@ with gr.Blocks(css=css) as demo:
                     )
                     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>
+                        <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. </p>
+                        <p>Set it to <strong>"Don't sleep"</strong> or <strong>more than 1 hour</strong> to be safe.</p>
+                        <p>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)