Alexander Miroshnichenko
b8cb8908e7
Some checks failed
CI / check (push) Has been cancelled
Start engine only after MainScreen mounts, while preserving explicit MainScreen widget lookup.
460 lines
22 KiB
Diff
460 lines
22 KiB
Diff
--- a/haivemind_tui.py
|
|
+++ b/haivemind_tui.py
|
|
@@ -1197,0 +1198,8 @@
|
|
+ def on_mount(self) -> None:
|
|
+ app = self.app
|
|
+ if (
|
|
+ getattr(app, "_prompt", None) is not None
|
|
+ and getattr(app, "_config", None) is not None
|
|
+ ):
|
|
+ app.start_run(app._prompt, app._config)
|
|
+
|
|
@@ -1220,13 +1220,13 @@
|
|
self._engine: HaivemindEngine | None = None
|
|
self._run_started = False
|
|
self._run_start_time: float = 0.0
|
|
+ self._main_screen = None
|
|
|
|
def on_mount(self) -> None:
|
|
- self.push_screen(MainScreen())
|
|
+ self._main_screen = MainScreen()
|
|
+ self.push_screen(self._main_screen)
|
|
self.set_interval(0.15, self._tick_spinner)
|
|
- if self._prompt is not None and self._config is not None:
|
|
- self.start_run(self._prompt, self._config)
|
|
- else:
|
|
+ if self._prompt is None or self._config is None:
|
|
self.push_screen(ConfigScreen())
|
|
|
|
def start_run(self, prompt: str, config: HaivemindConfig) -> None:
|
|
@@ -1303,7 +1303,7 @@
|
|
def _tick_spinner(self) -> None:
|
|
_advance_spinner()
|
|
try:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
if "streaming" in hbar._model_status.values():
|
|
hbar._refresh()
|
|
except NoMatches:
|
|
@@ -1315,20 +1315,20 @@
|
|
def _on_probe_start(self) -> None:
|
|
import time as _time
|
|
self._run_start_time = _time.time()
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
chat.add_summary("[bold yellow]Probing models...[/]")
|
|
if self._config is not None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_models(self._config.models)
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.consensus_info = "probing"
|
|
header.progress_pct = 0.0
|
|
|
|
@on(ProbeResult)
|
|
def _on_probe_result(self, event: ProbeResult) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "done" if event.ok else "failed")
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
if event.ok:
|
|
color = hbar.model_color(event.model)
|
|
chat.add_summary(f"[{color}][probe] ok: {event.model}[/]")
|
|
@@ -1339,44 +1339,44 @@
|
|
|
|
@on(ProbeDone)
|
|
def _on_probe_done(self, event: ProbeDone) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
chat.add_summary(
|
|
f"[bold yellow]Probe complete:[/] {len(event.healthy)} healthy, "
|
|
f"{len(event.failures)} failed"
|
|
)
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.consensus_info = ""
|
|
if event.healthy:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_models(event.healthy)
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.update_layout(event.healthy, hbar._model_colors)
|
|
|
|
@on(RoundStart)
|
|
def _on_round_start(self, event: RoundStart) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
chat.add_summary(f"\n[bold cyan]\u2500\u2500 Round {event.round_idx}/{event.total} \u2500\u2500[/]")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.round_info = f"hAIvemind \u2502 Round {event.round_idx}/{event.total}"
|
|
header.consensus_info = ""
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.clear_all()
|
|
except NoMatches:
|
|
pass
|
|
|
|
@on(ModelStart)
|
|
def _on_model_start(self, event: ModelStart) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "streaming")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
total = len(hbar._model_status) if hbar._model_status else 0
|
|
done_count = sum(
|
|
1 for s in hbar._model_status.values() if s in ("streaming", "done")
|
|
)
|
|
header.model_info = f"Models: {done_count}/{total}"
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
active = [
|
|
m for m, s in hbar._model_status.items() if s != "dropped"
|
|
]
|
|
@@ -1386,27 +1386,27 @@
|
|
|
|
@on(ModelStream)
|
|
def _on_model_stream(self, event: ModelStream) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
color = hbar.model_color(event.model)
|
|
t = event.fragment.strip()
|
|
if t:
|
|
chat.add_summary(f"[{color}][{event.model}][/] {t}")
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.stream_to_model(event.model, event.fragment)
|
|
except NoMatches:
|
|
pass
|
|
|
|
@on(ModelDone)
|
|
def _on_model_done(self, event: ModelDone) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
ar = event.agent_round
|
|
if ar.error and ar.error not in ("stderr",):
|
|
hbar.set_status(event.model, "failed")
|
|
else:
|
|
hbar.set_status(event.model, "done")
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
color = hbar.model_color(event.model)
|
|
if ar.has_valid_draft:
|
|
chat.add_summary(
|
|
@@ -1419,14 +1419,14 @@
|
|
|
|
@on(ModelDropped)
|
|
def _on_model_dropped(self, event: ModelDropped) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "dropped")
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
chat.add_summary(
|
|
f"[red]\u26a0 {event.model} dropped:[/] {event.reason}"
|
|
)
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
remaining = [
|
|
m for m, s in hbar._model_status.items() if s != "dropped"
|
|
]
|
|
@@ -1436,7 +1436,7 @@
|
|
|
|
@on(RoundDone)
|
|
def _on_round_done(self, event: RoundDone) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
dur = event.record.get("duration_s", 0)
|
|
chat.add_summary(
|
|
f"[dim]\u2500\u2500 end round {event.round_idx} ({dur}s) \u2500\u2500[/]"
|
|
@@ -1446,15 +1446,15 @@
|
|
best_sim = _compute_best_similarity(drafts)
|
|
threshold = self._config.threshold if self._config else 0.75
|
|
pct = min(1.0, best_sim / threshold) if threshold > 0 else 0.0
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.progress_pct = pct
|
|
header.consensus_info = f"sim: {best_sim:.2f}"
|
|
|
|
@on(Consensus)
|
|
def _on_consensus(self, event: Consensus) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
total_models = len(hbar._model_status) if hbar._model_status else 0
|
|
import time as _time
|
|
elapsed = (
|
|
@@ -1493,9 +1493,9 @@
|
|
|
|
@on(Fallback)
|
|
def _on_fallback(self, event: Fallback) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
total_models = len(hbar._model_status) if hbar._model_status else 0
|
|
header.consensus_info = "fallback"
|
|
header.progress_pct = 0.0
|
|
@@ -1518,30 +1518,30 @@
|
|
|
|
@on(EngineError)
|
|
def _on_engine_error(self, event: EngineError) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
chat.add_summary(f"[red]Error:[/] {event.message}")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.consensus_info = "error"
|
|
|
|
@on(EngineLog)
|
|
def _on_engine_log(self, event: EngineLog) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
chat.add_summary(event.message)
|
|
|
|
@on(MoaRefStart)
|
|
def _on_moa_ref_start(self, event: MoaRefStart) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "streaming")
|
|
color = hbar.model_color(event.model)
|
|
chat.add_summary(f"[{color}]\U0001f4e1 [ref] {event.model} analyzing...[/]")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.round_info = "hAIvemind \u2502 MoA \u2502 References"
|
|
|
|
@on(MoaRefDone)
|
|
def _on_moa_ref_done(self, event: MoaRefDone) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
color = hbar.model_color(event.model)
|
|
if event.err and not event.raw.strip():
|
|
hbar.set_status(event.model, "failed")
|
|
@@ -1552,32 +1552,32 @@
|
|
|
|
@on(MoaRefStream)
|
|
def _on_moa_ref_stream(self, event: MoaRefStream) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
color = hbar.model_color(event.model)
|
|
t = event.fragment.strip()
|
|
if t:
|
|
chat.add_summary(f"[{color}][ref:{event.model}][/] {t}")
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.stream_to_model(event.model, event.fragment)
|
|
except NoMatches:
|
|
pass
|
|
|
|
@on(MoaAggStart)
|
|
def _on_moa_agg_start(self, event: MoaAggStart) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "streaming")
|
|
color = hbar.model_color(event.model)
|
|
chat.add_summary(f"\n[bold {color}]\U0001f9e0 [aggregator] {event.model} synthesizing...[/]")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.round_info = "hAIvemind \u2502 MoA \u2502 Aggregating"
|
|
|
|
@on(MoaAggDone)
|
|
def _on_moa_agg_done(self, event: MoaAggDone) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
color = hbar.model_color(event.model)
|
|
if event.raw.strip():
|
|
hbar.set_status(event.model, "done")
|
|
@@ -1588,14 +1588,14 @@
|
|
|
|
@on(MoaAggStream)
|
|
def _on_moa_agg_stream(self, event: MoaAggStream) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
color = hbar.model_color(event.model)
|
|
t = event.fragment.strip()
|
|
if t:
|
|
chat.add_summary(f"[{color}][agg:{event.model}][/] {t}")
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.stream_to_model(event.model, event.fragment)
|
|
except NoMatches:
|
|
pass
|
|
@@ -1604,18 +1604,18 @@
|
|
|
|
@on(FusionPanelStart)
|
|
def _on_fusion_panel_start(self, event: FusionPanelStart) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "streaming")
|
|
color = hbar.model_color(event.model)
|
|
chat.add_summary(f"[{color}]\U0001f52c [panel] {event.model} analyzing...[/]")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.round_info = "hAIvemind \u2502 Fusion \u2502 Panel"
|
|
|
|
@on(FusionPanelDone)
|
|
def _on_fusion_panel_done(self, event: FusionPanelDone) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
color = hbar.model_color(event.model)
|
|
if event.err and not event.raw.strip():
|
|
hbar.set_status(event.model, "failed")
|
|
@@ -1626,32 +1626,32 @@
|
|
|
|
@on(FusionPanelStream)
|
|
def _on_fusion_panel_stream(self, event: FusionPanelStream) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
color = hbar.model_color(event.model)
|
|
t = event.fragment.strip()
|
|
if t:
|
|
chat.add_summary(f"[{color}][panel:{event.model}][/] {t}")
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.stream_to_model(event.model, event.fragment)
|
|
except NoMatches:
|
|
pass
|
|
|
|
@on(FusionJudgeStart)
|
|
def _on_fusion_judge_start(self, event: FusionJudgeStart) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "streaming")
|
|
color = hbar.model_color(event.model)
|
|
chat.add_summary(f"\n[bold {color}]\u2696\ufe0f [judge] {event.model} analyzing...[/]")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.round_info = "hAIvemind \u2502 Fusion \u2502 Judging"
|
|
|
|
@on(FusionJudgeDone)
|
|
def _on_fusion_judge_done(self, event: FusionJudgeDone) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
color = hbar.model_color(event.model)
|
|
if event.raw.strip():
|
|
hbar.set_status(event.model, "done")
|
|
@@ -1662,32 +1662,32 @@
|
|
|
|
@on(FusionJudgeStream)
|
|
def _on_fusion_judge_stream(self, event: FusionJudgeStream) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
color = hbar.model_color(event.model)
|
|
t = event.fragment.strip()
|
|
if t:
|
|
chat.add_summary(f"[{color}][judge:{event.model}][/] {t}")
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.stream_to_model(event.model, event.fragment)
|
|
except NoMatches:
|
|
pass
|
|
|
|
@on(FusionSynthStart)
|
|
def _on_fusion_synth_start(self, event: FusionSynthStart) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
hbar.set_status(event.model, "streaming")
|
|
color = hbar.model_color(event.model)
|
|
chat.add_summary(f"\n[bold {color}]\u270d\ufe0f [synth] {event.model} writing final answer...[/]")
|
|
- header = self.query_one("#header-bar", HeaderBar)
|
|
+ header = self._main_screen.query_one("#header-bar", HeaderBar)
|
|
header.round_info = "hAIvemind \u2502 Fusion \u2502 Synthesizing"
|
|
|
|
@on(FusionSynthDone)
|
|
def _on_fusion_synth_done(self, event: FusionSynthDone) -> None:
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
color = hbar.model_color(event.model)
|
|
if event.raw.strip():
|
|
hbar.set_status(event.model, "done")
|
|
@@ -1698,14 +1698,14 @@
|
|
|
|
@on(FusionSynthStream)
|
|
def _on_fusion_synth_stream(self, event: FusionSynthStream) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
- hbar = self.query_one("#health-bar", ModelHealthBar)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
+ hbar = self._main_screen.query_one("#health-bar", ModelHealthBar)
|
|
color = hbar.model_color(event.model)
|
|
t = event.fragment.strip()
|
|
if t:
|
|
chat.add_summary(f"[{color}][synth:{event.model}][/] {t}")
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.stream_to_model(event.model, event.fragment)
|
|
except NoMatches:
|
|
pass
|
|
@@ -1720,7 +1720,7 @@
|
|
|
|
def action_toggle_theatre(self) -> None:
|
|
try:
|
|
- tc = self.query_one("#theatre-container")
|
|
+ tc = self._main_screen.query_one("#theatre-container")
|
|
if tc.has_class("visible"):
|
|
tc.remove_class("visible")
|
|
else:
|
|
@@ -1734,13 +1734,13 @@
|
|
|
|
def action_clear_theatre(self) -> None:
|
|
try:
|
|
- theatre = self.query_one("#theatre-container", TheatrePanel)
|
|
+ theatre = self._main_screen.query_one("#theatre-container", TheatrePanel)
|
|
theatre.clear_all()
|
|
except NoMatches:
|
|
pass
|
|
|
|
def action_show_artifacts(self) -> None:
|
|
- chat = self.query_one("#chat-log", ChatLog)
|
|
+ chat = self._main_screen.query_one("#chat-log", ChatLog)
|
|
if self._engine and self._engine.run_dir:
|
|
chat.add_summary(f"[bold]Artifacts:[/] {self._engine.run_dir}")
|
|
else:
|