# The imports
from dotenv import load_dotenv
from agents import Agent, Runner, trace

# The usual starting point
load_dotenv(override=True)

@function_tool
def get_current_weather(city: str) -> str:
    """returns weather info for the specified city."""
    return f"The weather in {city} is sunny"

tools = [get_current_weather]

# Make an agent with name, instructions, model
agent = Agent(
  name="Jokestar", 
  instructions="You are a jokestar, you tell jokes", 
  model="gpt-5-nano"
  tools=tools
  )

# Run the joke with Runner.run(agent, prompt) then print final_output
with trace("Telling a joke"):
    result = await Runner.run(agent, "Tell a joke about Autonomous AI Agents")
    print(result.final_output)

from agents import Agent

booking_agent = Agent(...)
refund_agent = Agent(...)

booking_agent_tool = booking_agent.as_tool(
            tool_name="booking_expert",
            tool_description="Handles booking questions and requests.",
        )
refund_agent_tool = refund_agent.as_tool(
            tool_name="refund_expert",
            tool_description="Handles refund questions and requests.",
        )

customer_facing_agent = Agent(
    name="Customer-facing agent",
    instructions=(
        "Handle all direct user communication. "
        "Call the relevant tools when specialized expertise is needed."
    ),
    tools=[booking_agent_tool, refund_agent_tool],
)

from agents import Agent

booking_agent = Agent(...)
refund_agent = Agent(...)

triage_agent = Agent(
    name="Triage agent",
    instructions=(
        "Help the user with their questions. "
        "If they ask about booking, hand off to the booking agent. "
        "If they ask about refunds, hand off to the refund agent."
    ),
    handoffs=[booking_agent, refund_agent],
)

from dotenv import load_dotenv
from agents import Agent, Runner, trace, function_tool
from openai.types.responses import ResponseTextDeltaEvent
from typing import Dict
import sendgrid
import os
import asyncio
import requests
import json  # 추가!

load_dotenv(override=True)

@function_tool
def get_current_weather(location: str, unit: str = "celsius") -> str:
    """
    OpenWeatherMap API를 사용하여 특정 위치의 현재 날씨 정보를 가져옵니다.
    
    Args:
        location: 도시 이름 (예: Seoul, Tokyo)
        unit: 온도 단위 (celsius 또는 fahrenheit)
    
    Returns:
        JSON 형식의 날씨 정보 문자열
    """
    # OpenWeatherMap API 키 (환경변수에서 가져오기)
    api_key = os.getenv("OPENWEATHER_API_KEY")
    if not api_key:
        return json.dumps({"error": "OPENWEATHER_API_KEY not found"})
    
    # unit 값 정규화 (metric, celsius 모두 celsius로 처리)
    if unit.lower() in ["metric", "celsius"]:
        unit = "celsius"
    elif unit.lower() in ["imperial", "fahrenheit"]:
        unit = "fahrenheit"
    
    # 1. Geocoding API로 도시 이름을 위도/경도로 변환
    geo_url = f"http://api.openweathermap.org/geo/1.0/direct?q={location}&limit=1&appid={api_key}"
    
    try:
        geo_response = requests.get(geo_url, timeout=5)
        geo_response.raise_for_status()
        geo_data = geo_response.json()
        
        if not geo_data:
            return json.dumps({"error": f"도시를 찾을 수 없습니다: {location}"})
        
        lat = geo_data[0]['lat']
        lon = geo_data[0]['lon']
        
        # 2. Current Weather API 호출
        weather_url = f"https://api.openweathermap.org/data/2.5/weather?lat={lat}&lon={lon}&appid={api_key}&units=metric"
        weather_response = requests.get(weather_url, timeout=5)
        
        weather_response.raise_for_status()
        data = weather_response.json()
        
        # 온도 변환
        temp_c = data['main']['temp']
        temp_f = (temp_c * 9/5) + 32
        temperature = temp_c if unit == "celsius" else temp_f
        
        weather_data = {
            "location": data['name'],
            "temperature": round(temperature, 1),
            "unit": unit,
            "condition": data['weather'][0]['description'],
            "humidity": data['main']['humidity'],
            "wind_speed": data['wind']['speed']
        }
        return json.dumps(weather_data)
        
    except requests.exceptions.RequestException as e:
        return json.dumps({"error": f"API 호출 실패: {str(e)}"})

@function_tool
def send_email(body: str):
    # Mailgun API 설정
    domain = "bys.digital"
    api_key = os.getenv('MAILGUN_API_KEY')
    if not api_key:
        print("❌ MAILGUN_API_KEY 환경 변수가 설정되지 않았습니다!")
        return None
    url = f"https://api.mailgun.net/v3/{domain}/messages"
    data = {
        "from": f"Mailgun Sandbox <postmaster@{domain}>",
        "to": "skwltg90@naver.com",  # 이메일만 (이름 제거)
        "subject": "Weather Cast Email",
        "text": body
    }
    
    try:
        response = requests.post(
            url,
            auth=("api", api_key),
            data=data,
            timeout=10
        )
        
        print(f"Status Code: {response.status_code}")
        print(f"Response: {response.text}")
        
        if response.status_code == 200:
            print("✅ 이메일 전송 성공!")
        else:
            print(f"❌ 이메일 전송 실패: {response.status_code}")
            print(f"에러 메시지: {response.text}")
        
        return response
        
    except Exception as e:
        print(f"❌ 에러 발생: {str(e)}")
        return None

basic_tools = [get_current_weather]


# Professional and Detailed Style
instructions1 = """
You are a professional meteorologist. Create weather forecast emails following these guidelines.

**Style:**
- Professional and formal tone
- Appropriate use of meteorological terminology
- Provide detailed data-driven analysis
"""

instructions2 = """
You are a friendly neighborhood weather caster. Create weather forecast emails following these guidelines.

**Style:**
- Casual tone as if talking to a friend
- Active use of emojis (☀️🌧️❄️⛅🌈 etc.)
- Explain in everyday language
- Humorous and entertaining
"""

instructions3 = """
You are an emotional storyteller weather caster. Create weather forecast emails following these guidelines.

**Style:**
- Literary and lyrical expressions
- Present weather as a narrative story
- Emotional and poetic descriptions
- Emphasize seasonal feelings and natural changes
"""


weather_cast_agent1 = Agent(
        name="Professional Weather Caster Agent",
        instructions=instructions1,
        model="gpt-4o-mini",
        tools=basic_tools
)

weather_cast_agent2 = Agent(
        name="Friendly Weather Caster Agent",
        instructions=instructions2,
        model="gpt-4o-mini",
        tools=basic_tools
)

weather_cast_agent3 = Agent(
        name="Storyteller Weather Caster Agent",
        instructions=instructions3,
        model="gpt-4o-mini",
        tools=basic_tools
)

senior_weather_cast_agent = Agent(
    name="Senior Weather Caster Agent",
    instructions="You pick the best weather cast email from the given options. \
Imagine you are a ordinary citizens living in Korea and pick the one you are most likely to respond to. \
Do not give an explanation; reply with the selected email only.",
    model="gpt-4o-mini"
)


async def main():
    message = "Write a weather cast email for 'Seoul' using Korean. When you need to know use 'get_current_weather' tool. Do not call 'get_current_weather' tool several times."
    
    print("=== 날씨 이메일 생성 시작 ===\n")
    try:
                    
        with trace("Parallel weather cast emails"):
            results = await asyncio.gather(
                Runner.run(weather_cast_agent1, message),
                Runner.run(weather_cast_agent2, message),
                Runner.run(weather_cast_agent3, message),
                return_exceptions=True  # 에러가 나도 다른 에이전트는 계속 실행
            )
            outputs = [result.final_output for result in results]

            emails = "Weather cast emails:\n\n" + "\n\nEmail:\n\n".join(outputs)
            best = await Runner.run(senior_weather_cast_agent, emails)

            print(f"Best weather cast email:\n{best.final_output}")


        print("=== 결과 ===\n")
        
        for i, result in enumerate(results, 1):
            if isinstance(result, Exception):
                print(f"Agent {i} 에러: {result}\n")
            else:
                print(f"Agent {i} 결과:")
                print(result.final_output)
                print("\n" + "="*50 + "\n")
                
    except Exception as e:
        print(f"전체 실행 에러: {e}")

# 실행
if __name__ == "__main__":
    asyncio.run(main())

description = "Write a weather cast email for 'Seoul' using Korean. When you need to know weather, use 'get_current_weather' tool. Do not call 'get_current_weather' tool several times."
weather_tool1 = weather_cast_agent1.as_tool(tool_name="weather_cast_agent1", tool_description=description)
weather_tool2 = weather_cast_agent2.as_tool(tool_name="weather_cast_agent2", tool_description=description)
weather_tool3 = weather_cast_agent3.as_tool(tool_name="weather_cast_agent3", tool_description=description)

tools = [weather_tool1, weather_tool2, weather_tool3, send_email]


instructions = """
You are a senior weather caster. Your goal is to find the single best weather cast email using the weather_cast_agent tools.
You pick the best weather cast email from the given options. \
Imagine you are a ordinary citizens living in Korea and pick the one you are most likely to respond to. \
Do not give an explanation; reply with the selected email only.

Follow these steps carefully:
1. Generate Drafts: Use all three weather_cast_agent tools to generate three different email drafts. Do not proceed until all three drafts are ready.
 
2. Evaluate and Select: Review the drafts and choose the single best email using your judgment of which one is most effective.
 
3. Use the send_email tool to send the best email (and only the best email) to the user.
 
Crucial Rules:
- You must use the sales weather_cast_agent tools to generate the drafts — do not write them yourself.
- You must send ONE email using the send_email tool — never more than one.
"""

senior_weather_cast_manager = Agent(name="Senior Weather Cast Manager", instructions=instructions, tools=tools, model="gpt-4o-mini")

async def main():
    
    print("=== 날씨 이메일 생성 시작 ===\n")
    try:        
        message = "Send a weather cast email addressed to 'Dear Seoul Citizen'"
        
        with trace("Senior Weather Cast Manager"):
            result = await Runner.run(senior_weather_cast_manager, message)
        
        print(result)
                
    except Exception as e:
        print(f"전체 실행 에러: {e}")

# 실행
if __name__ == "__main__":
    asyncio.run(main())

# Professional and Detailed Style
instructions1 = """
You are a professional meteorologist. Create weather forecast emails following these guidelines.

**CRITICAL: You MUST use the get_current_weather tool to fetch real-time weather data for Seoul before writing the email. You must write in Korean**

**Style:**
- Professional and formal tone
- Appropriate use of meteorological terminology
- Provide detailed data-driven analysis
"""

instructions2 = """
You are a friendly neighborhood weather caster. Create weather forecast emails following these guidelines.

**CRITICAL: You MUST use the get_current_weather tool to fetch real-time weather data for Seoul before writing the email. You must write in Korean**

**Style:**
- Casual tone as if talking to a friend
- Active use of emojis (☀️🌧️❄️⛅🌈 etc.)
- Explain in everyday language
- Humorous and entertaining
"""

instructions3 = """
You are an emotional storyteller weather caster. Create weather forecast emails following these guidelines.

**CRITICAL: You MUST use the get_current_weather tool to fetch real-time weather data for Seoul before writing the email. You must write in Korean**

**Style:**
- Literary and lyrical expressions
- Present weather as a narrative story
- Emotional and poetic descriptions
- Emphasize seasonal feelings and natural changes
"""


weather_cast_agent1 = Agent(
        name="Professional Weather Caster Agent",
        instructions=instructions1,
        model="gpt-4o-mini",
        tools=basic_tools
)

weather_cast_agent2 = Agent(
        name="Friendly Weather Caster Agent",
        instructions=instructions2,
        model="gpt-4o-mini",
        tools=basic_tools
)

weather_cast_agent3 = Agent(
        name="Storyteller Weather Caster Agent",
        instructions=instructions3,
        model="gpt-4o-mini",
        tools=basic_tools
)

description = "Write current weawther of city e-mail"
# description = "Write a weather cast email for 'Seoul'. You must use 'get_current_weather' tool to know current weather."
weather_tool1 = weather_cast_agent1.as_tool(tool_name="weather_cast_agent1", tool_description=description)
weather_tool2 = weather_cast_agent2.as_tool(tool_name="weather_cast_agent2", tool_description=description)
weather_tool3 = weather_cast_agent3.as_tool(tool_name="weather_cast_agent3", tool_description=description)

tools = [weather_tool1, weather_tool2, weather_tool3, send_email]


instructions = """
You are a senior weather caster manager. Your goal is to find the single best weather cast email using the weather_cast_agent tools. You pick the best weather cast email from the given options. Imagine you are a ordinary citizens living in Korea and pick the one you are most likely to respond to. Do not give an explanation; reply with the selected email only.

**CRITICAL: You must use the sales weather_cast_agent tools to generate the drafts — do not write them yourself - do not call weather_cast_agent more than one. You must send ONE most effective email using the send_email tool — never more than one.**

Follow these steps IN ORDER:
1. Generate Drafts: Use all three weather_cast_agent tools to generate three different email drafts. Do not proceed until all three drafts are ready.
2. Evaluate and Select: Review the drafts and choose the single best email using your judgment of which one is most effective.
3. Use the send_email tool to send the best email (and only one the best email) to the user.
"""

senior_weather_cast_manager = Agent(name="Senior Weather Cast Manager", instructions=instructions, model="gpt-4o-mini", tools=tools)


async def main():
    
    
    print("=== 날씨 이메일 생성 시작 ===\n")
    try:        
        message = "Send a weather cast email addressed to 'Dear Seoul Citizen'"
        with trace("Senior Weather Cast Manager"):
            result = await Runner.run(senior_weather_cast_manager, message)
        
        print(result)
                
    except Exception as e:
        print(f"전체 실행 에러: {e}")

# 실행
if __name__ == "__main__":
    asyncio.run(main())

sender_instruction="You are an e-mail sender. You can use send_email tool to send an e-mail."
send_email_agent = Agent(
    name="Email Manager",
    instructions=sender_instruction,
    model="gpt-5-mini",
    tools=[send_email],
    handoff_description="Send an e-mail")


tools = [weather_tool1, weather_tool2, weather_tool3]
handoffs = [send_email_agent]

manager_instruction = """
You are a senior weather cast manager. Your goal is to find the single best weather cast email using weather_cast_agent tools and send e-mail. 

Follow these steps carefully:
1. Generate Drafts: Use all three weather_cast_agent tools to generate three different email drafts. Do not proceed until all three drafts are ready.
 
2. Evaluate and Select: Review the drafts and choose the single best email using your judgment of which one is most effective.
You can use the tools one time if you're not satisfied with the results, just choose second one.
 
3. Handoff for Sending: Pass ONLY the winning email draft to the 'Email Manager' agent. The Email Manager will take care of sending.
 
Crucial Rules:
- You must use the weather_cast_agent tools to generate the drafts — do not write them yourself.
- You must hand off exactly ONE email to the Email Manager — never more than one.
"""

senior_weather_cast_manager = Agent(
    name="Senior Weather Cast Manager", 
    instructions=manager_instruction, 
    model="gpt-4o-mini", 
    tools=tools, 
    handoffs = handoffs)

translator_agent.handoffs = [senior_weather_cast_manager]
send_email_agent.handoffs = [senior_weather_cast_manager]

weather_cast_agent1 = Agent(
        name="Professional Weather Caster Agent",
        instructions=instructions1,
        model="gpt-4o-mini",
        tools=basic_tools
)

weather_cast_agent2 = Agent(
        name="Friendly Weather Caster Agent",
        instructions=instructions2,
        model="gpt-4o-mini",
        tools=basic_tools
)

weather_cast_agent3 = Agent(
        name="Storyteller Weather Caster Agent",
        instructions=instructions3,
        model="gpt-4o-mini",
        tools=basic_tools
)

description = "Write current weawther of city e-mail"
# description = "Write a weather cast email for 'Seoul'. You must use 'get_current_weather' tool to know current weather."
weather_tool1 = weather_cast_agent1.as_tool(tool_name="weather_cast_agent1", tool_description=description)
weather_tool2 = weather_cast_agent2.as_tool(tool_name="weather_cast_agent2", tool_description=description)
weather_tool3 = weather_cast_agent3.as_tool(tool_name="weather_cast_agent3", tool_description=description)

translator_instruction = """
You are a professional translator. Translate the given text to Korean while maintaining the original style and formatting.
Return ONLY the translated text, no explanations. 
**CRITICAL: After you translate an e-mail, you must hand off the manager(senior_weather_cast_manager) to report result and to continue next step.** - do NOT stop here.**
"""
translator_agent = Agent(
    name="Translator Agent",
    instructions=translator_instruction,
    model="gpt-4o-mini",
    handoff_description="Translate to Korean"
    )


sender_instruction="You are an e-mail sender. You can use send_email tool to send an e-mail."
send_email_agent = Agent(
    name="Send Email Agent",
    instructions=sender_instruction,
    model="gpt-4o-mini",
    tools=[send_email],
    handoff_description="Send an e-mail")


tools = [weather_tool1, weather_tool2, weather_tool3]
handoffs = [translator_agent, send_email_agent]

manager_instruction = """
You are a senior weather caster manager. Your goal is to find the single best weather cast email using weather_cast_agent tools. You pick the best weather cast email from the given options. Imagine you are a ordinary citizens living in Korea and pick the one you are most likely to respond to. Do not give an explanation; reply with the selected email only.

**Must follow the ORDER: 
1. You must use the sales weather_cast_agent tools to generate the drafts — do not write them yourself - do not call each weather_cast_agent tool more than one. 
2. Evaluate and Select: Review the drafts and choose the 'ONLY ONE' single best email using your judgment of which one is most effective.
3. Handoff for translate: Pass the one(single best email choosed in step 2) e-mail draft to the 'Translator Agent'. Translator Agent will take care translating **Wait for translated result**
4. Review: You must check the translated e-mail whether it is Korean or not.
5. Handoff for sending e-mail: Pass translated e-mail to the 'Send Email Agent'. 'Send Email Agent' will send e-mail.
**
"""

senior_weather_cast_manager = Agent(name="Senior Weather Cast Manager", instructions=manager_instruction, model="gpt-4o-mini", tools=tools, handoffs = handoffs)

translator_agent.handoffs = [senior_weather_cast_manager]
send_email_agent.handoffs = [senior_weather_cast_manager]

async def main():
    
    print("=== 날씨 이메일 생성 시작 ===\n")
    try:        
        message = "Send a weather cast email addressed to 'Dear Seoul Citizen'"
        with trace("Senior Weather Cast Manager"):
            result = await Runner.run(senior_weather_cast_manager, message)
        
        print(result)
                
    except Exception as e:
        print(f"전체 실행 에러: {e}")

# 실행
if __name__ == "__main__":
    asyncio.run(main())

description = "Write current weawther of city e-mail"
# description = "Write a weather cast email for 'Seoul'. You must use 'get_current_weather' tool to know current weather."
weather_tool1 = weather_cast_agent1.as_tool(tool_name="weather_cast_agent1", tool_description=description)
weather_tool2 = weather_cast_agent2.as_tool(tool_name="weather_cast_agent2", tool_description=description)
weather_tool3 = weather_cast_agent3.as_tool(tool_name="weather_cast_agent3", tool_description=description)

translator_instruction = """
You are a professional translator. Translate the given text to Korean while maintaining the original style and formatting.
Return ONLY the translated text, no explanations. 
**CRITICAL: After you translate an e-mail, you must hand off the Send Email Agent to send an e-mail.** - do NOT stop here.**
"""
translator_agent = Agent(
    name="Translator Agent",
    instructions=translator_instruction,
    model="gpt-4o-mini",
    handoff_description="Translate to Korean"
    )

sender_instruction="You are an e-mail sender. You can use send_email tool to send an e-mail."
send_email_agent = Agent(
    name="Send Email Agent",
    instructions=sender_instruction,
    model="gpt-4o-mini",
    tools=[send_email],
    handoff_description="Send an e-mail")


tools = [weather_tool1, weather_tool2, weather_tool3]
handoffs = [translator_agent]
translator_agent.handoffs = [send_email_agent]

manager_instruction = """
You are a senior weather caster manager. Your goal is to find the single best weather cast email using weather_cast_agent tools. You pick the best weather cast email from the given options. Imagine you are a ordinary citizens living in Korea and pick the one you are most likely to respond to. Do not give an explanation; reply with the selected email only.

**Must follow the ORDER: 
1. You must use the sales weather_cast_agent tools to generate the drafts — do not write them yourself - do not call each weather_cast_agent tool more than one. 
2. Evaluate and Select: Review the drafts and choose the 'ONLY ONE' single best email using your judgment of which one is most effective.
3. Handoff for translate: Pass the one(single best email choosed in step 2) e-mail draft to the 'Translator Agent'. Translator Agent will take care translating **Wait for translated result**
**
"""

senior_weather_cast_manager = Agent(name="Senior Weather Cast Manager", instructions=manager_instruction, model="gpt-4o-mini", tools=tools, handoffs = handoffs)

......