import os import streamlit as st import googleapiclient.discovery from google.oauth2 import service_account def auth(): with open("/tmp/token.json", "w") as token_f: google_key = os.getenv("GOOGLE_KEY") token_f.write(google_key) credentials = service_account.Credentials.from_service_account_file('/tmp/token.json') return credentials def get_status(credentials): status = False client = googleapiclient.discovery.build('compute', 'v1', credentials=credentials) result = client.instances().list(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c').execute() for item in result['items']: if item["name"] == "oj": status = True if item["status"] == "RUNNING" else False return status def activate_server(): return True st.title("Lucky Reactor") credentials = auth() status = True st.markdown("credentials: " + credentials) st.image("cover.jpg", caption="Reactor is currently " + "running 🚀" if status else "sleeping 😴" + ".") if not status: if st.button("Ignite"): st.write("Here you go: [LINK]")