import asyncio
import os
from typing import List, Dict, Any, Optional

# Assuming imports succeed
from agents import Agent, Runner, trace, gen_trace_id, ModelSettings
from agents.mcp import MCPServerSse

PLAYWRIGHT_MCP_URL = "http://localhost:8931/sse"

async def run_simple_web_task():
    print(f"INFO: Attempting to connect to Playwright MCP Server at {PLAYWRIGHT_MCP_URL}")

    async with MCPServerSse(
        name="PlaywrightMCPExampleServer",
        params={"url": PLAYWRIGHT_MCP_URL}
    ) as server:
        print("INFO: MCP Server connection established.")

        browser_agent = Agent(
            name="BrowserToolAgent",
            instructions=(
                "You are a browser interaction specialist. Execute the specific browser commands "
                "given to you (e.g., navigate, get_text, click). Use only the provided tools. "
                "Be precise with URLs and selectors."
                "When asked to play tictactoe, you are expected to make the move."
                "When you finish, tell the user the result of the task, what you did, and what the final is. Especially detail the key information on the page."
            ),
            mcp_servers=[server],
            model_settings=ModelSettings(tool_choice="required"),
            handoff_description="Controls a web browser via Playwright MCP (navigate, get text, click)."
        )

        orchestrator_agent = Agent(
            name="WebTaskOrchestrator",
            instructions=(
                "Your goal is to accomplish the user's web-based task. Analyze the request. "
                "Do not attempt to perform the task directly. Instead, "
                "use the 'browser_tool' to perform those specific actions. "
                "Provide clear instructions to the tool, including URLs and what specific text/element is needed. "
                "Iterate until the task is complete. "

                "for example, if the task is to play a game, "
                "ask the browser tool to navigate to the game page, "
                "then ask the browser tool to click on the game board, "
                "ask the browser tool to give you the current game state so you can decide your next move, "
                "and finally, ask the browser tool to click on the game board again to make your move. "
                "if you are not sure what to do next, ask the browser tool to give you the current game state. "
                "Report the final result to the user after the task is complete."
            ),
            tools=[
                browser_agent.as_tool(
                    tool_name="browser_tool",
                    tool_description="Use to interact with web pages: navigate to a URL, get text from elements."
                )
            ]
        )
        print("INFO: Orchestrator agent created.")

        task_message = "Go to https://www.wikidata.org/wiki/Q28865 and traverse the page to the page concept of Math(Q395). Do not use the search bar. Do not ask the browser agent to directly navigate to the Math page either. This is meant to be a challenge for you, collaborate with the browser agent to solve this problem iteratively."

        print(f"\nINFO: Executing task: '{task_message}'")
        trace_id = gen_trace_id()
        trace_url = f"https://platform.openai.com/traces/trace?trace_id={trace_id}"
        print(f"INFO: Trace URL: {trace_url}")

        with trace(f"Minimal Example - {trace_id}", trace_id=trace_id):
            #here I can adjust the number of max_turn for the orchestrator agent, but I don't see how to adjust the max_turn for the agents as tools.
            result = await Runner.run(
                starting_agent=orchestrator_agent,
                input=task_message
            )

        print("\n--- Orchestration Result ---")
        final_output = result.final_output if result.final_output else "No final output generated."
        print(f"Final Output: {final_output}")
        print("--------------------------")

if __name__ == "__main__":
    print("INFO: Starting Minimal Playwright MCP Agent Example (No Error Handling)...")
    # Set API keys if needed
    # import openai
    # openai.api_key = "YOUR_API_KEY"
    asyncio.run(run_simple_web_task())
    print("INFO: Script finished.")

npx @playwright/mcp@latest --port 8931

@function_tool
def browser_tool(input:str):
  """Use to interact with web pages: navigate to a URL, get text from elements."""
  browser_agent = Agent(...)
  result = await Runner.run(browser_agent, input, max_turns=50)
  return ItemHelpers.text_message_outputs(result.new_items)