refactored every component thoroughly.

src/gemini/prompt.rs

@@ -0,0 +1,51 @@
+use crate::models::{FileCategory, OrganizationPlan};
+use crate::gemini::types::OrganizationPlanResponse;
+
+impl OrganizationPlanResponse {
+    pub fn to_organization_plan(self) -> OrganizationPlan {
+        OrganizationPlan {
+            files: self
+                .files
+                .into_iter()
+                .map(|f| FileCategory {
+                    filename: f.filename,
+                    category: f.category,
+                    sub_category: String::new(),
+                })
+                .collect(),
+        }
+    }
+}
+
+#[derive(Debug)]
+pub struct PromptBuilder {
+    file_list: String,
+}
+
+impl PromptBuilder {
+    pub fn new(file_list: Vec<String>) -> Self {
+        Self {
+            file_list: file_list.join(", "),
+        }
+    }
+
+    pub fn build_categorization_prompt(&self) -> String {
+        format!(
+            "I have these files in my Downloads folder: [{}]. \
+             Categorize them into these folders: 'Images', 'Documents', 'Installers', 'Music', 'Archives', 'Code', 'Misc'. \
+             Return ONLY a JSON object with this structure: {{ 'files': [ {{ 'filename': 'name', 'category': 'folder' }} ] }}",
+            self.file_list
+        )
+    }
+
+    pub fn build_subcategory_prompt(
+        filename: &str,
+        parent_category: &str,
+        content: &str,
+    ) -> String {
+        format!(
+            "I have a file named '{}' inside the '{}' folder. Here is the first 1000 characters of content:\n---\n{}\n---\nBased on this, suggest a single short sub-folder name (e.g., 'Invoices', 'Notes', 'Config'). Return ONLY the name of the sub-folder. Do not use markdown or explanations.",
+            filename, parent_category, content
+        )
+    }
+}