Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import subprocess
+import sys
 
 # Install BrowserGym dependencies before running the main application
 def install_browsergym():
@@ -7,23 +8,38 @@ def install_browsergym():
         print("Installing BrowserGym dependencies...")
         subprocess.run("cd BrowserGym && make install", shell=True, check=True)
         print("BrowserGym installation completed successfully")
+        
+        # Add BrowserGym directory to sys.path directly
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        browsergym_path = os.path.join(current_dir, "BrowserGym")
+        if browsergym_path not in sys.path:
+            sys.path.insert(0, browsergym_path)
+            print(f"Added BrowserGym to sys.path: {browsergym_path}")
+        
+        # Also set PYTHONPATH environment variable for child processes
+        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']}")
+        
+        # Verify BrowserGym is importable
+        try:
+            import browsergym.core
+            print("BrowserGym successfully imported")
+        except ImportError as e:
+            print(f"BrowserGym import verification failed: {e}")
+            print(f"Current sys.path: {sys.path}")
+            raise
+            
     except subprocess.CalledProcessError as e:
         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 first
 install_browsergym()
 
-import os
+# Now import the rest of the modules
 import logging
 import gradio as gr
 import openai