Mirrors/webamp
1
0
Fork 0
mirror of https://github.com/captbaritone/webamp.git synced 2026-07-25 11:04:00 +00:00
Code Issues Projects Releases Wiki Activity

Hack on openAi tags

Browse source
This commit is contained in:
Jordan Eldredge 2025-01-14 22:50:51 -08:00
parent afff5d63d2
commit b01cdc5876
2 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

4
packages/skin-database/cli.ts
View file

@ -136,8 +136,10 @@ program
if (ai) {
const skin = await SkinModel.fromMd5Assert(ctx, md5);
const description = await generateDescription(skin);
console.log("Generated description for", skin.getFileName());
console.log("Generated description for", await skin.getFileName());
console.log("====================================");
console.log(description);
console.log("====================================");
}
if (del) {
await Skins.deleteSkin(md5);

19
packages/skin-database/services/openAi.ts
View file

@ -18,10 +18,12 @@ The description should be concise and descriptive, and should not include any
personal opinions or judgement. The description should be written in English and
should be no longer than 200 characters.`;
const prompt2 = `"Identify thematic and stylistic details in this screenshot of a media player.
Give a comma separated list of tags, and ignore the text."`;
const prompt2 = `You are a staff digital preservationist working at the Internet Archive.
You have been tasked with identifying a comma separated list of tags for a Winamp skin to
be used for indexing purposes. Ignore the names of the tracks in the playlist.`;
export async function generateDescription(skin: SkinModel): Promise<string> {
const url = skin.getScreenshotUrl();
const response = await client.chat.completions.create({
model: "gpt-4o-mini",
messages: [
@ -29,16 +31,13 @@ export async function generateDescription(skin: SkinModel): Promise<string> {
role: "user",
content: [
{ type: "text", text: prompt2 },
{
type: "image_url",
image_url: {
url: skin.getScreenshotUrl(),
},
},
{ type: "image_url", image_url: { url } },
],
},
],
});
console.log(response.choices[0].message.content);
return "Done";
if (typeof response.choices[0].message.content !== "string") {
throw new Error("OpenAI response is not a string");
}
return response.choices[0].message.content;
}