Update app.py

app.py

@@ -27,10 +27,25 @@ def capture_output():
     old_stdout = sys.stdout
     old_stderr = sys.stderr
 
+    class Tee:
+        def __init__(self, buffer, original):
+            self.buffer = buffer
+            self.original = original
+
+        def write(self, data):
+            self.buffer.write(data)
+            self.original.write(data)
+
+        def flush(self):
+            self.buffer.flush()
+            self.original.flush()
+
     try:
-        sys.stdout = StringIO()
-        sys.stderr = StringIO()
-        yield sys.stdout, sys.stderr
+        stdout_buffer = StringIO()
+        stderr_buffer = StringIO()
+        sys.stdout = Tee(stdout_buffer, old_stdout)
+        sys.stderr = Tee(stderr_buffer, old_stderr)
+        yield stdout_buffer, stderr_buffer
     finally:
         sys.stdout = old_stdout
         sys.stderr = old_stderr