Update app.py

app.py

@@ -25,13 +25,16 @@ def install_browsergym():
         
         # Verify BrowserGym is importable
         try:
-            from browsergym.core.action.highlevel import HighLevelActionSet
-            from browsergym.utils.obs import (
-                flatten_axtree_to_str,
-                flatten_dom_to_str,
-                prune_html,
-            )
-            from browsergym.experiments import Agent
+            import importlib.util
+
+            # Try to import HighLevelActionSet
+            spec = importlib.util.find_spec("browsergym.core.action.highlevel")
+            if spec is None:
+                print("Module browsergym.core.action.highlevel not found")
+            else:
+                module = importlib.util.module_from_spec(spec)
+                spec.loader.exec_module(module)
+                HighLevelActionSet = getattr(module, "HighLevelActionSet")
             print("BrowserGym successfully imported")
         except ImportError as e:
             print(f"BrowserGym import verification failed: {e}")