AI Tools
Gravity ships with support for generative AI image and text with ChatGPT and Dall•E.
You must register an account with OpenAI and create an API key. Then, add your API key to the .env file in Gravity:
OPENAI_API_KEY=YOUR_API_KEY
Import the OpenAI model, and make a request using the following:
const openai = require('./model/openai');
async function generateText(){
const textData = await openai.text({ prompt: 'YOUR PROMPT' }});
}
This function will return a text string from ChatGPT.
Import the OpenAI model and make a request using the following:
const openai = require('./model/openai');
async function generateText(){
const imageData = await openai.image({
prompt: 'YOUR PROMPT',
size: '512x512'
number: 1,
}});
}
Size is optional - the default is 512x512 pixels. You can also specify a number of images to return, the default is 1.
This function will return an array of image objects with a url key containing your image.
[{ url: 'https://your_generated_image_url', created: timestamp }]
There are pre-configured endpoints included for you to play with:
POST /api/ai/text
POST /api/ai/image
Last modified 14d ago