Update app.py

app.py

@@ -2,163 +2,18 @@ import os
 import subprocess
 import sys
 
-# Install BrowserGym dependencies before running the main application
-def install_browsergym():
-    try:
-        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 importlib.util
+print("Installing BrowserGym dependencies...")
+subprocess.run("cd BrowserGym && make install", shell=True, check=True)
+print("BrowserGym installation completed successfully")
 
-            # 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}")
-            print(f"Current sys.path: {sys.path}")
-            raise
-            
-    except subprocess.CalledProcessError as e:
-        print(f"Error installing BrowserGym: {e}")
-        raise
+# Add BrowserGym directory to sys.path directly
+current_dir = os.path.dirname(os.path.abspath(__file__))
+browsergym_path = os.path.join(current_dir, "BrowserGym")
+print(f"Added BrowserGym path: {browsergym_path}")
 
-# Install BrowserGym first
-install_browsergym()
+# Prepare env with updated PYTHONPATH
+env = os.environ.copy()
+env["PYTHONPATH"] = f"{browsergym_path}:{env.get('PYTHONPATH', '')}"
 
-# Now import the rest of the modules
-import logging
-import gradio as gr
-import openai
-import multiprocessing
-
-from process_run import process_run
-
-# Configure logging
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    handlers=[
-        logging.StreamHandler(),
-    ]
-)
-logger = logging.getLogger(__name__)
-logger.setLevel('INFO')
-
-# Set your OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-
-# Example instructions to display
-EXAMPLES = [
-    "When did the solar system form? Find on wikipedia.",
-    "Find the rating of Monopoly (1935) on boardgamegeek.com",
-]
-
-URL_EXAMPLES = [
-    "about:blank",
-    "https://www.wikipedia.org",
-    "https://www.boardgamegeek.com"
-]
-
-def main():
-    logger.info("Starting BrowserGym web agent")
-    
-    with gr.Blocks(title="WebShephered Demo") as demo:
-        # Add CSS for outlined groups
-        gr.Markdown("# WebShephered Demo")
-        with gr.Row():
-            with gr.Column(scale=2):
-                with gr.Column():
-                    instruction = gr.Textbox(
-                        label="Instruction",
-                        placeholder="Enter your instruction here",
-                        lines=2,
-                    )
-                    gr.Examples(
-                        examples=[[e] for e in EXAMPLES],
-                        inputs=instruction,
-                        cache_examples=False,
-                    )
-                
-                gr.Markdown("\n\n")
-                
-                with gr.Column():
-                    start_url = gr.Textbox(
-                        label="Starting URL",
-                        placeholder="URL to start the browser at",
-                        value="about:blank"
-                    )
-                    gr.Examples(
-                        examples=URL_EXAMPLES,
-                        inputs=start_url,
-                        cache_examples=False,
-                    )
-
-                gr.Markdown("\n\n")
-
-                model_name = gr.Dropdown(
-                    label="Agent Model",
-                    choices=["gpt-4o"],
-                    value="gpt-4o"
-                )
-                run_btn = gr.Button("Run Demo")
-
-                gr.Markdown("---")
-                
-                with gr.Column():
-                    gr.Markdown("## Current State")
-                    state_view = gr.Markdown()
-                    browser_view = gr.Image(label="Browser View")
-
-                    gr.Markdown("### Task Checklist from WebShephered")
-                    checklist_view = gr.Markdown()
-                    
-                    gr.Markdown("### Action Selection in current step")
-                    with gr.Row() as rm_row:
-                        rm_cards_container = gr.HTML()
-            with gr.Column(scale=2):
-                gr.Markdown("## Trajectory")
-                trajectory_container = gr.HTML() # Placeholder for our custom trajectory component
-
-        
-            
-        run_btn.click(
-            fn=process_run,
-            inputs=[instruction, model_name, start_url],
-            outputs=[state_view, browser_view, checklist_view, rm_cards_container, trajectory_container],
-            api_name="run_agent",
-            concurrency_limit=32,
-            show_progress=True
-        )
-
-    logger.info("Launching Gradio interface")
-    # Set max_threads to allow multiple concurrent requests
-    demo.launch(share=True, max_threads=32)
-
-if __name__ == "__main__":
-    # Add support for multiprocessing on Windows
-    multiprocessing.freeze_support()
-    main()
+# 🔁 Now run app.py in a clean subprocess with correct PYTHONPATH
+subprocess.run([sys.executable, "app.py"], env=env)