import os

import arcade_math
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel

from arcade.actor.fastapi.actor import FastAPIActor
from arcade.client import AsyncArcade
from arcade.core.config import config
from arcade.core.toolkit import Toolkit

if not config.api or not config.api.key:
    raise ValueError("Arcade API key not set. Please run `arcade login`.")

client = AsyncArcade(api_key=config.api.key)

app = FastAPI()

actor_secret = os.environ.get("ARCADE_ACTOR_SECRET")
actor = FastAPIActor(app, secret=actor_secret)
actor.register_toolkit(Toolkit.from_module(arcade_math))


class ChatRequest(BaseModel):
    message: str


@app.post("/chat")
async def postChat(request: ChatRequest, tool_choice: str = "execute"):
    try:
        raw_response = await client.chat.completions.create(
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": request.message},
            ],
            model="gpt-4o-mini",
            max_tokens=500,
            tools=[
                # "Google.GetEmails",
                # "Google.SearchEmailsByHeader",
                # "Google.WriteDraft",
                # "GitHub.CountStargazers",
                # "GitHub.SetStarred",
                # "GitHub.SearchIssues",
                # "Slack.SendDmToUser",
                # "Slack.SendMessageToChannel",
            ],
            tool_choice=tool_choice,
            user=config.user.email if config.user else None,
        )

    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))
    else:
        return raw_response.choices