Added stop app message in bug hunter and troubleshooter

core/agents/bug_hunter.py

@@ -108,9 +108,9 @@ async def check_logs(self, logs_message: str = None):
         return AgentResponse.done(self)
 
     async def ask_user_to_test(self, awaiting_bug_reproduction: bool = False, awaiting_user_test: bool = False):
+        await self.ui.stop_app()
         test_instructions = self.current_state.current_iteration["bug_reproduction_description"]
         await self.send_message("You can reproduce the bug like this:\n\n" + test_instructions)
-
         await self.ui.send_test_instructions(test_instructions)
 
         if self.current_state.run_command:

core/agents/troubleshooter.py

@@ -74,6 +74,7 @@ async def create_iteration(self) -> AgentResponse:
         else:
             await self.send_message("Here are instructions on how to test the app:\n\n" + user_instructions)
 
+        await self.ui.stop_app()
         await self.ui.send_test_instructions(user_instructions)
 
         # Developer sets iteration as "completed" when it generates the step breakdown, so we can't

core/ui/base.py

@@ -356,6 +356,12 @@ async def generate_diff(
         """
         raise NotImplementedError()
 
+    async def stop_app(self):
+        """
+        Stop the App.
+        """
+        raise NotImplementedError()
+
     async def close_diff(self):
         """
         Close all diff views.

core/ui/console.py

@@ -168,6 +168,9 @@ async def generate_diff(
     ):
         pass
 
+    async def stop_app(self):
+        pass
+
     async def close_diff(self):
         pass
 

core/ui/ipc_client.py

@@ -51,6 +51,7 @@ class MessageType(str, Enum):
     MODIFIED_FILES = "modifiedFiles"
     IMPORTANT_STREAM = "importantStream"
     TEST_INSTRUCTIONS = "testInstructions"
+    STOP_APP = "stopApp"
 
 
 class Message(BaseModel):
@@ -460,6 +461,10 @@ async def generate_diff(
             },
         )
 
+    async def stop_app(self):
+        log.debug("Sending signal to stop the App")
+        await self._send(MessageType.STOP_APP)
+
     async def close_diff(self):
         log.debug("Sending signal to close the generated diff file")
         await self._send(MessageType.CLOSE_DIFF)

core/ui/virtual.py

@@ -164,6 +164,9 @@ async def generate_diff(
     ):
         pass
 
+    async def stop_app(self):
+        pass
+
     async def close_diff(self):
         pass