opencode-avatar 0.1.0 → 0.2.0

package/README.md

@@ -42,12 +42,14 @@ Then add to your OpenCode config:
 
 ## Configuration
 
-### Environment Variables
+### API Key Configuration
 
-Create a `.env` file in the plugin directory:
+Create a config file at `~/.config/opencode/opencode-avatar.json`:
 
-```env
-FAL_KEY=your_fal_ai_api_key_here
+```json
+{
+  "falKey": "your_fal_ai_api_key_here"
+}
 ```
 
 Get your FAL.ai API key from [fal.ai](https://fal.ai).
@@ -168,7 +170,7 @@ The output shows:
 ### Avatar Not Showing
 
 1. Check console output for errors
-2. Verify FAL.ai API key in `.env`
+2. Verify FAL.ai API key in `~/.config/opencode/opencode-avatar.json`
 3. Ensure port 47291 is available
 4. Try restarting OpenCode
 

package/dist/electron.js

@@ -402,8 +402,23 @@ import { app, BrowserWindow, screen, Tray, Menu, nativeImage } from "electron";
 import * as path from "path";
 import * as http from "http";
 import * as fs from "fs";
+import * as os from "os";
 import { fileURLToPath } from "url";
 require_main().config();
+function getFalKey() {
+  try {
+    const configPath = path.join(os.homedir(), ".config", "opencode", "opencode-avatar.json");
+    const config = JSON.parse(fs.readFileSync(configPath, "utf8"));
+    if (!config.falKey) {
+      console.warn("Warning: falKey not found in config file. Avatar generation will not work. Please set falKey in ~/.config/opencode/opencode-avatar.json");
+      return null;
+    }
+    return config.falKey;
+  } catch (error) {
+    console.warn(`Warning: Failed to read config file: ${error.message}. Avatar generation will not work. Please ensure ~/.config/opencode/opencode-avatar.json exists and contains falKey.`);
+    return null;
+  }
+}
 var FAL_CDN_URL = "https://v3.fal.media";
 var FAL_REST_URL = "https://rest.alpha.fal.ai";
 var FAL_NANO_BANANA_URL = "https://fal.run/fal-ai/nano-banana-pro/edit";
@@ -551,13 +566,13 @@ function getAvatarPath() {
   }
   return path.join(AVATAR_DIR, "avatar.png");
 }
-async function uploadFile(filePath) {
+async function uploadFile(filePath, falKey) {
   const fileBuffer = fs.readFileSync(filePath);
   const fileName = path.basename(filePath);
   const tokenResponse = await fetch(`${FAL_REST_URL}/storage/auth/token?storage_type=fal-cdn-v3`, {
     method: "POST",
     headers: {
-      Authorization: `Key ${process.env.FAL_KEY}`,
+      Authorization: `Key ${falKey}`,
       Accept: "application/json",
       "Content-Type": "application/json"
     },
@@ -585,11 +600,11 @@ async function uploadFile(filePath) {
   const result = await response.json();
   return result.access_url || result.url || "";
 }
-async function generateAvatarImage(imageUrl, prompt) {
+async function generateAvatarImage(imageUrl, prompt, falKey) {
   const response = await fetch(FAL_NANO_BANANA_URL, {
     method: "POST",
     headers: {
-      Authorization: `Key ${process.env.FAL_KEY}`,
+      Authorization: `Key ${falKey}`,
       "Content-Type": "application/json"
     },
     body: JSON.stringify({
@@ -612,6 +627,11 @@ async function downloadImage(url, outputPath) {
   fs.writeFileSync(outputPath, buffer);
 }
 async function generateAvatarForPrompt(prompt) {
+  const falKey = getFalKey();
+  if (!falKey) {
+    console.warn("falKey is not set. Cannot generate avatar. Using default avatar.");
+    return path.join(AVATAR_DIR, "avatar.png");
+  }
   const cachedFilename = promptToFilename(prompt);
   const cachedPath = path.join(AVATAR_DIR, cachedFilename);
   if (fs.existsSync(cachedPath)) {
@@ -627,9 +647,9 @@ async function generateAvatarForPrompt(prompt) {
         return;
       }
       const sourceAvatar = path.join(AVATAR_DIR, "avatar.png");
-      const uploadedUrl = await uploadFile(sourceAvatar);
+      const uploadedUrl = await uploadFile(sourceAvatar, falKey);
       const fullPrompt = `make a character variant: ${prompt}. Keep the background as a solid green screen color. Do not let the green screen color appear in reflections or on the subject.`;
-      const result = await generateAvatarImage(uploadedUrl, fullPrompt);
+      const result = await generateAvatarImage(uploadedUrl, fullPrompt, falKey);
       const outputUrl = result.images?.[0]?.url || result.image?.url || result.url;
       if (!outputUrl) {
         throw new Error("No output image URL in response: " + JSON.stringify(result, null, 2));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-avatar",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Dynamic desktop avatar plugin for OpenCode that reacts to your coding activities",
   "type": "module",
   "main": "./dist/index.js",