AI Tools

Gravity ships with support for generative AI image and text with ChatGPT and Dall•E.
1. Create an OpenAI API Key
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
2. Generating Text with ChatGPT
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.
3. Generating Images with Dall•E
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 }]
Included Endpoints
There are pre-configured endpoints included for you to play with:
POST /api/ai/text
POST /api/ai/image

Last modified 14d ago