changes

app.py

@@ -1,19 +1,10 @@
 import streamlit as st
-import io, sys, time
 from utils import *
 from client import CobotController
 
-# render_log_window()
-
-class StreamlitRedirector:
-    def write(self, message):
-        if message.strip():  # Avoids writing empty lines
-            st.write(message)
-
-# Set up the redirection
-sys.stdout = StreamlitRedirector()
-
+render_log_window()
 
 user_id = get_user_id()
 user, passwd, host, endpoint, port = get_credentials(False)
-client = CobotController(user, passwd, host, port, endpoint, user_id)
+client = CobotController(user, passwd, host, port, endpoint, user_id)
+refresh_once()

utils.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from streamlit_js_eval import streamlit_js_eval
 import uuid, io, sys, os
 
 def get_user_id():
@@ -26,40 +27,17 @@ def get_credentials(use_own_creds: bool):
 		PORT = int(os.environ.get("PORT", 8883))
 	return HIVEMQ_USERNAME, HIVEMQ_PASSWORD, HIVEMQ_HOST, DEVICE_ENDPOINT, PORT
 
-class StdoutRedirector(io.StringIO):
-	def __init__(self, log_area, max_lines=20):
-		super().__init__()
-		self.log_area = log_area
-		self.logs = []
-		self.max_lines = max_lines
-
-	def write(self, message):
-		if message.strip():
-			self.logs.append(message.strip())
-			if len(self.logs) > self.max_lines:
-				self.logs.pop(0)
-			self.log_area.markdown(
-				f"<div class='log-window'>{'<br>'.join(self.logs)}</div>", 
-				unsafe_allow_html=True
-			)
-
 def render_log_window():
-	st.markdown("""
-		<style>
-		.log-window {
-			height: 150px;
-			overflow-y: auto;
-			font-family: monospace;
-			background-color: var(--background-color);
-			color: var(--text-color);
-			padding: 10px;
-			border-radius: 5px;
-			border: 2px solid white;
-		}
-		</style>
-	""", unsafe_allow_html=True)
+	class StreamlitRedirector:
+		def write(self, message):
+			if message.strip():
+				st.write(message)
 
-	log_area = st.empty()
-	stdout_redirector = StdoutRedirector(log_area)
+	sys.stdout = StreamlitRedirector()
 
-	sys.stdout = stdout_redirector
+def refresh_once():
+	has_refreshed = st.query_params.get("has_refreshed", None)
+	if not has_refreshed:
+		st.query_params["has_refreshed"] = True
+		sys.stdout = sys.__stdout__
+		streamlit_js_eval(js_expressions="parent.window.location.reload()")