✨ Aurora Dashboard
Streamlined task orchestration for BroJS. Send tasks, monitor progress, and control your course automation—all in one place.
""", unsafe_allow_html=True)
# ──── SESSION STATE ────
for key in ["chat_history", "chat_events", "chat_thread_id", "task_statuses"]:
if key not in st.session_state:
if key == "chat_thread_id":
st.session_state[key] = f"aurora-{int(time.time())}"
else:
st.session_state[key] = [] if key != "task_statuses" else {}
# ──── AGENT HELPERS ────
@st.cache_resource(show_spinner="🚀 Starting Aurora engine...")
def get_agent():
from src.agent.agent import agent
return agent
@st.cache_resource(show_spinner="📡 Loading orchestrator...")
def get_main_agent():
from src.agent.agent import agent
return agent
def _load_statuses():
from src.agent.mcp_client import load_journal_toolsets
async def _fetch():
j = load_journal_toolsets()
tools = {t.name: t for t in j.tasks_submissions_tools}
key = "mcp__journal-bh-professor__tasks_list"
tool = tools.get(key) or next((v for k, v in tools.items() if "tasks_list" in k), None)
if not tool:
return [], f"Task list tool not found."
raw = await tool.ainvoke({"courseId": "698b49da77cb6d4d2e43ce78"})
text = next((x["text"] for x in raw if x.get("type") == "text"), str(raw)) if isinstance(raw, list) else str(raw)
data = json.loads(text)
return (data.get("tasks", data) if isinstance(data, dict) else data), None
result = {"items": [], "error": None}
def _run():
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
try:
result["items"], result["error"] = loop.run_until_complete(_fetch())
except Exception as e:
result["error"] = str(e)
finally:
loop.close()
t = threading.Thread(target=_run, daemon=True)
t.start()
t.join()
return result["items"], result["error"]
# ──── CALLBACKS & RUNNERS ────
_SOLVE_TOOLS = {"validate_teacher_comment", "generate_code_solution"}
_JOURNAL_PREFIX = "mcp__journal-bh-professor__"
class AgentCallback(BaseCallbackHandler):
def __init__(self, q: queue.Queue):
self.q = q
def _ts(self) -> str:
return datetime.now().strftime("%H:%M:%S")
def on_tool_start(self, serialized, input_str, **kwargs):
name = serialized.get("name", "?")
try:
args = json.loads(str(input_str)) if isinstance(input_str, str) else input_str
except:
args = {}
self.q.put({"t": "tool_start", "ts": self._ts(), "name": name, "args": args})
def on_tool_end(self, output, **kwargs):
self.q.put({"t": "tool_end", "ts": self._ts(), "output": str(output)[:280]})
def on_tool_error(self, error, **kwargs):
self.q.put({"t": "tool_error", "ts": self._ts(), "msg": str(error)[:200]})
def on_llm_start(self, *a, **kw):
self.q.put({"t": "thinking", "ts": self._ts()})
def on_llm_end(self, response, **kwargs):
try:
text = response.generations[0][0].text[:120]
self.q.put({"t": "llm_end", "ts": self._ts(), "preview": text})
except:
pass
def _run_agent_thread(agent, messages, config, q: queue.Queue, cb: AgentCallback):
async def _inner():
try:
result = await agent.ainvoke(messages, {**config, "callbacks": [cb]})
q.put({"t": "done", "result": result})
except Exception as e:
q.put({"t": "fatal", "msg": str(e)})
asyncio.run(_inner())
def _render_event(ev: dict) -> str:
ts = ev.get("ts", "")
kind = ev.get("t", "")
if kind == "thinking":
return f'💭 {ts} Thinking...
'
if kind == "tool_start":
name = ev["name"]
args = ev.get("args", {})
short = name.replace(_JOURNAL_PREFIX, "mcp::")
icon = "🔧"
if name in _SOLVE_TOOLS:
icon = "🧠"
short = f"[Subagent] {short}"
elif "gitea" in name:
icon = "📦"
elif "journal" in name:
icon = "📡"
hint = ""
for k in ("taskId", "path", "repo", "repo_name", "name"):
if k in args:
hint = f' — {args[k]}'
break
return f'✏ {ts} {preview}...
'
return ""
# ──── TABS ────
tab1, tab2, tab3 = st.tabs(["💬 Chat", "⚡ Pipeline", "📊 Status"])
# ═══════════════════════════════════════════════════════════════
# TAB 1: CHAT
# ═══════════════════════════════════════════════════════════════
with tab1:
st.subheader("Ask Aurora")
st.write("Chat with the agent. Ask questions, request task solutions, or check your course progress.")
# Display chat history
chat_col = st.container()
with chat_col:
for msg in st.session_state.chat_history:
if msg["role"] == "user":
st.markdown(f'