Skip to content

Commit

Permalink
separate commander and inquirer logic
Browse files Browse the repository at this point in the history
  • Loading branch information
@ajcwebdev
ajcwebdev committed Oct 1, 2024
1 parent e54d9d6 commit 9529db8
Show file tree
Hide file tree
Showing 2 changed files with 200 additions and 195 deletions.
207 changes: 12 additions & 195 deletions src/autoshow.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,15 @@
*/

import { Command } from 'commander'
import inquirer from 'inquirer'
import { handleInteractivePrompt } from './inquirer.js'
import { processVideo } from './commands/processVideo.js'
import { processPlaylist } from './commands/processPlaylist.js'
import { processURLs } from './commands/processURLs.js'
import { processFile } from './commands/processFile.js'
import { processRSS } from './commands/processRSS.js'
import { argv } from 'node:process'

/** @import { ProcessingOptions, InquirerAnswers, InquirerQuestions, HandlerFunction, LLMOption, TranscriptOption, WhisperModelType } from './types.js' */
/** @import { ProcessingOptions, HandlerFunction, LLMOption, TranscriptOption } from './types.js' */

// Initialize the command-line interface
const program = new Command()
Expand Down Expand Up @@ -51,189 +51,6 @@ program
.option('--gemini [model]', 'Use Gemini for processing with optional model specification')
.option('--noCleanUp', 'Do not delete intermediary files after processing')

// Interactive prompts using inquirer
/** @type {InquirerQuestions} */
const INQUIRER_PROMPT = [
{
type: 'list',
name: 'action',
message: 'What would you like to process?',
choices: [
{ name: 'Single YouTube Video', value: 'video' },
{ name: 'YouTube Playlist', value: 'playlist' },
{ name: 'List of URLs from File', value: 'urls' },
{ name: 'Local Audio/Video File', value: 'file' },
{ name: 'Podcast RSS Feed', value: 'rss' },
],
},
{
type: 'input',
name: 'video',
message: 'Enter the YouTube video URL:',
when: (answers) => answers.action === 'video',
validate: (input) => (input ? true : 'Please enter a valid URL.'),
},
{
type: 'input',
name: 'playlist',
message: 'Enter the YouTube playlist URL:',
when: (answers) => answers.action === 'playlist',
validate: (input) => (input ? true : 'Please enter a valid URL.'),
},
{
type: 'input',
name: 'urls',
message: 'Enter the file path containing URLs:',
when: (answers) => answers.action === 'urls',
validate: (input) => (input ? true : 'Please enter a valid file path.'),
},
{
type: 'input',
name: 'file',
message: 'Enter the local audio/video file path:',
when: (answers) => answers.action === 'file',
validate: (input) => (input ? true : 'Please enter a valid file path.'),
},
{
type: 'input',
name: 'rss',
message: 'Enter the podcast RSS feed URL:',
when: (answers) => answers.action === 'rss',
validate: (input) => (input ? true : 'Please enter a valid URL.'),
},
{
type: 'confirm',
name: 'specifyItem',
message: 'Do you want to process specific episodes by providing their audio URLs?',
when: (answers) => answers.action === 'rss',
default: false,
},
{
type: 'input',
name: 'item',
message: 'Enter the audio URLs of the episodes (separated by commas):',
when: (answers) => answers.action === 'rss' && answers.specifyItem,
validate: (input) => (input ? true : 'Please enter at least one valid audio URL.'),
},
{
type: 'list',
name: 'llmOpt',
message: 'Select the Language Model (LLM) you want to use:',
choices: [
{ name: 'OpenAI ChatGPT', value: 'chatgpt' },
{ name: 'Anthropic Claude', value: 'claude' },
{ name: 'Cohere', value: 'cohere' },
{ name: 'Mistral', value: 'mistral' },
{ name: 'OctoAI', value: 'octo' },
{ name: 'node-llama-cpp (local inference)', value: 'llama' },
{ name: 'Google Gemini', value: 'gemini' },
{ name: 'Skip LLM Processing', value: null },
],
},
{
type: 'list',
name: 'llamaModel',
message: 'Select the LLAMA model you want to use:',
choices: [
{ name: 'LLAMA 3 8B Q4 Model', value: 'LLAMA_3_1_8B_Q4_MODEL' },
{ name: 'LLAMA 3 8B Q6 Model', value: 'LLAMA_3_1_8B_Q6_MODEL' },
{ name: 'GEMMA 2 2B Q4 Model', value: 'GEMMA_2_2B_Q4_MODEL' },
{ name: 'GEMMA 2 2B Q6 Model', value: 'GEMMA_2_2B_Q6_MODEL' },
],
when: (answers) => answers.llmOpt === 'llama',
},
{
type: 'list',
name: 'transcriptOpt',
message: 'Select the transcription service you want to use:',
choices: [
{ name: 'Whisper.cpp', value: 'whisper' },
{ name: 'Deepgram', value: 'deepgram' },
{ name: 'AssemblyAI', value: 'assembly' },
],
},
{
type: 'confirm',
name: 'useDocker',
message: 'Do you want to run Whisper.cpp in a Docker container?',
when: (answers) => answers.transcriptOpt === 'whisper',
default: false,
},
{
type: 'list',
name: 'whisperModel',
message: 'Select the Whisper model type:',
choices: ['tiny', 'tiny.en', 'base', 'base.en', 'small', 'small.en', 'medium', 'medium.en', 'large', 'large-v1', 'large-v2'],
when: (answers) => answers.transcriptOpt === 'whisper',
default: 'large',
},
{
type: 'confirm',
name: 'speakerLabels',
message: 'Do you want to use speaker labels?',
when: (answers) => answers.transcriptOpt === 'assembly',
default: false,
},
{
type: 'checkbox',
name: 'prompt',
message: 'Select the prompt sections to include:',
choices: [
{ name: 'Titles', value: 'titles' },
{ name: 'Summary', value: 'summary' },
{ name: 'Short Chapters', value: 'shortChapters' },
{ name: 'Medium Chapters', value: 'mediumChapters' },
{ name: 'Long Chapters', value: 'longChapters' },
{ name: 'Key Takeaways', value: 'takeaways' },
{ name: 'Questions', value: 'questions' },
],
default: ['summary', 'longChapters'],
},
{
type: 'confirm',
name: 'noCleanUp',
message: 'Do you want to keep intermediary files after processing?',
default: false,
},
]

/**
* Prompts the user for input if no command-line options are provided.
* @param {ProcessingOptions} options - The initial command-line options.
* @returns {Promise<ProcessingOptions>} - The updated options after user input.
*/
async function handleInteractivePrompt(options) {
/** @type {InquirerAnswers} */
const answers = await inquirer.prompt(INQUIRER_PROMPT)
options = {
...options,
...answers,
}

// Handle LLM options
if (answers.llmOpt) {
options[answers.llmOpt] = answers.llmOpt === 'llama' ? answers.llamaModel : true
}

// Handle transcription options
if (answers.transcriptOpt === 'whisper') {
if (answers.useDocker) {
options.whisperDocker = /** @type {WhisperModelType} */ (answers.whisperModel)
} else {
options.whisper = /** @type {WhisperModelType} */ (answers.whisperModel)
}
} else {
options[answers.transcriptOpt] = true
}

// Handle 'item' for RSS feed
if (answers.item && typeof answers.item === 'string') {
options.item = answers.item.split(',').map((url) => url.trim())
}

return options
}

/**
* Main action for the program.
* @param {ProcessingOptions} options - The command-line options provided by the user.
Expand All @@ -242,12 +59,12 @@ async function handleInteractivePrompt(options) {
program.action(async (options) => {
console.log(`Options received:\n`)
console.log(options)
const { video, playlist, urls, file, rss } = options

// Check if no input options are provided and if so, prompt the user interactively
const noInputOptions = !options.video && !options.playlist && !options.urls && !options.file && !options.rss
if (noInputOptions) {
options = await handleInteractivePrompt(options)
}
options = [video, playlist, urls, file, rss].every(opt => !opt)
? await handleInteractivePrompt(options)
: options

// Ensure options.item is an array if provided via command line
if (options.item && !Array.isArray(options.item)) {
Expand All @@ -270,17 +87,17 @@ program.action(async (options) => {
* Determine the selected LLM option
* @type {LLMOption | undefined}
*/
const llmOpt = /** @type {LLMOption | undefined} */ (['chatgpt', 'claude', 'cohere', 'mistral', 'octo', 'llama', 'ollama', 'gemini'].find(
(option) => options[option]
))
const llmOpt = /** @type {LLMOption | undefined} */ ([
'chatgpt', 'claude', 'cohere', 'mistral', 'octo', 'llama', 'ollama', 'gemini'
].find((option) => options[option]))

/**
* Determine the transcription service to use
* @type {TranscriptOption | undefined}
*/
const transcriptOpt = /** @type {TranscriptOption | undefined} */ (['whisper', 'whisperDocker', 'deepgram', 'assembly'].find(
(option) => options[option]
))
const transcriptOpt = /** @type {TranscriptOption | undefined} */ ([
'whisper', 'whisperDocker', 'deepgram', 'assembly'
].find((option) => options[option]))

// Execute the appropriate handler based on the action
for (const [key, handler] of Object.entries(handlers)) {
Expand Down
Loading

0 comments on commit 9529db8

Please sign in to comment.