Update app.py

app.py

@@ -40,9 +40,31 @@ def swap_hardware(hf_token, hardware="cpu-basic"):
 
 
 def swap_sleep_time(sleep_time):
+    
+    if sleep_time == "5 minutes":
+        new_sleep_time = 300
+    elif sleep_time == "15 minutes":
+        new_sleep_time = 900
+    elif sleep_time == "30 minutes":
+        new_sleep_time = 1800
+    elif sleep_time == "1 hour":
+        new_sleep_time = 3600
+    elif sleep_time == "10 hours":
+        new_sleep_time = 36000
+    elif sleep_time == "24 hours":
+        new_sleep_time = 86400
+    elif sleep_time == "48 hours":
+        new_sleep_time = 172800
+    elif sleep_time == "72 hours":
+        new_sleep_time = 259200
+    elif sleep_time == "1 week":
+        new_sleep_time = 604800
+    elif sleep_time == "Don't sleep":
+        new_sleep_time = -1
+    
     sleep_time_url = f"https://huggingface.co/api/spaces/{os.environ['SPACE_ID']}/sleeptime"
     headers = { "authorization" : f"Bearer {hf_token}"}
-    body = {'seconds':sleep_time}
+    body = {'seconds':new_sleep_time}
     requests.post(sleep_time_url,json=body,headers=headers)
 
 
@@ -60,9 +82,28 @@ def get_sleep_time():
 def check_sleep_time():
     sleep_time = get_sleep_time()
     if sleep_time is None :
-        return sleep_time, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+        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)
     else :
-        return sleep_time, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        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:
+            sleep_time_value = "1 hour"
+        elif sleep_time == 36000:
+            sleep_time_value = "10 hours"
+        elif sleep_time == 86400:
+            sleep_time_value = "24 hours"
+        elif sleep_time == 172800:
+            sleep_time_value = "48 hours"
+        elif sleep_time == 259200:
+            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)
     
 def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instance_prompt, class_prompt, max_train_steps, checkpoint_steps):
     
@@ -274,7 +315,7 @@ with gr.Blocks(css=css) as demo:
                         current_sleep_time = gr.Dropdown(
                             label="current space sleep time",
                             choices = [
-                                -1, 300, 900, 1800, 3600, 36000, 86400, 172800, 259200, 604800
+                                "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)