import os
from dotenv import load_dotenv
from agents import AsyncOpenAI, OpenAIChatCompletionsModel, set_tracing_disabled

set_tracing_disabled(True)
load_dotenv()

openai_api_key = os.getenv("OPENAI_API_KEY")
gemini_api_key = os.getenv("GEMINI_API_KEY")  # Optional, ignore if not set

# No strict check—use OpenAI directly (Gemini fallback if you want later)
if not openai_api_key:
    raise ValueError(
        "OPENAI_API_KEY is not set. Please add it to your .env file: OPENAI_API_KEY=your_key_here"
    )

client_provider = AsyncOpenAI(
    api_key=openai_api_key,
    base_url="https://api.openai.com/v1/",
)

# If you want Gemini fallback (uncomment below, but CEO ke against hai abhi)
# if openai_api_key:
#     ... (OpenAI part)
# else:
#     if not gemini_api_key:
#         raise ValueError("No API key found!")
#     client_provider = AsyncOpenAI(
#         api_key=gemini_api_key,
#         base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
#     )

model = OpenAIChatCompletionsModel(
    model="gpt-4o",  # FIXED: Using valid OpenAI model (fastest GPT-4 variant)
    openai_client=client_provider
)

print("Setup complete! Model ready with OpenAI GPT-4o")  # Debug line