Update app.py

app.py

@@ -64,11 +64,19 @@ st.title("Lucky Reactor Panel")
 
 instance_name = "lucky-reactor"
 credentials = auth()
-status, ip_address = get_status(credentials, instance_name)
 
-st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running 🚀" if status else "sleeping 😴") + ".")
+# Get instance info
+instance_info = get_server(credentials, instance_name)
+status = instance_info['status']
+ip_address = instance_info['networkInterfaces'][0]['accessConfigs'][0]['natIP']
+max_run_duration = instance_info['scheduling']['maxRunDuration']['seconds']
+last_start_timestamp = instance_info['lastStartTimestamp']
+max_remaining_time = get_time_remaining(last_start_timestamp, max_run_duration)
 
-if not status:            
+# UI
+st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running 🚀" if status == 'RUNNING' else "sleeping 😴") + ".")
+
+if status != 'RUNNING':            
     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 🚀")
@@ -79,11 +87,7 @@ if not status:
             response = activate_server(credentials, instance_name)
 else:
     st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")
-    info = get_server(credentials, instance_name)
-    max_run_duration = info['scheduling']['maxRunDuration']['seconds']
-    last_start_timestamp = info['lastStartTimestamp']
-    max_remaining_time = get_time_remaining(last_start_timestamp, max_run_duration)
-    st.write(f'Max Remaining Seconds: {max_remaining_time} s')
+    st.write(f'Lucky Reactor can still run: {max_remaining_time} s. You can terminate it in advance.')
     
     with st.form("deactivate_form"):
         token = st.text_input("Token")