Update app.py

app.py

@@ -1,38 +1,52 @@
 import streamlit as st
 import asyncio
 from playwright.async_api import async_playwright
-import threading
 
-logs = []
+if "logs" not in st.session_state:
+    st.session_state.logs = []
+
+if "running" not in st.session_state:
+    st.session_state.running = False
+
 
 async def run_bot(url):
+    st.session_state.logs.append("Starting bot...")
+
     async with async_playwright() as p:
-        browser = await p.chromium.launch(headless=True)
+        browser = await p.chromium.launch(
+            headless=True,
+            args=["--no-sandbox", "--disable-dev-shm-usage"]
+        )
+
         page = await browser.new_page()
 
-        page.on("console", lambda msg: logs.append(f"LOG: {msg.text}"))
-        page.on("request", lambda req: logs.append(f"REQ: {req.url}"))
-        page.on("response", lambda res: logs.append(f"RES: {res.url} {res.status}"))
+        page.on("console", lambda msg: st.session_state.logs.append(f"LOG: {msg.text}"))
+        page.on("request", lambda req: st.session_state.logs.append(f"REQ: {req.url}"))
+        page.on("response", lambda res: st.session_state.logs.append(f"RES: {res.url} {res.status}"))
 
         await page.goto(url)
-        await page.evaluate("console.log('Injected script running');")
+
+        # keep session alive
+        for _ in range(15):
+            await asyncio.sleep(1)
+
         await browser.close()
 
-# Wrapper to run async function in thread
-def run_bot_thread(url):
-    asyncio.run(run_bot(url))
+    st.session_state.logs.append("Bot finished.")
+    st.session_state.running = False
+
 
 st.title("DevTools Bot Dashboard")
 
-target_url = st.text_input("Enter the URL to monitor:", "https://example.com")
+url = st.text_input("Enter URL", "https://example.com")
+
+if st.button("Run Bot") and not st.session_state.running:
+    st.session_state.logs.clear()
+    st.session_state.running = True
+    asyncio.run(run_bot(url))
 
-if st.button("Run Bot"):
-    logs.clear()
-    # Start bot in separate thread
-    threading.Thread(target=run_bot_thread, args=(target_url,), daemon=True).start()
-    st.success("Bot started in background. Logs will appear below shortly.")
 
-# Display logs
 st.subheader("Logs")
-for log in logs:
+
+for log in st.session_state.logs:
     st.text(log)