move pretty printing to response_state_manager

charles_app.py

@@ -89,7 +89,6 @@ class CharlesApp:
         
         process_speech_to_text_future = []
         human_preview_text = ""
-        robot_preview_text = ""
         additional_prompt = None
         previous_prompt = ""
         is_talking = False
@@ -128,14 +127,10 @@ class CharlesApp:
 
                     if speaker_finished and len(prompt) > 0 and prompt not in prompts_to_ignore:
                         print(f"Prompt: {prompt}")
-                        line = ""
-                        for i, response in enumerate(response_state.current_responses):
-                            line += "🤖 " if len(line) == 0 else ""
-                            line += f"[{response_state.speech_chunks_per_response[i]}] {response}  \n"
-                        if len(line) > 0:
-                            await add_debug_output(line)
+                        response_preview_text = self._response_state_manager.pretty_print_current_responses()
+                        if len(response_preview_text) > 0:
+                            await add_debug_output(response_preview_text)
                         human_preview_text = ""
-                        robot_preview_text = ""
                         if additional_prompt is not None:
                             prompt = additional_prompt + ". " + prompt
                         await add_debug_output(f"👨 {prompt}")
@@ -167,20 +162,14 @@ class CharlesApp:
                         human_preview_text = f"👨❓ {prompt}"
 
             for new_response in response_step_obs.llm_responses:
-                # add_debug_output(f"🤖 {new_response}")
                 self._prompt_manager.append_assistant_message(new_response)
-                robot_preview_text = ""
-            if len(response_step_obs.llm_preview):
-                robot_preview_text = f"🤖❓ {response_step_obs.llm_preview}"
 
             list_of_strings = debug_output_history.copy()
-            line = ""
-            for i, response in enumerate(response_state.current_responses):
-                line += "🤖 " if len(line) == 0 else ""
-                line += f"[{response_state.speech_chunks_per_response[i]}] {response}  \n"
+            robot_preview_text = self._response_state_manager.pretty_print_preview_text()
+            response_preview_text = self._response_state_manager.pretty_print_current_responses()
             if len(robot_preview_text) > 0:
-                line += robot_preview_text+"  \n"
-            list_of_strings.append(line)
+                response_preview_text += robot_preview_text+"  \n"
+            list_of_strings.append(response_preview_text)
             if len(human_preview_text) > 0:
                 list_of_strings.append(human_preview_text)
             if len(list_of_strings) > 10:

response_state_manager.py

@@ -20,6 +20,7 @@ class ResponseState:
         self.step = step
         self.current_responses = []
         self.speech_chunks_per_response = []
+        self.llm_preview = ''
         self.is_speaking = False
 
     def __str__(self):
@@ -50,13 +51,28 @@ class ResponseStateManager:
 
     def set_llm_preview(self, llm_preview):
         self.response_step_obs.llm_preview = llm_preview
+        self.response_state.llm_preview = llm_preview
 
     def add_llm_response_and_clear_llm_preview(self, llm_response):
         self.response_state.current_responses.append(llm_response)
         self.response_state.speech_chunks_per_response.append(0)
         self.response_step_obs.llm_responses.append(llm_response)
         self.response_step_obs.llm_preview = ''
+        self.response_state.llm_preview = ''
 
     def add_tts_raw_chunk_id(self, chunk_id, llm_sentence_id):
         self.response_state.speech_chunks_per_response[llm_sentence_id] += 1
-        self.response_step_obs.tts_raw_chunk_ids.append(chunk_id)
+        self.response_step_obs.tts_raw_chunk_ids.append(chunk_id)
+
+    def pretty_print_current_responses(self)->str:
+        line = ""
+        for i, response in enumerate(self.response_state.current_responses):
+            line += "🤖 " if len(line) == 0 else ""
+            line += f"[{self.response_state.speech_chunks_per_response[i]}] {response}  \n"
+        return line
+    
+    def pretty_print_preview_text(self)->str:
+        robot_preview_text = ""
+        if len(self.response_state.llm_preview):
+                robot_preview_text = f"🤖❓ {self.response_state.llm_preview}"
+        return robot_preview_text