Update app.py

app.py

@@ -1,8 +1,11 @@
 import os
 import streamlit as st
+from dateutil import parser
 import googleapiclient.discovery
+from datetime import datetime, timedelta
 from google.oauth2 import service_account
 
+
 def auth():
     with open("/tmp/token.json", "w") as token_f:
         google_key = os.getenv("GOOGLE_KEY")
@@ -25,6 +28,13 @@ def get_status(credentials, instance_name):
             status = True if item["status"] == "RUNNING" else False
     return status, ip_address
 
+def get_time_remaining(last_start_timestamp, run_duration_seconds):
+    last_start_dt = parser.isoparse(last_start_timestamp)
+    end_run_dt = last_start_dt + timedelta(seconds=run_duration_seconds)
+    now_dt = datetime.now(tz=last_start_dt.tzinfo)
+    delta = (end_run_dt - now_dt).total_seconds()
+    return max(delta, 0.0)
+
 def get_server(credentials, instance_name):
     service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
     request = service.instances().get(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name)
@@ -69,7 +79,11 @@ if not status:
 else:
     st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")
     info = get_server(credentials, instance_name)
-    st.write(info)
+    max_run_duration = info['scheduling']['maxRunDuration']['seconds']
+    last_start_timestamp = info['lastStartTimestamp']
+    get_time_remaining(last_start_timestamp, run_duration_seconds)
+    st.write(f'Remaining seconds: {max_run_duration} s')
+    
     with st.form("deactivate_form"):
         token = st.text_input("Token")
         submitted = st.form_submit_button("Terminate 🕊️")