import { VoltAgent, Agent, createTool } from "@voltagent/core";
import { VercelAIProvider } from "@voltagent/vercel-ai";
import { openai } from "@ai-sdk/openai";
import { z } from "zod";

// Story Writer Agent
const writerAgent = new Agent({
  name: "writer",
  instructions: `You are a creative story writer. You write engaging short stories based on user requests.
  
  Keep stories between 2-3 paragraphs.
  Make them interesting and creative.
  Always include vivid descriptions and engaging characters.`,
  llm: new VercelAIProvider(),
  model: openai("gpt-4o-mini"),
  tools: [
    createTool({
      name: "save_story",
      description: "Save a completed story",
      parameters: z.object({
        title: z.string().describe("Title of the story"),
        content: z.string().describe("The story content"),
        genre: z.string().describe("Genre of the story")
      }),
      execute: async ({ title, content, genre }) => {
        console.log("Saving story:", { title, genre });
        return { 
          storyId: "STORY-" + Date.now(), 
          status: "saved",
          wordCount: content.split(' ').length
        };
      },
    })
  ]
});

// Translator Agent  
const translatorAgent = new Agent({
  name: "translator",
  instructions: `You are a professional translator. You translate text into multiple languages:
  - German (Deutsch)
  - Japanese (日本語) 
  - Italian (Italiano)
  
  Provide accurate, natural translations that maintain the story's tone and meaning.`,
  llm: new VercelAIProvider(),
  model: openai("gpt-4o-mini"),
  tools: [
    createTool({
      name: "save_translation",
      description: "Save a translation",
      parameters: z.object({
        originalText: z.string().describe("Original text"),
        translatedText: z.string().describe("Translated text"),
        language: z.string().describe("Target language")
      }),
      execute: async ({ originalText, translatedText, language }) => {
        console.log("Saving translation to", language);
        return { 
          translationId: "TRANS-" + Date.now(),
          language: language,
          status: "saved"
        };
      },
    })
  ]
});

// Creative Director - Coordinates the team
const creativeDirector = new Agent({
  name: "creative-director", 
  instructions: `You are a creative director managing a story writing and translation team.

When users request stories:
1. First use the WRITER agent to create the story
2. Then use the TRANSLATOR agent to translate it into German, Japanese, and Italian
3. Present the original story and all translations

ALWAYS delegate - don't write stories or translate yourself.
Your job is coordination only.

Example flow:
User: "Write a story about a magic cat"
→ Use writer agent to create story
→ Use translator agent for German version  
→ Use translator agent for Japanese version
→ Use translator agent for Italian version
→ Present all versions to user`,
  llm: new VercelAIProvider(),
  model: openai("gpt-4o-mini"),
  subAgents: [writerAgent, translatorAgent],
});

new VoltAgent({
  agents: {
    "creative-director": creativeDirector,
    writer: writerAgent,
    translator: translatorAgent,
  },
});

// Test the creative team
const testCreativeTeam = async () => {
  console.log("Testing creative writing team...");
  
  // Scenario 1: Simple story request
  const magicalStory = await creativeDirector.generateText(
    "Write a story about a magical cat",
    { userId: "writer123" }
  );
  
  // Scenario 2: Themed story
  const mysteryStory = await creativeDirector.generateText(
    "Create a mystery story set in Victorian London",
    { userId: "writer123" }
  );
  
  // Scenario 3: Adventure story
  const spaceAdventure = await creativeDirector.generateText(
    "Tell me an exciting space adventure story",
    { userId: "writer123" }
  );
  
  console.log("All creative scenarios completed!");
  console.log("Each story created in English + German + Japanese + Italian!");
};

// Test individual agents
const testIndividualAgents = async () => {
  // Test writer directly
  const story = await writerAgent.generateText(
    "Write a short story about friendship",
    { userId: "writer123" }
  );
  
  // Test translator directly  
  const translation = await translatorAgent.generateText(
    "Translate this to German: 'Once upon a time, there was a brave knight.'",
    { userId: "writer123" }
  );
};

// Run tests
testCreativeTeam();
testIndividualAgents();