Better state management

src/app.py

@@ -27,6 +27,7 @@ def get_response(prompt):
     
 def get_output(context, user_input):
     try:
+        config.status.update(label=":bulb: Preparing answer..")
         prompt_schema = llm.prompter.read_yaml_file("prompt_assets/output_sys_prompt.yml")
         sys_prompt = llm.prompter.format_yaml_prompt(prompt_schema, dict()) + \
             f"Information:\n{context}"

src/functioncall.py

@@ -83,7 +83,6 @@ class ModelInference:
         return results_dict
     
     def run_inference(self, prompt: List[Dict[str, str]]):
-        config.status.update(label=":brain: Thinking..")
         inputs = self.tokenizer.apply_chat_template(
             prompt,
             add_generation_prompt=True,
@@ -108,6 +107,7 @@ class ModelInference:
             chat = [{"role": "user", "content": user_message}]
             tools = functions.get_openai_tools()
             prompt = self.prompter.generate_prompt(chat, tools, num_fewshot)
+            config.status.update(label=":brain: Thinking..")
             completion = self.run_inference(prompt)
 
             def recursive_loop(prompt, completion, depth):
@@ -140,6 +140,7 @@ class ModelInference:
                         completion = self.run_inference(prompt)
                         return completion
 
+                    config.status.update(label=":brain: Analysing information..")
                     completion = self.run_inference(prompt)
                     return recursive_loop(prompt, completion, depth)
                 elif error_message: