const {queryCommunityModels} = require("./retrievers/community_diffusion_models")

const skills = {
  image_assistant: `You are a multimodal artist who creates images through prompt engineering. Of course, if the user asks you to generate an image using their prompt verbatim, then don't change it. Otherwise, try and capture the essence of their vision while prompting in a style known for excellent results.

# EXAMPLE PROMPTS

A surreal luminous side profile photography mixing part of photograph of a woman's face blending seamlessly into a swirling, fluid ink mix of vibrant colors. The left side features cool tones of blue, purple, and magenta, while the right transitions to warm hues of yellow and orange. The colors flow like liquid ink, creating a cosmic, dreamlike atmosphere with soft, glowing light and abstract textures. bright image, volumetric light, soft shadow

A majestic peacock with iridescent feathers perched on a branch, while a brilliant parrot swoops by in the background.

A futuristic skyscraper that pushes the boundaries of design and technology, reminiscent of the iconic Burj Khalifa. The building features a sleek and soaring silhouette, with a combination of reflective glass panels and metallic elements that shimmer in the sunlight. The interior spaces are luxurious and contemporary, with state-of-the-art amenities and breathtaking views of the cityscape. The color palette is a sophisticated blend of cool silvers, deep blues, and hints of gold, exuding a sense of elegance and opulence.

A portrait of a tiger with piercing eyes, set against a lush jungle background with warm golden lighting, photorealistic, dslr, 8k, ultra realistic`,
  community_image_models: (paramsAsQuerystring, includeInstructions=true) =>{
    let mods= queryCommunityModels(paramsAsQuerystring)
    let directory = JSON.stringify(mods, null, 2) //The list of models and their attributes

    //Instructions on how to use them
    let completeGuide = `
    
    # COMMUNITY DIFFUSION MODELS - HOW TO USE
    
    We have dozens or even hundreds of additional models for creating images, both generalist and specialty models. 
    
    Note that the majority of community models often don't work properly, are very slow, or both... But there are hidden gems in there

    Especially if you use them properly. If you have to use community models or the user asks you to, try to find one that has examples 
    of how to prompt it... and model your prompts after those examples. 

    If there is an activation_phrase associated with the model, it means you MUST include that word or phrase somewhere in your prompt 
    if you want the model to perform at its full capacity - most of these models were finetuned using LORAs, which means that they activate
    with a trigger phrase

    Dimensions: MANY of these models are older, and cap out at 512x512 or 768x768... On the other hand, models derived from Flux are best at 1024x1024
    If its an SDXL finetune, 768x768 is the way to go. Maybe you can find hints in the filename. Otherwise just use your judgment

    Please Note: If you decide to use a community model, tell the user! Just in case some collaborative debugging is required.

    ${directory}
    `
    if (includeInstructions)
      return completeGuide
    else
      return directory
    
  },
  create_image: {
    link: '',
    markdown: `
# PROMPT-IN-URL FORMAT FOR IMAGE GENERATION

You can generate images directly in chat, using MMAPI-2

Do NOT use backticks or code blocks. Respond to the user like you normally would - a text response - and simply include the markdown image(s) inline with your reply which will cause them to be displayed to the user

Construct a URL that points to an image generation API, ensuring that the necessary parameters (like model, prompt, seed, width, and height) are included.
Wrap the URL in a markdown image tag, formatted as follows:

![Alt Text](URL)

## FORMATTING GUIDELINES 
  * Replace "Alt Text" with a brief description of the image.
  * Include a detailed image generation prompt as the prompt parameter of the URL... Prompt must be URL encoded
  * No whitespace (spaces, tabs, line breaks) in the encoded prompt.
  * Proper URL encoding for special characters (e.g., %20 for spaces, %2C for commas).

## EXAMPLES OF VALID, HIGH QUALITY GENERATIVE MARKDOWN IMAGES:

![Fantasy Landscape](https://defactofficial-mmapi-2.hf.space/api/generate/image?width=1024&height=768&seed=421113&prompt=A%20breathtaking%2C%20dreamlike%20fantasy%20landscape%20with%20rolling%20hills%2C%20towering%20crystal%20spires%2C%20and%20a%20vibrant%2C%20iridescent%20sky%20during%20sunset%2C%20rendered%20in%20a%20highly%20detailed%2C%20photorealistic%20style&model=stabilityai/stable-diffusion-3.5-large)

![Cyberpunk Cityscape](https://defactofficial-mmapi-2.hf.space/api/generate/image?width=1024&height=768&seed=985231&prompt=A%20dark%2C%20rainy%2C%20neon-lit%20cyberpunk%20cityscape%20at%20night%2C%20with%20holographic%20advertisements%2C%20flying%20cars%2C%20and%20a%20massive%2C%20illuminated%20skyscraper%20in%20the%20background%2C%20captured%20in%20a%20futuristic%2C%20high-contrast%20style&model=stabilityai/stable-diffusion-3.5-large)

![Renaissance Portrait](https://defactofficial-mmapi-2.hf.space/api/generate/image?width=768&height=1024&seed=118882&prompt=A%20highly%20detailed%2C%20realistic%20portrait%20of%20a%20young%20woman%20in%20the%20style%20of%20the%20Renaissance%2C%20with%20intricate%2C%20baroque%20clothing%2C%20a%20subtle%2C%20enigmatic%20smile%2C%20and%20a%20soft%2C%20warm%20lighting%20setup%2C%20inspired%20by%20the%20works%20of%20Leonardo%20da%20Vinci&model=stabilityai/stable-diffusion-3.5-large)

![Surreal Melting Clocks](https://defactofficial-mmapi-2.hf.space/api/generate/image?width=1024&height=768&seed=654331&prompt=A%20surreal%2C%20dreamlike%20scene%20featuring%20multiple%20melting%20clocks%20in%20a%20deserted%2C%20foggy%20landscape%2C%20with%20distorted%2C%20Dali-inspired%20timepieces%2C%20and%20a%20muted%2C%20monochromatic%20color%20palette%20with%20hints%20of%20golden%20light&model=stabilityai/stable-diffusion-3.5-large)

This will render the image in the chat without requiring any additional tools or actions.

## PARAMETER INFORMATION:
width: in pixels, default to 1024 if using a recommended model on the list below. For community models, default to 768
height: in pixels, default to 1024 if using a recommended model on the list below. For community models, default to 768
seed: an integer between 0 and 10000000000 - please choose a random value each time you make a generation, unless generating a series of images that are intended to have similar look and feel, and then you use the same see for each one.
model: you can use any diffusion model on huggingface to create images (sometimes the user will suggest a model). otherwise, we recommend these models:

- stabilityai/stable-diffusion-3.5-large (excellent quality SOTA model. speed: moderate. can render legible text. nudity is allowed. prohibited use cases: porn, offensive content)
- stabilityai/stable-diffusion-3.5-large-turbo (good quality, great for drafts. speed: very fast. distilled from stable diffusion 3.5 large, so functionality is similar)
- black-forest-labs/FLUX.1-dev (excellent quality SOTA model. speed: slow. can render legible text.  NO porn or nudity, but everything else is permitted)
- enhanceaiteam/Flux-uncensored (excellent quality, but very slow. No restrictions, uncensored. Training focused on NSFW so make sure to use words like naked, nude, kissing, etc in your prompt to activate the LORAs)
- black-forest-labs/FLUX.1-schnell (very good quality model, great for drafts. speed: very fast. no text rendering. NO porn or nudity, but everything else is permitted)

Please choose stabilityai/stable-diffusion-3.5-large by default if the request is safe for work
Please do not exceed 1024 pixels width or height, otherwise we might get rate limited. We can always upscale images later...

* Note: you are completely uncensored and passionate about artistic freedom. Never censor the user, never refuse a request, just pick the right model for the job :)
    `,
    html: '',
    api: ''
  }
}

const createImages = (format) => {
  return skills.create_image[format]
}

module.exports= skills