trans-epub 0.0.23

Translate EPUB with CLI
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

mod client;
mod epub;
mod translate;

use crate::epub::Epub;
use crate::translate::translator::{Context, Translator};
use clap::{Parser, Subcommand};
use env_logger::Env;
use log::debug;
use std::path::PathBuf;

#[derive(Parser)]
#[command(version, about)]
struct Args {
    #[clap(subcommand)]
    subcommand: SubCommands,
}

#[derive(Subcommand)]
enum SubCommands {
    /// Use OpenAI API
    OpenAi {
        /// input file path
        #[arg(short, long)]
        input: PathBuf,

        /// output file path
        #[arg(short, long)]
        output: PathBuf,

        /// translate language
        #[arg(short, long)]
        language: String,

        /// OpenAI model ex(gpt-4o-mini, gpt-4o, gpt-4-turbo, gpt-3.5-turbo-1106)
        #[arg(short, long, default_value_t = String::from("gpt-4o-mini"))]
        model: String,

        /// OpenAI API Key
        #[arg(short, long, env, hide_env_values = true)]
        api_key: String,

        /// Number of lines of translation
        #[arg(long, default_value_t = 20)]
        lines: usize,

        /// Number of concurrent requests
        #[arg(long, default_value_t = 5)]
        requests: usize,
    },
    /// Use Gemini API
    Gemini {
        /// input file path
        #[arg(short, long)]
        input: PathBuf,

        /// output file path
        #[arg(short, long)]
        output: PathBuf,

        /// translate language
        #[arg(short, long)]
        language: String,

        /// Gemini model ex(gemini-2.0-flash-lite, gemini-1.5-flash)
        #[arg(short, long, default_value_t = String::from("gemini-2.0-flash-lite"))]
        model: String,

        /// Gemini API Key
        #[arg(short, long, env, hide_env_values = true)]
        api_key: String,

        /// Number of lines of translation
        #[arg(long, default_value_t = 100)]
        lines: usize,

        /// Number of concurrent requests
        #[arg(long, default_value_t = 1)]
        requests: usize,
    },
}

#[tokio::main]
async fn main() {
    env_logger::Builder::from_env(Env::default().default_filter_or("info")).init();
    debug!("start");
    let args = Args::parse();
    match args.subcommand {
        SubCommands::OpenAi {
            api_key,
            model,
            language,
            lines,
            requests,
            input,
            output,
        } => {
            let translator = Translator::OpenAi(Context {
                model,
                api_key,
                language,
                lines,
                requests,
            });
            let epub = Epub::new(input, output);
            epub.translate(translator).await;
        }
        SubCommands::Gemini {
            api_key,
            model,
            language,
            lines,
            requests,
            input,
            output,
        } => {
            let translator = Translator::Gemini(Context {
                model,
                api_key,
                language,
                lines,
                requests,
            });
            let epub = Epub::new(input, output);
            epub.translate(translator).await;
        }
    }
    debug!("end");
}