import OpenAI from "openai";
import { ObjectiveAI } from "objectiveai";

export type Category =
  | "Job Application"
  | "Inquiry"
  | "Meeting Request"
  | "Invoice"
  | "Requires Human Review";

export async function categorizeEmail(emailContent: string): Promise<Category> {
  const openai = new OpenAI({
    apiKey: "<YOUR_OBJECTIVE_AI_API_KEY>",
    baseURL: "https://api.objective-ai.io",
  });
  const request: ObjectiveAI.Chat.Completions.ChatCompletionCreateParams = {
    messages: [
      {
        role: "user",
        content: `Select the correct category for the following email:\n\n${emailContent}`,
      },
    ],
    model: "objectiveai/7GgWgWm7LLuk1LBd5UmYsh",
    response_format: {
      type: "json_schema",
      json_schema: {
        name: "email_categorization",
        description: "Categorize the provided email into a predefined category.",
        strict: true,
        schema: {
          type: "object",
          properties: {
            category: {
              type: "string",
              enum: [
                "Job Application",
                "Inquiry",
                "Meeting Request",
                "Invoice"
              ],
            },
          },
          required: ["category"],
        },
      },
    },
  };
  const completion = await ObjectiveAI.Chat.Completions.create(openai, request);
  let winner_confidence = 0;
  let winner_category = "";
  for (const choice of completion.choices) {
    // ignore choices generated by `select` models
    if (choice.generate_id === null) continue;
    // ignore choices that did not complete successfully
    if (choice.finish_reason !== "stop" || choice.confidence === null) continue;
    // get the winning choice
    if (choice.confidence > winner_confidence) {
      const parsedContent = JSON.parse(choice.content);
      winner_confidence = choice.confidence;
      winner_category = parsedContent.category;
    }
  }
  if (winner_confidence < 0.8 || winner_category === "") {
    return "Requires Human Review";
  } else {
    return winner_category as Category;
  }
}