Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import subprocess
-import multiprocessing
 
 # Install BrowserGym dependencies before running the main application
 def install_browsergym():
@@ -12,6 +11,16 @@ def install_browsergym():
         print(f"Error installing BrowserGym: {e}")
         raise
 
+# Add BrowserGym to Python path
+current_dir = os.path.dirname(os.path.abspath(__file__))
+browsergym_path = os.path.join(current_dir, "BrowserGym")
+if "PYTHONPATH" in os.environ:
+    os.environ["PYTHONPATH"] = f"{browsergym_path}:{os.environ['PYTHONPATH']}"
+else:
+    os.environ["PYTHONPATH"] = browsergym_path
+print(f"Updated PYTHONPATH: {os.environ['PYTHONPATH']}")
+
+
 install_browsergym()
 
 import os