Update logs

app.py

@@ -258,8 +258,6 @@ def text_to_text_generation(verbose, prompt):
         )
     response = response[len(is_kubectl_prompt) :]
 
-    print(f'{" Query Start ":-^40}')
-    print("Classified as: " + response)
 
     response_num = 0 if "0" in response else (1 if "1" in response else 2)
 
@@ -310,11 +308,17 @@ def text_to_text_generation(verbose, prompt):
         return (prompt, answer[start:end].strip())
 
     modes = ["Kubectl command", "Kubernetes definition", "Normal"]
+
+    print(f'{" Query Start ":-^40}')
+    print("Classified as: " + modes[response_num])
+
     modes[response_num] = f"**{modes[response_num]}**"
     modes = " / ".join(modes)
 
+
     if response_num == 2:
         prompt = create_generation_prompt(response_num, prompt, False)
+        print('Prompt given to model:\n' + prompt + '\n')
         original, new = generate_batch(prompt)[0]
         prompt, response = cleanup(original, new)
         if verbose:
@@ -324,6 +328,7 @@ def text_to_text_generation(verbose, prompt):
 
     if response_num == 0:
         prompt = create_generation_prompt(response_num, prompt, False)
+        print('Prompt given to model:\n' + prompt + '\n')
         original, new = generate_batch(prompt)[0]
         prompt, response = cleanup(original, new)
         model_response = new[len(original):].strip()
@@ -341,8 +346,11 @@ def text_to_text_generation(verbose, prompt):
             return f"{modes}\n\n" f"# Answer:\n" f"```bash\n{str_to_md(response)}\n```\n"
 
     res_prompt = create_generation_prompt(response_num, prompt, False)
+    print(f'Prompt given to finetuned model:\n{res_google_search_prompt}\n')
     res_semantic_search_prompt = create_generation_prompt(response_num, prompt, "semantic_search")
+    print(f'Prompt given to model with RAG:\n{res_semantic_search_prompt}\n')
     res_google_search_prompt = create_generation_prompt(response_num, prompt, "google_search")
+    print(f'Prompt given to model with Google search:\n{res_google_search_prompt}\n')
 
     gen_normal, gen_semantic_search, gen_google_search = generate_batch(
         res_prompt, res_semantic_search_prompt, res_google_search_prompt