fix the issue task completed will not shown in chat history

chatbot_simulator.py

@@ -170,7 +170,7 @@ Example Output Format:
 
         try:
             if updated_state['task_completed'].lower() == 'true':
-                print(f"Task completed! You took {self.prompt_count} steps.")
+                return f"Task completed! You took {self.prompt_count} steps."
         except:
             updated_state['task_completed'] = 'False'
 
@@ -185,32 +185,6 @@ Example Output Format:
         self.conversation.append({"role": "assistant", "content": gpt_instruction})
         return gpt_instruction
 
-    def _generate_agent_input(self):
-        """
-        Simulate the agent generating input based on the conversation state.
-        """
-        agent_prompt = f"""
-Imagine you are an agent navigate through the Uber environment.
-Your overarching task is: {self.task}. You may have done some part of the task, or none at all.
-You will have access to all of your previous actions in the environment, as well as the last message from the assistant giving the current state of the environment.
-The last message from the assistant was: {self.conversation[-1]['content']}
-Respond first with a brief "Plan" which suggests what steps you are going to take to accomplish the task, and what your immediate. 
-Then generate an "Action" which is the immediate next step you can take.
-        """
-
-        messages = [{"role": "system", "content": agent_prompt}]
-        for action in self.actions:
-            messages.append({"role": "user", "content": action})
-        messages.append({"role": "assistant", "content": self.conversation[-1]['content']})
-
-        agent_response = self.client.chat.completions.create(
-            model="gpt-4",
-            messages=messages,
-            temperature=0.7,
-        )
-        print(f"LLM-Agent: {agent_response.choices[0].message.content}")
-        return agent_response.choices[0].message.content
-
     def start_conversation(self):
         greeting = f'\n Welcome to {self.app_name} simulator! Your task is: {self.task}. \n'
         system_prompt = self._generate_system_prompt()