Update app.py

app.py

@@ -1,29 +1,41 @@
 import streamlit as st
 import subprocess
 import os
+import signal
 
-# Track process
-process = None
+# Track PID in Streamlit session state
+if "process_pid" not in st.session_state:
+    st.session_state.process_pid = None
 
 st.title("Powers Control Panel ⚡")
 st.write("Use the buttons below to **start** or **stop** the `powers` Python process.")
 
+def is_process_running(pid):
+    """Check if a process with the given PID is running."""
+    try:
+        os.kill(pid, 0)
+        return True
+    except OSError:
+        return False
+
 if st.button("Start Powers"):
-    if process is None or process.poll() is not None:
-        process = subprocess.Popen(["python3", "-m", "powers"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        st.success("`powers` process started!")
+    if st.session_state.process_pid is None or not is_process_running(st.session_state.process_pid):
+        # Start the background process and store its PID
+        process = subprocess.Popen(["python3", "-m", "powers"])
+        st.session_state.process_pid = process.pid
+        st.success(f"`powers` process started with PID {process.pid}!")
     else:
         st.warning("`powers` is already running!")
 
 if st.button("Stop Powers"):
-    if process is not None:
-        process.terminate()  # Stop the process
-        process = None
+    if st.session_state.process_pid and is_process_running(st.session_state.process_pid):
+        os.kill(st.session_state.process_pid, signal.SIGTERM)  # Kill the process
+        st.session_state.process_pid = None
         st.success("`powers` process stopped!")
     else:
         st.warning("No running process to stop.")
 
-if process and process.poll() is None:
-    st.write("`powers` is **running**! 🚀")
+if st.session_state.process_pid and is_process_running(st.session_state.process_pid):
+    st.write(f"`powers` is **running** with PID {st.session_state.process_pid}! 🚀")
 else:
     st.write("`powers` is **not running**.")