Update train_agent.py

train_agent.py

@@ -1,16 +1,43 @@
 # train_agent.py
+import sys
+import os
+from pathlib import Path
+
+IS_COLAB = 'google.colab' in sys.modules
+
+# Get the current file's directory
+current_dir = Path(__file__).parent.absolute()
+
+# Search for agent.py in the current directory and its parent directories
+agent_path = None
+search_dir = current_dir
+while search_dir != search_dir.parent:  # Stop at root directory
+    possible_path = search_dir / 'agent.py'
+    if possible_path.exists():
+        agent_path = str(search_dir)
+        break
+    search_dir = search_dir.parent
+
+if agent_path:
+    sys.path.insert(0, agent_path)
+    print(f"Added {agent_path} to Python path")
+else:
+    print("Could not find agent.py")
+
+# Now try to import AutonomousWebAgent
+try:
+    from agent import AutonomousWebAgent
+    print("Successfully imported AutonomousWebAgent")
+except ImportError as e:
+    print(f"Error importing AutonomousWebAgent: {e}")
+    sys.exit(1)
 
+# Rest of your imports
 from twisted.internet import reactor, defer, task
 import random
 import logging
-import sys
 import time
 import codecs
-
-IS_COLAB = 'google.colab' in sys.modules
-from agent import AutonomousWebAgent
-
-
 # Configure logging
 if IS_COLAB:
     logging.basicConfig(level=logging.INFO,