diff --git a/tools/stdout_wrapper.py b/tools/stdout_wrapper.py
new file mode 100644
index 0000000..c2c9fe0
--- /dev/null
+++ b/tools/stdout_wrapper.py
@@ -0,0 +1,17 @@
+import sys
+
+
+class StdoutWrapper:
+    def write(self, message: str):
+        print(message, end="")
+
+    def flush(self):
+        pass
+
+
+def get_stdout():
+    # Colab 環境をチェックする
+    if "google.colab" in sys.modules:
+        return StdoutWrapper()
+    else:
+        return sys.stdout