diff --git a/main.py b/main.py
index 8f8b750..c43f2c7 100644
--- a/main.py
+++ b/main.py
@@ -1136,9 +1136,9 @@ def main():
             
             # Otherwise, treat it as a chat message
             try:
-                print("MAIN: before invoke")
+                print("MAIN: before invoke", flush=True)
                 result = agent.invoke({"messages": [HumanMessage(content=user_input)]})
-                print(f"MAIN: after invoke, result type={type(result)}")
+                print(f"MAIN: after invoke, result type={type(result)}", flush=True)
                 
                 # Extract the AI's response (works for both simple and tool agent)
                 print(f"MAIN: result keys={result.keys() if hasattr(result, 'keys') else 'NO KEYS'}")