Merge pull request #9 from zeke/support-filenames-as-input

support filenames as input

Author: zeke

README.md

@@ -29,6 +29,13 @@ $ yolox "add 100px white padding around dots.png and save it as dots-with-room.p
 # convert dots.png -bordercolor white -border 100 dots-with-room.png
 ```
 
+You can also pass a file containing the prompt. This lets you write long prompts and iterate on them without retyping:
+
+```
+echo "do some stuff" > PROMPT.md
+$ yolox PROMPT.md
+```
+
 ## Caution
 
 This tool should be used with caution. It's called "YOLO X" because it's dangerous. **yolo** as in "you only live once" and *x* as in "execute this code". It lets an AI write code for you, then blindly executes that code on your system. There are a few guardrails in its prompt to prevent the result from taking destructive actions like deleting files or directories, but there's always still a danger that the resulting commands will have unintended consequences. You've been warned!
@@ -47,6 +54,10 @@ npx yolox@latest "use ffmpeg to convert foo.mkv to foo.mp4"
 
 ## Usage
 
+```
+yolox <prompt-string-or-filename-containing-prompt-string>
+```
+
 yolox supports [GPT4o](https://openai.com/index/hello-gpt-4o/) on OpenAI and [Llama 3](https://replicate.com/meta/meta-llama-3-70b-instruct) on Replicate.
 
 To add support for other models or providers, [open a pull request](https://github.com/zeke/yolox/issues)!

index.js

@@ -4,8 +4,12 @@ import { spawn } from 'node:child_process'
 import minimist from 'minimist'
 import { createClient } from './client.js'
 
+import { readFile } from 'node:fs/promises'
+import { existsSync } from 'node:fs'
+import { resolve } from 'node:path'
+
 const argv = minimist(process.argv.slice(2))
-const englishCommand = argv._.join(' ')
+let englishCommand = argv._.join(' ')
 const model = argv.model || 'gpt-4o'
 
 const models = {
@@ -28,6 +32,18 @@ if (!models[model]) {
   process.exit()
 }
 
+if (englishCommand && !englishCommand.includes(' ')) {
+  const filePath = resolve(process.cwd(), englishCommand)
+  if (existsSync(filePath)) {
+    try {
+      englishCommand = (await readFile(filePath, 'utf8')).trim()
+    } catch (error) {
+      console.error(`Error reading file: ${error.message}`)
+      process.exit(1)
+    }
+  }
+}
+
 const provider = models[model].split(':')[0]
 const modelFullName = models[model].split(':')[1]
 const client = createClient(provider)
@@ -76,9 +92,13 @@ child.stderr.on('data', (data) => {
 })
 
 child.on('error', (error) => {
-  console.error(`Error executing command: ${error.message}`)
+  console.error(`\nError executing command: ${error.message}`)
 })
 
 child.on('close', (code) => {
-  console.log(`Child process exited with code ${code}`)
+  if (code === 0) {
+    console.log(`\n✅ Success! Child process exited with code ${code}`)
+  } else {
+    console.log(`\n❌ Error! Child process exited with code ${code}`)
+  }
 })