nhcarrigan/nomena
9
0
Fork 0
generated from nhcarrigan/template
Code Issues Pull Requests 7 Actions Releases Activity

revert: don't clarify the full body in the prompt, no safety filter

Browse Source
This commit is contained in:
Naomi Carrigan
2025-10-31 12:51:51 -07:00
parent 3a2cc3f992
commit 2c9eb1e360
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/classes/ai.ts
+7 -8
View File
@@ -5,7 +5,7 @@
*/ */
import { Anthropic } from "@anthropic-ai/sdk"; import { Anthropic } from "@anthropic-ai/sdk";
import { GoogleGenAI, PersonGeneration, SafetyFilterLevel } from "@google/genai"; import { GoogleGenAI, PersonGeneration } from "@google/genai";
import { AttachmentBuilder, type MessageCreateOptions } from "discord.js"; import { AttachmentBuilder, type MessageCreateOptions } from "discord.js";
/** /**
@@ -50,7 +50,7 @@ export class Ai {
join(", ")}`, join(", ")}`,
prompt, prompt,
); );
const image = await this.generateImage(`Your task is to generate a full body anime girl mascot for this project. This means the full character should be visible. The image should have a transparent background. NEVER include text in the image. The project description is: ${prompt}`); const image = await this.generateImage(`Your task is to generate a full body anime girl mascot for this project. The image should have a transparent background. NEVER include text in the image. The project description is: ${prompt}`);
if (image === null) { if (image === null) {
return { content: `Project Name: ${names}\nProject Description: ${prompt}\nSorry, I was unable to generate an image for you.` }; return { content: `Project Name: ${names}\nProject Description: ${prompt}\nSorry, I was unable to generate an image for you.` };
} }
@@ -85,12 +85,11 @@ export class Ai {
private async generateImage(prompt: string): Promise<Buffer | null> { private async generateImage(prompt: string): Promise<Buffer | null> {
const response = await this.gemini.models.generateImages({ const response = await this.gemini.models.generateImages({
config: { config: {
aspectRatio: "3:4", aspectRatio: "3:4",
imageSize: "2K", imageSize: "2K",
numberOfImages: 1, numberOfImages: 1,
outputMimeType: "image/png", outputMimeType: "image/png",
personGeneration: PersonGeneration.ALLOW_ADULT, personGeneration: PersonGeneration.ALLOW_ADULT,
safetyFilterLevel: SafetyFilterLevel.BLOCK_NONE,
}, },
model: "models/imagen-4.0-generate-001", model: "models/imagen-4.0-generate-001",
prompt: prompt, prompt: prompt,