"use client";

import { Agent, run, setDefaultOpenAIClient, setTracingDisabled } from '@openai/agents';
import OpenAI from 'openai';
import { useMemo, useEffect, useState } from 'react';

// Early call to disable tracing
try {
  setTracingDisabled(true);
  console.log("[Orchestrator] Tracing explicitly disabled for Agent SDK.");
} catch (e) {
  console.warn("[Orchestrator] Could not disable tracing for Agent SDK:", e);
}

export const useOrchestrator = () => {
  const [message, setMessage] = useState("");

  const openaiClient = useMemo(() => {
    const apiKey = process.env.NEXT_PUBLIC_OPENAI_API_KEY;
    if (!apiKey) {
      console.warn("API key not set");
      return null;
    }
    try {
      return new OpenAI({ apiKey, dangerouslyAllowBrowser: true });
    } catch (e) {
      console.error("Failed to create OpenAI client", e);
      return null;
    }
  }, []);

  // Set default client immediately if available
  if (openaiClient) {
    try {
      setDefaultOpenAIClient(openaiClient);
      console.log("[Orchestrator] Default OpenAI client attempted to be set synchronously.");
    } catch (e) {
      console.error("[Orchestrator] Failed to set default OpenAI client synchronously:", e);
    }
  } else {
    console.warn("[Orchestrator] OpenAI client is null, cannot set default client.");
  }

  const testAgent = useMemo(() => {
    if (!openaiClient) return null; // Ensure client is set before agent creation
    return new Agent({
      name: 'TestAgent',
      instructions: 'You are a test agent.',
      // model: 'gpt-3.5-turbo', // Or your desired model
    });
  }, [openaiClient]);

  const handleRunTest = async () => {
    if (!testAgent) {
      setMessage("Agent not initialized.");
      return;
    }
    console.log("Attempting to run TestAgent...");
    try {
      const result = await run(testAgent, "Hello, agent!");
      setMessage(JSON.stringify(result.finalOutput) || "No output");
      console.log("Agent run successful:", result.finalOutput);
    } catch (error) {
      const err = error as Error;
      console.error("[Orchestrator] Error running TestAgent:", err);
      console.error("[Orchestrator] Full error object:", err);
      console.error("[Orchestrator] Stack trace:", err.stack);
      setMessage(`Error: ${err.message} - Check console for stack trace.`);
    }
  };

  return { message, handleRunTest };
};