From 0fa5b5dd8f65e4b8ae2873dc8244a4684e308786 Mon Sep 17 00:00:00 2001
From: Anthony Campolo <12433465+ajcwebdev@users.noreply.github.com>
Date: Tue, 5 Nov 2024 10:51:00 -0800
Subject: [PATCH] refactor interactive cli file

---
 src/globals.ts     | 142 +++++++++++++++++++++++++++++++++++++++++++++
 src/interactive.ts | 141 +++-----------------------------------------
 2 files changed, 149 insertions(+), 134 deletions(-)

diff --git a/src/globals.ts b/src/globals.ts
index 3e569590..9ad32ccc 100644
--- a/src/globals.ts
+++ b/src/globals.ts
@@ -15,6 +15,148 @@ import type { WhisperModelType, ChatGPTModelType, ClaudeModelType, CohereModelTy
 export const execPromise = promisify(exec)
 export const execFilePromise = promisify(execFile)
 
+export const PROCESS_CHOICES = [
+  { name: 'Single YouTube Video', value: 'video' },
+  { name: 'YouTube Playlist', value: 'playlist' },
+  { name: 'YouTube Channel', value: 'channel' },
+  { name: 'List of URLs from File', value: 'urls' },
+  { name: 'Local Audio/Video File', value: 'file' },
+  { name: 'Podcast RSS Feed', value: 'rss' },
+]
+
+export const TRANSCRIPT_CHOICES = [
+  { name: 'Whisper.cpp', value: 'whisper' },
+  { name: 'Whisper.cpp (Docker)', value: 'whisperDocker' },
+  { name: 'Whisper Python', value: 'whisperPython' },
+  { name: 'Whisper Diarization', value: 'whisperDiarization' },
+  { name: 'Deepgram', value: 'deepgram' },
+  { name: 'AssemblyAI', value: 'assembly' },
+]
+
+export const WHISPER_MODEL_CHOICES = [
+  'tiny',
+  'tiny.en',
+  'base',
+  'base.en',
+  'small',
+  'small.en',
+  'medium',
+  'medium.en',
+  'large-v1',
+  'large-v2',
+  'turbo',
+]
+
+export const WHISPER_LIBRARY_CHOICES = [
+  'whisper',
+  'whisperDocker',
+  'whisperPython',
+  'whisperDiarization'
+]
+
+export const LLM_SERVICE_CHOICES = [
+  { name: 'Skip LLM Processing', value: null },
+  { name: 'Ollama (local inference)', value: 'ollama' },
+  { name: 'OpenAI ChatGPT', value: 'chatgpt' },
+  { name: 'Anthropic Claude', value: 'claude' },
+  { name: 'Google Gemini', value: 'gemini' },
+  { name: 'Cohere', value: 'cohere' },
+  { name: 'Mistral', value: 'mistral' },
+  { name: 'Fireworks AI', value: 'fireworks' },
+  { name: 'Together AI', value: 'together' },
+  { name: 'Groq', value: 'groq' },
+]
+
+export const LLM_CHOICES = [
+  'ollama',
+  'chatgpt',
+  'claude',
+  'cohere',
+  'mistral',
+  'fireworks',
+  'together',
+  'groq',
+  'gemini',
+]
+
+export const OLLAMA_CHOICES = [
+  { name: 'LLAMA 3 2 1B', value: 'LLAMA_3_2_1B' },
+  { name: 'LLAMA 3 2 3B', value: 'LLAMA_3_2_3B' },
+  { name: 'GEMMA 2 2B', value: 'GEMMA_2_2B' },
+  { name: 'PHI 3 5', value: 'PHI_3_5' },
+  { name: 'QWEN 2 5 1B', value: 'QWEN_2_5_1B' },
+  { name: 'QWEN 2 5 3B', value: 'QWEN_2_5_3B' },
+]
+
+export const CHATGPT_CHOICES = [
+  { name: 'GPT 4 o MINI', value: 'GPT_4o_MINI' },
+  { name: 'GPT 4 o', value: 'GPT_4o' },
+  { name: 'GPT 4 TURBO', value: 'GPT_4_TURBO' },
+  { name: 'GPT 4', value: 'GPT_4' },
+]
+
+export const CLAUDE_CHOICES = [
+  { name: 'Claude 3.5 Sonnet', value: 'CLAUDE_3_5_SONNET' },
+  { name: 'Claude 3 Opus', value: 'CLAUDE_3_OPUS' },
+  { name: 'Claude 3 Sonnet', value: 'CLAUDE_3_SONNET' },
+  { name: 'Claude 3 Haiku', value: 'CLAUDE_3_HAIKU' },
+]
+
+export const COHERE_CHOICES = [
+  { name: 'Command R', value: 'COMMAND_R' },
+  { name: 'Command R Plus', value: 'COMMAND_R_PLUS' },
+]
+
+export const MISTRAL_CHOICES = [
+  { name: 'Mixtral 8x7b', value: 'MIXTRAL_8x7b' },
+  { name: 'Mixtral 8x22b', value: 'MIXTRAL_8x22b' },
+  { name: 'Mistral Large', value: 'MISTRAL_LARGE' },
+  { name: 'Mistral Nemo', value: 'MISTRAL_NEMO' },
+]
+
+export const FIREWORKS_CHOICES = [
+  { name: 'LLAMA 3 1 405B', value: 'LLAMA_3_1_405B' },
+  { name: 'LLAMA 3 1 70B', value: 'LLAMA_3_1_70B' },
+  { name: 'LLAMA 3 1 8B', value: 'LLAMA_3_1_8B' },
+  { name: 'LLAMA 3 2 3B', value: 'LLAMA_3_2_3B' },
+  { name: 'LLAMA 3 2 1B', value: 'LLAMA_3_2_1B' },
+  { name: 'QWEN 2 5 72B', value: 'QWEN_2_5_72B' },
+]
+
+export const TOGETHER_CHOICES = [
+  { name: 'LLAMA 3 2 3B', value: 'LLAMA_3_2_3B' },
+  { name: 'LLAMA 3 1 405B', value: 'LLAMA_3_1_405B' },
+  { name: 'LLAMA 3 1 70B', value: 'LLAMA_3_1_70B' },
+  { name: 'LLAMA 3 1 8B', value: 'LLAMA_3_1_8B' },
+  { name: 'Gemma 2 27B', value: 'GEMMA_2_27B' },
+  { name: 'Gemma 2 9B', value: 'GEMMA_2_9B' },
+  { name: 'QWEN 2 5 72B', value: 'QWEN_2_5_72B' },
+  { name: 'QWEN 2 5 7B', value: 'QWEN_2_5_7B' },
+]
+
+export const GROQ_CHOICES = [
+  { name: 'LLAMA 3 1 70B Versatile', value: 'LLAMA_3_1_70B_VERSATILE' },
+  { name: 'LLAMA 3 1 8B Instant', value: 'LLAMA_3_1_8B_INSTANT' },
+  { name: 'LLAMA 3 2 1B Preview', value: 'LLAMA_3_2_1B_PREVIEW' },
+  { name: 'LLAMA 3 2 3B Preview', value: 'LLAMA_3_2_3B_PREVIEW' },
+  { name: 'Mixtral 8x7b 32768', value: 'MIXTRAL_8X7B_32768' },
+]
+
+export const GEMINI_CHOICES = [
+  { name: 'Gemini 1.5 Flash', value: 'GEMINI_1_5_FLASH' },
+  { name: 'Gemini 1.5 Pro', value: 'GEMINI_1_5_PRO' },
+]
+
+export const PROMPT_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' },
+]
+
 /**
  * Configure XML parser for RSS feed processing
  * Handles attributes without prefixes and allows boolean values
diff --git a/src/interactive.ts b/src/interactive.ts
index 6d4e1ab6..b0474f4a 100644
--- a/src/interactive.ts
+++ b/src/interactive.ts
@@ -2,110 +2,9 @@
 
 import inquirer from 'inquirer'
 import type { ProcessingOptions, InquirerAnswers, WhisperModelType } from './types.js'
-import { l } from './globals.js'
-
-const TRANSCRIPT_CHOICES = [
-  { name: 'Whisper.cpp', value: 'whisper' },
-  { name: 'Whisper.cpp (Docker)', value: 'whisperDocker' },
-  { name: 'Whisper Python', value: 'whisperPython' },
-  { name: 'Whisper Diarization', value: 'whisperDiarization' },
-  { name: 'Deepgram', value: 'deepgram' },
-  { name: 'AssemblyAI', value: 'assembly' },
-]
-
-const WHISPER_MODEL_CHOICES = [
-  'tiny',
-  'tiny.en',
-  'base',
-  'base.en',
-  'small',
-  'small.en',
-  'medium',
-  'medium.en',
-  'large-v1',
-  'large-v2',
-  'turbo',
-]
-
-const LLM_CHOICES = [
-  'ollama',
-  'chatgpt',
-  'claude',
-  'cohere',
-  'mistral',
-  'fireworks',
-  'together',
-  'groq',
-  'gemini',
-]
-
-const OLLAMA_CHOICES = [
-  { name: 'LLAMA 3 2 1B', value: 'LLAMA_3_2_1B' },
-  { name: 'LLAMA 3 2 3B', value: 'LLAMA_3_2_3B' },
-  { name: 'GEMMA 2 2B', value: 'GEMMA_2_2B' },
-  { name: 'PHI 3 5', value: 'PHI_3_5' },
-  { name: 'QWEN 2 5 1B', value: 'QWEN_2_5_1B' },
-  { name: 'QWEN 2 5 3B', value: 'QWEN_2_5_3B' },
-]
-
-const CHATGPT_CHOICES = [
-  { name: 'GPT 4 o MINI', value: 'GPT_4o_MINI' },
-  { name: 'GPT 4 o', value: 'GPT_4o' },
-  { name: 'GPT 4 TURBO', value: 'GPT_4_TURBO' },
-  { name: 'GPT 4', value: 'GPT_4' },
-]
-
-const CLAUDE_CHOICES = [
-  { name: 'Claude 3.5 Sonnet', value: 'CLAUDE_3_5_SONNET' },
-  { name: 'Claude 3 Opus', value: 'CLAUDE_3_OPUS' },
-  { name: 'Claude 3 Sonnet', value: 'CLAUDE_3_SONNET' },
-  { name: 'Claude 3 Haiku', value: 'CLAUDE_3_HAIKU' },
-]
-
-const COHERE_CHOICES = [
-  { name: 'Command R', value: 'COMMAND_R' },
-  { name: 'Command R Plus', value: 'COMMAND_R_PLUS' },
-]
-
-const MISTRAL_CHOICES = [
-  { name: 'Mixtral 8x7b', value: 'MIXTRAL_8x7b' },
-  { name: 'Mixtral 8x22b', value: 'MIXTRAL_8x22b' },
-  { name: 'Mistral Large', value: 'MISTRAL_LARGE' },
-  { name: 'Mistral Nemo', value: 'MISTRAL_NEMO' },
-]
-
-const FIREWORKS_CHOICES = [
-  { name: 'LLAMA 3 1 405B', value: 'LLAMA_3_1_405B' },
-  { name: 'LLAMA 3 1 70B', value: 'LLAMA_3_1_70B' },
-  { name: 'LLAMA 3 1 8B', value: 'LLAMA_3_1_8B' },
-  { name: 'LLAMA 3 2 3B', value: 'LLAMA_3_2_3B' },
-  { name: 'LLAMA 3 2 1B', value: 'LLAMA_3_2_1B' },
-  { name: 'QWEN 2 5 72B', value: 'QWEN_2_5_72B' },
-]
-
-const TOGETHER_CHOICES = [
-  { name: 'LLAMA 3 2 3B', value: 'LLAMA_3_2_3B' },
-  { name: 'LLAMA 3 1 405B', value: 'LLAMA_3_1_405B' },
-  { name: 'LLAMA 3 1 70B', value: 'LLAMA_3_1_70B' },
-  { name: 'LLAMA 3 1 8B', value: 'LLAMA_3_1_8B' },
-  { name: 'Gemma 2 27B', value: 'GEMMA_2_27B' },
-  { name: 'Gemma 2 9B', value: 'GEMMA_2_9B' },
-  { name: 'QWEN 2 5 72B', value: 'QWEN_2_5_72B' },
-  { name: 'QWEN 2 5 7B', value: 'QWEN_2_5_7B' },
-]
-
-const GROQ_CHOICES = [
-  { name: 'LLAMA 3 1 70B Versatile', value: 'LLAMA_3_1_70B_VERSATILE' },
-  { name: 'LLAMA 3 1 8B Instant', value: 'LLAMA_3_1_8B_INSTANT' },
-  { name: 'LLAMA 3 2 1B Preview', value: 'LLAMA_3_2_1B_PREVIEW' },
-  { name: 'LLAMA 3 2 3B Preview', value: 'LLAMA_3_2_3B_PREVIEW' },
-  { name: 'Mixtral 8x7b 32768', value: 'MIXTRAL_8X7B_32768' },
-]
-
-const GEMINI_CHOICES = [
-  { name: 'Gemini 1.5 Flash', value: 'GEMINI_1_5_FLASH' },
-  { name: 'Gemini 1.5 Pro', value: 'GEMINI_1_5_PRO' },
-]
+import {
+  l, PROCESS_CHOICES, TRANSCRIPT_CHOICES, WHISPER_MODEL_CHOICES, WHISPER_LIBRARY_CHOICES, LLM_SERVICE_CHOICES, LLM_CHOICES, OLLAMA_CHOICES, CHATGPT_CHOICES, CLAUDE_CHOICES, COHERE_CHOICES, MISTRAL_CHOICES, FIREWORKS_CHOICES, TOGETHER_CHOICES, GROQ_CHOICES, GEMINI_CHOICES, PROMPT_CHOICES
+} from './globals.js'
 
 /**
  * Prompts the user for input if interactive mode is selected.
@@ -125,14 +24,7 @@ export async function handleInteractivePrompt(
       type: 'list',
       name: 'action',
       message: 'What would you like to process?',
-      choices: [
-        { name: 'Single YouTube Video', value: 'video' },
-        { name: 'YouTube Playlist', value: 'playlist' },
-        { name: 'YouTube Channel', value: 'channel' },
-        { name: 'List of URLs from File', value: 'urls' },
-        { name: 'Local Audio/Video File', value: 'file' },
-        { name: 'Podcast RSS Feed', value: 'rss' },
-      ],
+      choices: PROCESS_CHOICES,
     },
     // Input prompts for different content sources
     {
@@ -237,18 +129,7 @@ export async function handleInteractivePrompt(
       type: 'list',
       name: 'llmServices',
       message: 'Select the Language Model (LLM) you want to use:',
-      choices: [
-        { name: 'Skip LLM Processing', value: null },
-        { name: 'Ollama (local inference)', value: 'ollama' },
-        { name: 'OpenAI ChatGPT', value: 'chatgpt' },
-        { name: 'Anthropic Claude', value: 'claude' },
-        { name: 'Google Gemini', value: 'gemini' },
-        { name: 'Cohere', value: 'cohere' },
-        { name: 'Mistral', value: 'mistral' },
-        { name: 'Fireworks AI', value: 'fireworks' },
-        { name: 'Together AI', value: 'together' },
-        { name: 'Groq', value: 'groq' },
-      ],
+      choices: LLM_SERVICE_CHOICES,
     },
     // Model selection based on chosen LLM service
     {
@@ -314,15 +195,7 @@ export async function handleInteractivePrompt(
       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' },
-      ],
+      choices: PROMPT_CHOICES,
       default: ['summary', 'longChapters'],
     },
     {
@@ -351,7 +224,7 @@ export async function handleInteractivePrompt(
   // Configure transcription service options based on user selection
   if (answers.transcriptServices) {
     if (
-      ['whisper', 'whisperDocker', 'whisperPython', 'whisperDiarization'].includes(
+      WHISPER_LIBRARY_CHOICES.includes(
         answers.transcriptServices
       )
     ) {