Update app.py

app.py

@@ -1,55 +1,47 @@
 import streamlit as st
 import subprocess
 import os
-import signal
+import pymongo
 
-# Track process for running 'Powers'
+# Track process
 process = None
 
-st.title("Powers Bot Control Panel ⚡")
-st.write("Use this panel to start/stop the bot and view real-time logs.")
+# Title
+st.title("Powers Control Panel ⚡")
+st.write("Use the buttons below to **start** or **stop** the `powers` Python process.")
 
-# Container to display logs dynamically
-log_container = st.empty()
+# MongoDB Connection Test
+mongo_uri = os.getenv("MONGO_URI")
+st.write(f"Connecting to MongoDB with URI: {mongo_uri}")
 
+try:
+    client = pymongo.MongoClient(mongo_uri)
+    client.admin.command('ping')  # Test the connection
+    st.success("Connected to MongoDB successfully!")
+except Exception as e:
+    st.error(f"Failed to connect to MongoDB: {e}")
+
+# Start button: Installs dependencies and runs `powers`
 if st.button("Start Powers"):
     if process is None or process.poll() is not None:
-        # Step 1: Install dependencies
-        with st.spinner("Installing dependencies... This might take a while."):
-            install_process = subprocess.run(["pip3", "install", "-U", "-r", "requirements.txt"], text=True, capture_output=True)
-            if install_process.returncode == 0:
-                st.success("Dependencies installed successfully!")
-            else:
-                st.error("Error installing dependencies:")
-                st.error(install_process.stderr)
-                st.stop()  # Stop here if installation fails
-
-        # Step 2: Launch the bot and stream logs
-        process = subprocess.Popen(["python3", "-m", "Powers"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, preexec_fn=os.setsid, text=True)
-        st.success("`Powers` bot started successfully!")
+        with st.spinner("Installing dependencies..."):
+            subprocess.run(["pip3", "install", "-U", "-r", "requirements.txt"])
+        process = subprocess.Popen(["python3", "-m", "powers"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        st.success("`powers` process started!")
     else:
-        st.warning("`Powers` bot is already running!")
+        st.warning("`powers` is already running!")
 
+# Stop button: Terminates the `powers` process
 if st.button("Stop Powers"):
     if process is not None:
-        os.killpg(os.getpgid(process.pid), signal.SIGTERM)  # Stop the entire process group
+        process.terminate()  # Stop the process
         process = None
-        st.success("`Powers` bot stopped!")
+        st.success("`powers` process stopped!")
     else:
-        st.warning("No running bot process to stop.")
+        st.warning("No running process to stop.")
 
-# Check if the process is running and show logs dynamically
+# Display the status of the `powers` process
 if process and process.poll() is None:
-    st.write("`Powers` bot is **running**! 🚀")
-
-    # Display logs in real-time
-    with log_container:
-        st.text("Real-time Logs:")
-        while True:
-            output = process.stdout.readline()
-            if output == "" and process.poll() is not None:
-                break  # Stop loop when process ends
-            if output:
-                st.write(output.strip())  # Display log line by line in the UI
+    st.write("`powers` is **running**! 🚀")
 else:
-    st.write("`Powers` bot is **not running**.")
+    st.write("`powers` is **not running**.")