Update app.py

app.py

@@ -25,6 +25,12 @@ def get_status(credentials, instance_name):
             status = True if item["status"] == "RUNNING" else False
     return status, ip_address
 
+def schedule_server(credentials, instance_name):
+    service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
+    request = service.instances().setScheduling(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name, scheduling_resource={"maxRunDuration": {"seconds": 7200}})
+    response = request.execute()
+    return response
+
 def activate_server(credentials, instance_name):
     service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
     request = service.instances().start(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name)
@@ -51,7 +57,8 @@ if not status:
         submitted = st.form_submit_button("Ignite 🚀")
         
         if submitted and token == os.getenv("EASY_TOKEN"):
-            st.write("Lucky Reactor has been ignited. Please wait a few minutes (3~5) to preceed...")            
+            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:
     st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")