1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
package cli
import (
"fmt"
"os"
"path/filepath"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"codeberg.org/snonux/totalrecall/internal"
)
// CreateRootCommand creates and configures the root cobra command
func CreateRootCommand(flags *Flags) *cobra.Command {
rootCmd := &cobra.Command{
Use: "totalrecall [word]",
Short: "Bulgarian Anki Flashcard Generator",
Long: `totalrecall generates Anki flashcard materials from Bulgarian words.
It creates audio pronunciation files using OpenAI TTS and downloads
representative images from web search APIs.
Examples:
totalrecall # Launch interactive GUI (default)
totalrecall ябълка # Generate materials for "apple" via CLI
totalrecall --batch words.txt # Process multiple words from file
totalrecall --archive # Archive existing cards directory
Batch file formats:
ябълка # Bulgarian word (will be translated to English)
ябълка = apple # Bulgarian with translation provided
= apple # English only (will be translated to Bulgarian)`,
Args: cobra.MaximumNArgs(1),
Version: internal.Version,
}
// Set up flags
setupFlags(rootCmd, flags)
return rootCmd
}
func setupFlags(cmd *cobra.Command, flags *Flags) {
// Set default output directory to match GUI mode
home, _ := os.UserHomeDir()
defaultOutputDir := filepath.Join(home, ".local", "state", "totalrecall", "cards")
// Global flags
// AGENT: The default config file location should be ~/.config/totalrecall/config.yaml
cmd.PersistentFlags().StringVar(&flags.CfgFile, "config", "", "config file (default is $HOME/.totalrecall.yaml)")
// Local flags
cmd.Flags().StringVarP(&flags.OutputDir, "output", "o", defaultOutputDir, "Output directory")
cmd.Flags().StringVarP(&flags.AudioFormat, "format", "f", flags.AudioFormat, "Audio format (wav or mp3)")
cmd.Flags().StringVar(&flags.ImageAPI, "image-api", flags.ImageAPI, "Image source (only openai supported)")
cmd.Flags().StringVar(&flags.BatchFile, "batch", "", "Process words from file (one per line)")
cmd.Flags().BoolVar(&flags.SkipAudio, "skip-audio", false, "Skip audio generation")
cmd.Flags().BoolVar(&flags.SkipImages, "skip-images", false, "Skip image download")
cmd.Flags().BoolVar(&flags.GenerateAnki, "anki", false, "Generate Anki import file (APKG format by default, use --anki-csv for legacy CSV)")
cmd.Flags().BoolVar(&flags.AnkiCSV, "anki-csv", false, "Generate legacy CSV format instead of APKG when using --anki")
cmd.Flags().StringVar(&flags.DeckName, "deck-name", flags.DeckName, "Deck name for APKG export")
cmd.Flags().BoolVar(&flags.ListModels, "list-models", false, "List available OpenAI models for the current API key")
cmd.Flags().BoolVar(&flags.AllVoices, "all-voices", false, "Generate audio in all available voices (creates multiple files)")
cmd.Flags().BoolVar(&flags.NoAutoPlay, "no-auto-play", false, "Disable automatic audio playback in GUI mode (auto-play is enabled by default)")
cmd.Flags().BoolVar(&flags.Archive, "archive", false, "Archive existing cards directory with timestamp")
// OpenAI flags
cmd.Flags().StringVar(&flags.OpenAIModel, "openai-model", flags.OpenAIModel, "OpenAI TTS model: tts-1, tts-1-hd, gpt-4o-mini-tts")
cmd.Flags().StringVar(&flags.OpenAIVoice, "openai-voice", "", "OpenAI voice: alloy, ash, ballad, coral, echo, fable, onyx, nova, sage, shimmer, verse (default: random)")
cmd.Flags().Float64Var(&flags.OpenAISpeed, "openai-speed", flags.OpenAISpeed, "OpenAI speech speed (0.25 to 4.0, may be ignored by gpt-4o-mini-tts)")
cmd.Flags().StringVar(&flags.OpenAIInstruction, "openai-instruction", "", "Voice instructions for gpt-4o-mini-tts model (e.g., 'speak slowly with a Bulgarian accent')")
// OpenAI Image Generation flags
cmd.Flags().StringVar(&flags.OpenAIImageModel, "openai-image-model", flags.OpenAIImageModel, "OpenAI image model: dall-e-2 or dall-e-3")
cmd.Flags().StringVar(&flags.OpenAIImageSize, "openai-image-size", flags.OpenAIImageSize, "Image size: 256x256, 512x512, 1024x1024 (dall-e-3: also 1024x1792, 1792x1024)")
cmd.Flags().StringVar(&flags.OpenAIImageQuality, "openai-image-quality", flags.OpenAIImageQuality, "Image quality: standard or hd (dall-e-3 only)")
cmd.Flags().StringVar(&flags.OpenAIImageStyle, "openai-image-style", flags.OpenAIImageStyle, "Image style: natural or vivid (dall-e-3 only)")
// Bind flags to viper
bindFlagsToViper(cmd)
}
func bindFlagsToViper(cmd *cobra.Command) {
viper.BindPFlag("audio.provider", cmd.Flags().Lookup("audio-provider"))
viper.BindPFlag("audio.voice", cmd.Flags().Lookup("voice"))
viper.BindPFlag("audio.format", cmd.Flags().Lookup("format"))
viper.BindPFlag("audio.pitch", cmd.Flags().Lookup("pitch"))
viper.BindPFlag("audio.amplitude", cmd.Flags().Lookup("amplitude"))
viper.BindPFlag("audio.word_gap", cmd.Flags().Lookup("word-gap"))
viper.BindPFlag("audio.openai_model", cmd.Flags().Lookup("openai-model"))
viper.BindPFlag("audio.openai_voice", cmd.Flags().Lookup("openai-voice"))
viper.BindPFlag("audio.openai_speed", cmd.Flags().Lookup("openai-speed"))
viper.BindPFlag("audio.openai_instruction", cmd.Flags().Lookup("openai-instruction"))
viper.BindPFlag("output.directory", cmd.Flags().Lookup("output"))
viper.BindPFlag("image.provider", cmd.Flags().Lookup("image-api"))
// Bind OpenAI image flags
viper.BindPFlag("image.openai_model", cmd.Flags().Lookup("openai-image-model"))
viper.BindPFlag("image.openai_size", cmd.Flags().Lookup("openai-image-size"))
viper.BindPFlag("image.openai_quality", cmd.Flags().Lookup("openai-image-quality"))
viper.BindPFlag("image.openai_style", cmd.Flags().Lookup("openai-image-style"))
}
// InitConfig initializes viper configuration
func InitConfig(cfgFile string) {
if cfgFile != "" {
// Use config file from the flag
viper.SetConfigFile(cfgFile)
} else {
// Find home directory
home, err := os.UserHomeDir()
if err != nil {
fmt.Fprintf(os.Stderr, "Error getting home directory: %v\n", err)
return
}
// Search config in home directory with name ".totalrecall" (without extension)
viper.AddConfigPath(home)
viper.AddConfigPath(".")
viper.SetConfigType("yaml")
viper.SetConfigName(".totalrecall")
}
// Environment variables
viper.SetEnvPrefix("TOTALRECALL")
viper.AutomaticEnv()
// Read config file
if err := viper.ReadInConfig(); err == nil {
fmt.Fprintln(os.Stderr, "Using config file:", viper.ConfigFileUsed())
}
}
// GetOpenAIKey retrieves the OpenAI API key from environment or config
func GetOpenAIKey() string {
// First check environment variable
if key := os.Getenv("OPENAI_API_KEY"); key != "" {
return key
}
// Then check config file
return viper.GetString("audio.openai_key")
}
|
