diff --git a/core/agents/frontend.py b/core/agents/frontend.py index ff76eb505b..60ee86e444 100644 --- a/core/agents/frontend.py +++ b/core/agents/frontend.py @@ -25,7 +25,7 @@ from core.llm.convo import Convo from core.llm.parser import DescriptiveCodeBlockParser, OptionalCodeBlockParser from core.log import get_logger from core.telemetry import telemetry -from core.ui.base import ProjectStage +from core.ui.base import ProjectStage, UISource log = get_logger(__name__) @@ -168,6 +168,10 @@ class Frontend(FileDiffMixin, GitMixin, BaseAgent): if user_input: await self.send_message("Errors detected, fixing...") else: + await self.ui.send_message( + "Use code CODE20 and subscribe https://pythagora.ai/pricing", + source=UISource("Congratulations", "success"), + ) answer = await self.ask_question( "Do you want to change anything or report a bug?" if frontend_only else FE_CHANGE_REQ, buttons={"yes": "I'm done building the UI"} if not frontend_only else None, diff --git a/core/agents/spec_writer.py b/core/agents/spec_writer.py index 2ca25c0d02..6d95a9ac5a 100644 --- a/core/agents/spec_writer.py +++ b/core/agents/spec_writer.py @@ -11,7 +11,7 @@ from core.llm.parser import StringParser from core.log import get_logger from core.telemetry import telemetry from core.templates.registry import PROJECT_TEMPLATES -from core.ui.base import ProjectStage +from core.ui.base import ProjectStage, UISource log = get_logger(__name__) @@ -122,6 +122,9 @@ class SpecWriter(BaseAgent): await self.ui.send_front_logs_headers("specs_0", ["E1 / T1", "Writing Specification", "working"], "") + await self.ui.send_message( + "Use code CODE20 and subscribe https://pythagora.ai/pricing", source=UISource("Congratulations", "success") + ) await self.send_message( "## Write specification\n\nPythagora is generating a detailed specification for app based on your input.", # project_state_id="setup",