Prefer logger over print

Signed-off-by: Brianna Major <brianna.major@kitware.com>

Author: bnmajor

src/vtk_prompt/generate_files.py

@@ -187,15 +187,15 @@ def main(
                 f.write(xml_content)
             logger.info("VTK XML content written to %s", output)
         else:
-            print(xml_content)
+            logger.info(xml_content)
     else:
         logger.warning("Generated content may not be valid VTK XML")
         if output:
             with open(output, "w") as f:
                 f.write(xml_content)
             logger.info("Content written to %s (please verify)", output)
         else:
-            print(xml_content)
+            logger.info(xml_content)
 
 
 if __name__ == "__main__":

src/vtk_prompt/rag_chat_wrapper.py

@@ -239,7 +239,7 @@ def main(database: str, collection_name: str, top_k: int, model: str) -> None:
         try:
             reply = chat.ask(user_input, collection_name, top_k, streaming=True)
             for item in reply["response"]:
-                print(item.delta, end="")
+                logger.info(item.delta)
                 full_reply += item.delta
 
             logger.info("\n Here are some relevant references:")