Skip to content

VassoD/cohere-plugin-figma

Repository files navigation

Overview

This is an Figma AI plugin template that demonstrates streaming LLM responses inside of a Figma plugin. This template is customized for integrating with the Cohere API. This template shows:

  • Securely storing Cohere keys / prompts on a server
  • Streaming a response to an iframe
  • Streaming a response to a Figma document
  • A fully functional React iframe using tailwind and Next.js
  • Deploying your plugin to production
  • Accessing the Figma API directly from the iframe

Getting Started

This plugin is set up to use Next.js.

First, create this template using create-next-app:

npx create-next-app@latest --example https://github.com/figma/ai-plugin-template/

Next you need to store you Cohere API key in the .env.local file. Create a .env.local file in the root of this project and add your API key:

COHERE_API_KEY=***

Then, run the development server:

npm i
npm run dev

You can then open up the Figma desktop app and import a plugin from the manifest file in this project. You can right click on the canvas and navigate to Plugins > Development > Import plugin from manifest... and select the manifest.json in {path to this project}/plugin/manifest.json.

Image showing how to import from manifest

Editing this template

The main files you'll want to edit are:

  • app/page.tsx: will let you update the plugin iframe. The page auto-updates as you edit the file and will let you update the user interface of your plugin.
  • app/completion/route.ts: This is the "server" of the plugin and is what talks to Cohere. This is where you can update the prompt that you are sending to Cohere.
  • plugin/manifest.json: this is the manifest file that will let you update the permissions and editor types of your plugin.

Publishing your plugin

In this example we will be publishing the Next.js app to Vercel. You can also publish to any other hosting provider that supports Next.js.

  1. If you haven't already, push your code to a git repo on GitHub.
  2. Create an account on Vercel and connect your GitHub account.
  3. Deploy your app to Vercel. You can follow the guide here.
  4. While deploying make sure to set the environment variable COHERE_API_KEY to your Cohere API key.
  5. Once your app is deployed you can update the siteURL section of your package.json file to point to the deployed URL. It will look something like https://your-site-here.vercel.app/
"config": {
  "siteURL": "https://your-site-here.vercel.app/"
}
  1. Run npm run build to create the production build of your plugin that points to your deployed URL.
  2. Test your plugin locally and make sure that it works after pointing to vercel.
  3. Publish your plugin to community
  4. After publishing to community your plugin will update automatically when you push to your git repo.

figmaAPI

This template includes a figmaAPI helper at @/lib/figmaAPI that lets you run plugin code from inside of the iframe. This is useful for avoiding the iframe <-> plugin postMessage API and reduces the amount of code you need to write.

Example:

import { figmaAPI } from "@/lib/figmaAPI";

const nodeId = "0:2";

const result = await figmaAPI.run(
  (figma, { nodeId }) => {
    return figma.getNodeById(nodeId)?.name;
  },
  // Any variable you want to pass to the function must be passed as a parameter.
  { nodeId },
);

console.log(result); // "Page 1"

A few things to note about this helper:

  1. The code cannot reference any variables outside of the function unless they are passed as a parameter to the second argument. This is because the code is stringified and sent to the plugin, and the plugin evals it. The plugin has no access to the variables in the iframe.
  2. The return value of the function must be JSON serializable. This is because the result is sent back to the iframe via postMessage, which only supports JSON.

Learn More