from autogen.agentchat import (
    Agent,
    GroupChat,
    GroupChatManager,
    AssistantAgent,
    UserProxyAgent,
)
import openai
from typing import Dict, List, Optional, Union
import os
from dotenv import load_dotenv
from HumanResources import HumanResources
from AgentSelector import AgentSelector
from AgentSpawner import AgentSpawner
from GroupChatSpawner import GroupChatSpawner

openai.api_key = os.getenv("OPENAI_API_KEY")

CHAT_INITIATOR = "finance"

hr = HumanResources()
all_available_agents = hr.select_agents()
print(all_available_agents)

task = "revenue in the east coast is falling and the competitor is doing great with their product."
selector = AgentSelector(task=task, available_agents=all_available_agents, n_agents=3)
selected_agents = selector.run_selection()
print("*" * 10, selected_agents)

selected_agents = ["sales", "marketing", "engineer"]
agent_spawner = AgentSpawner(selected_agents, CHAT_INITIATOR)
agents = agent_spawner.spawn()
print(agents)

messages = []
llm_config = {"config_list": [{"model": "gpt-4"}], "seed": 42, "request_timeout": 600}
groupchat = GroupChatSpawner(
    agents=agents, llm_config=llm_config, messages=messages, max_round=10
)

chat_initiator.initiate_chat(
    manager,
    message="I want to design an app to make me one million dollars in one month. Yes, your heard that right.",
)