feat: add loading of recommendations to movie page

lib/openai.ts

@@ -1,5 +1,7 @@
 import { OpenAI } from "https://deno.land/x/openai@1.4.2/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);
 
@@ -9,8 +11,8 @@ function extractListFromResponse(response?: string): string[] {
     .split(/[\n,]/)
     .map((line) => line.trim())
     .filter((line) => !line.endsWith(":"))
-    .map((line) => line.replace(/^[^(a-zA-Z)]*/, "").trim())
-    .filter((line) => line.length > 0);
+    .map((line) => line.replace(/^[^\s]*/, "").trim())
+    .filter((line) => line.length > 2);
 }
 
 export async function summarize(content: string) {
@@ -73,7 +75,7 @@ export async function extractAuthorName(content: string) {
   return author;
 }
 
-export async function createKeywords(
+export async function createGenres(
   type: string,
   description: string,
   title = "unknown",
@@ -85,7 +87,7 @@ export async function createKeywords(
       {
         "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. Also include some that describe the genre. ${
+          `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}` : ""
           }`,
       },
@@ -107,6 +109,82 @@ export async function createKeywords(
     .map((v) => v.replaceAll(" ", "-"));
 }
 
+export async function createKeywords(
+  type: string,
+  description: string,
+  title = "unknown",
+) {
+  if (!openAI) return;
+  const chatCompletion = await openAI.createChatCompletion({
+    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}
+description: ${description.slice(0, 2000).replaceAll("\n", " ")}}
+`,
+      },
+      {
+        "role": "user",
+        "content": "return a list of around 20 keywords seperated by commas",
+      },
+    ],
+  });
+
+  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.createChatCompletion({
+        model: "gpt-3.5-turbo",
+        messages: [
+          {
+            "role": "system",
+            "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.createChatCompletion({
@@ -126,4 +204,3 @@ export async function createTags(content: string) {
 
   return extractListFromResponse(res).map((v) => v.replaceAll(" ", "-"));
 }
-