import OpenAI from "https://deno.land/x/openai@v4.52.0/mod.ts";
import { OPENAI_API_KEY } from "@lib/env.ts";
import { cacheFunction } from "@lib/cache/cache.ts";
import { hashString } from "@lib/helpers.ts";

const openAI = OPENAI_API_KEY && new OpenAI(OPENAI_API_KEY);

function extractListFromResponse(response?: string): string[] {
  if (!response) return [];
  return response
    .split(/[\n,]/)
    .map((line) => line.trim())
    .filter((line) => !line.endsWith(":"))
    .map((line) => line.replace(/^[^\s]*/, "").trim())
    .filter((line) => line.length > 2);
}

export async function summarize(content: string) {
  if (!openAI) return;
  const chatCompletion = await openAI.chat.completions.create({
    model: "gpt-3.5-turbo",
    messages: [
      {
        role: "user",
        content:
          `Please summarize the article in one sentence as short as possible: ${
            content.slice(0, 2000)
          }`,
      },
    ],
  });

  return chatCompletion.choices[0].message.content;
}
export async function shortenTitle(content: string) {
  if (!openAI) return;
  const chatCompletion = await openAI.chat.completions.create({
    model: "gpt-3.5-turbo",
    messages: [
      {
        role: "system",
        content:
          "You shorten the title of the provided website as much as possible, don't rewrite it, just remove unneccesary information. Please remove for example any mention of the name of the website.",
      },
      {
        role: "user",
        content: content.slice(0, 2000),
      },
    ],
  });

  return chatCompletion.choices[0].message.content;
}

export async function extractAuthorName(content: string) {
  if (!openAI) return;
  const chatCompletion = await openAI.chat.completions.create({
    model: "gpt-3.5-turbo",
    messages: [
      {
        role: "system",
        content:
          "You extract the name of the author from the given text. If you are not able to extract the name, please respond with 'not found' and do not include any other output.",
      },
      {
        role: "user",
        content: content.slice(0, 2000),
      },
    ],
  });

  const author = chatCompletion.choices[0].message.content;

  if (
    author?.toLowerCase().includes("not") &&
    author?.toLowerCase().includes("found")
  ) return "";

  return author;
}

export async function createGenres(
  type: string,
  description: string,
  title = "unknown",
) {
  if (!openAI) return;
  const chatCompletion = await openAI.chat.completions.create({
    model: "gpt-3.5-turbo",
    messages: [
      {
        role: "system",
        content:
          `you create some keywords that can be used in a recommendation system. The keywords are based on a ${type} description or title. If you do not know the title, take into account the description aswell. Create a range of keywords from very specific ones that describe the general vibe. ${
            title ? `The name of the ${type} is ${title}` : ""
          }. Return a list of around 20 keywords seperated by commas`,
      },
      {
        role: "user",
        content: description.slice(0, 2000),
      },
    ],
  });

  const res = chatCompletion.choices[0].message.content?.toLowerCase();

  return extractListFromResponse(res)
    .map((v) => v.replaceAll(" ", "-"));
}

export async function createKeywords(
  type: string,
  description: string,
  title = "unknown",
) {
  if (!openAI) return;
  const chatCompletion = await openAI.chat.completions.create({
    model: "gpt-3.5-turbo",
    messages: [
      {
        "role": "system",
        "content":
          `you create some keywords that can be used in a recommendation system. The keywords are based on a ${type} description or title. If you do not know the title, take into account the description aswell. Create a range of keywords from very specific ones that describe the general vibe.
title: ${title}
return a list of around 20 keywords seperated by commas
`,
      },
      {
        "role": "user",
        "content": description.slice(0, 2000).replaceAll("\n", " "),
      },
    ],
  });

  const res = chatCompletion.choices[0].message.content?.toLowerCase();

  return extractListFromResponse(res)
    .map((v) => v.replaceAll(" ", "-"));
}

export const getMovieRecommendations = (keywords: string, exclude: string[]) =>
  cacheFunction({
    fn: async () => {
      if (!openAI) return;
      const chatCompletion = await openAI.chat.completions.create({
        model: "gpt-3.5-turbo",
        messages: [
          {
            role: "user",
            content:
              `Could you recommend me 10 movies based on the following attributes:

${keywords} 

The movies should be similar to but not include ${
                exclude.join(", ")
              } or remakes of that.

respond with a plain unordered list each item starting with the year the movie was released and then the title of the movie seperated by a -`,
          },
        ],
      });

      const res = chatCompletion.choices[0].message.content?.toLowerCase();

      if (!res) return;

      console.log("REsult:");
      console.log(res);

      const list = extractListFromResponse(res);

      console.log({ list });

      return res.split("\n").map((entry) => {
        const [year, ...title] = entry.split("-");

        return {
          year: parseInt(year.trim()),
          title: title.join(" ").replaceAll('"', "").trim(),
        };
      }).filter((y) => !Number.isNaN(y.year));
    },
    id: `openai:movierecs:${hashString(`${keywords}:${exclude.join()}`)}`,
  });

export async function createTags(content: string) {
  if (!openAI) return;
  const chatCompletion = await openAI.chat.completions.create({
    model: "gpt-3.5-turbo",
    messages: [
      {
        role: "system",
        content:
          "Please respond with a list of genres the corresponding article falls into, dont include any other informations, just a comma seperated list of top 5 categories. Please respond only with tags that make sense even if there are less than five.",
      },
      { role: "user", content: content.slice(0, 2000) },
    ],
  });

  const res = chatCompletion.choices[0].message.content?.toLowerCase();

  return extractListFromResponse(res).map((v) => v.replaceAll(" ", "-"));
}
