Update app.py

app.py

@@ -31,6 +31,12 @@ def activate_server(credentials, instance_name):
     response = request.execute()
     return response
 
+def deactivate_server(credentials, instance_name):
+    service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
+    request = service.instances().stop(project='nus-cisco-corp-lab-wp1',zone='asia-southeast1-c',instance=instance_name, discardLocalSsd=False)
+    response = request.execute()
+    return response
+
 st.title("Lucky Reactor Panel")
 
 instance_name = "lucky-reactor"
@@ -40,7 +46,7 @@ status, ip_address = get_status(credentials, instance_name)
 st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running 🚀" if status else "sleeping 😴") + ".")
 
 if not status:            
-    with st.form("my_form"):
+    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 🚀")
         
@@ -48,4 +54,11 @@ if not status:
             st.write("Lucky Reactor has been ignited. Please wait a few minutes (3~5) to preceed...")            
             response = activate_server(credentials, instance_name)
 else:
-    st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")
+    st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")
+    with st.form("deactivate_form"):
+        token = st.text_input("Token")
+        submitted = st.form_submit_button("Terminate 🕊️")
+        
+        if submitted and token == os.getenv("EASY_TOKEN"):
+            st.write("Lucky Reactor has been terminated.")            
+            response = deactivate_server(credentials, instance_name)