Update app.py

app.py

@@ -76,7 +76,7 @@ max_remaining_time = get_time_remaining(last_start_timestamp, max_run_duration)
 # UI
 st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running 🚀" if status == 'RUNNING' else "sleeping 😴") + ".")
 
-if status != 'RUNNING':            
+if status == 'TERMINATED':            
     with st.form("activate_form"):
         token = st.text_input("Token (FYI: The reactor costs USD $10 per hour to operate)")
         submitted = st.form_submit_button("Ignite 🚀")
@@ -85,7 +85,7 @@ if status != 'RUNNING':
             st.write("Lucky Reactor has been ignited. Please wait a few minutes (3~5) to preceed...")     
             response = schedule_server(credentials, instance_name)
             response = activate_server(credentials, instance_name)
-else:
+elif status == 'RUNNING':
     st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")
     st.write(f'Lucky Reactor can still run: {max_remaining_time} s. You can terminate it in advance.')
     
@@ -95,4 +95,6 @@ else:
         
         if submitted and token == os.getenv("EASY_TOKEN"):
             st.write("Lucky Reactor has been terminated.")            
-            response = deactivate_server(credentials, instance_name)
+            response = deactivate_server(credentials, instance_name)
+else:
+    st.write(f"Server is [{status}]. Please wait...")