Update app.py

app.py

@@ -42,9 +42,9 @@ def get_server(credentials, instance_name):
     response = request.execute()
     return response
 
-def schedule_server(credentials, instance_name):
+def schedule_server(credentials, instance_name, running_time):
     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, body={"automaticRestart": False,"maxRunDuration": {"seconds": 7200}})    
+    request = service.instances().setScheduling(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name, body={"automaticRestart": False,"maxRunDuration": {"seconds": running_time*3600}})    
     response = request.execute()
     return response
 
@@ -79,11 +79,12 @@ st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running 🚀" if
 if status == 'TERMINATED':            
     with st.form("activate_form"):
         token = st.text_input("Token (FYI: The reactor costs USD $10 per hour to operate)")
+        running_time = st.select_slider("Running Hours", options=[1,2,3,4,5,6])
         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...")     
-            response = schedule_server(credentials, instance_name)
+            response = schedule_server(credentials, instance_name, running_time)
             response = activate_server(credentials, instance_name)
 elif status == 'RUNNING':
     st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")