43 lines
1.2 KiB
Rust
43 lines
1.2 KiB
Rust
|
use std::path::PathBuf;
|
||
|
|
|
||
|
|
use noentropy::files::FileBatch;
|
||
|
|
use noentropy::files::OrganizationPlan;
|
||
|
|
use noentropy::files::execute_move;
|
||
|
|
use noentropy::gemini::GeminiClient;
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||
|
|
dotenv::dotenv().ok();
|
||
|
|
let api_key = std::env::var("GEMINI_API_KEY").expect("KEY not set");
|
||
|
|
let download_path_var = std::env::var("DOWNLOAD_FOLDER").expect("Set DOWNLOAD_FOLDER={path}");
|
||
|
|
|
||
|
|
// 1. Setup
|
||
|
|
let download_path: PathBuf = PathBuf::from(download_path_var.to_string());
|
||
|
|
let client: GeminiClient = GeminiClient::new(api_key);
|
||
|
|
|
||
|
|
// 2. Get Files (Using your previous FileBatch logic)
|
||
|
|
// Assuming FileBatch::from_path returns a struct with .filenames
|
||
|
|
let batch = FileBatch::from_path(download_path.clone());
|
||
|
|
|
||
|
|
if batch.filenames.is_empty() {
|
||
|
|
println!("No files found to organize!");
|
||
|
|
return Ok(());
|
||
|
|
}
|
||
|
|
|
||
|
|
println!(
|
||
|
|
"Found {} files. Asking Gemini to organize...",
|
||
|
|
batch.filenames.len()
|
||
|
|
);
|
||
|
|
|
||
|
|
// 3. Call Gemini
|
||
|
|
let plan: OrganizationPlan = client.organize_files(batch.filenames).await?;
|
||
|
|
|
||
|
|
println!("Gemini Plan received! Moving files...");
|
||
|
|
|
||
|
|
// 4. Execute
|
||
|
|
execute_move(&download_path, plan);
|
||
|
|
|
||
|
|
println!("Done!");
|
||
|
|
Ok(())
|
||
|
|
}
|