import os
from fastapi import FastAPI
from pydantic import BaseModel
from browser_use import Agent, Browser, BrowserConfig
from langchain_openai import ChatOpenAI

app = FastAPI()

# Используем Nvidia или OpenRouter (через OpenAI адаптер)
llm = ChatOpenAI(
    base_url="https://integrate.api.nvidia.com/v1",
    api_key=os.getenv("NVIDIA_API_KEY"),
    model="meta/llama-3.1-405b-instruct"
)

class Task(BaseModel):
    instruction: str

@app.post("/run")
async def run_task(task: Task):
    # Настройка браузера для работы с виртуальным дисплеем
    browser = Browser(config=BrowserConfig(headless=False)) 
    agent = Agent(task=task.instruction, llm=llm, browser=browser)
    
    result = await agent.run()
    return {"result": result.final_result()}