thelapyae 0.1.0 → 0.1.1

package/bin/config.js

@@ -0,0 +1,41 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const readline = require('readline');
+
+const CONFIG_PATH = path.join(os.homedir(), '.thelapyaerc');
+
+function loadConfig() {
+  if (fs.existsSync(CONFIG_PATH)) {
+    try {
+      return JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf8'));
+    } catch (e) {
+      return {};
+    }
+  }
+  return {};
+}
+
+function saveConfig(config) {
+  fs.writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2));
+}
+
+async function configure() {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+
+  return new Promise((resolve) => {
+    rl.question('🔑 Enter your Gemini API Key: ', (key) => {
+      const config = loadConfig();
+      config.GEMINI_API_KEY = key.trim();
+      saveConfig(config);
+      console.log('✅ API Key saved successfully!');
+      rl.close();
+      resolve();
+    });
+  });
+}
+
+module.exports = { loadConfig, configure };

package/bin/thelapyae.js

@@ -1,13 +1,176 @@
 #!/usr/bin/env node
 
-console.log(`
-👋 Hi, I'm La Pyae (@thelapyae)
+const { GoogleGenerativeAI } = require("@google/generative-ai");
+const { loadConfig, configure } = require("./config");
+const models = require("../data/mental-models.json");
+
+const args = process.argv.slice(2);
+const input = args.join(" ");
+
+function randomModel() {
+  const model = models[Math.floor(Math.random() * models.length)];
+  printModel(model);
+}
+
+function searchModels(query, limit = 1) {
+  const terms = query.toLowerCase().split(/\s+/).filter(w => w.length > 2);
+  
+  if (terms.length === 0) return [];
+
+  const scored = models.map(m => {
+    let score = 0;
+    const textTitle = m.title.toLowerCase();
+    const textProblem = (m.problem || "").toLowerCase();
+    const textExplanation = (m.explanation || "").toLowerCase();
+
+    terms.forEach(term => {
+      if (textTitle.includes(term)) score += 10;
+      if (textProblem.includes(term)) score += 5;
+      if (textExplanation.includes(term)) score += 1;
+    });
+
+    return { model: m, score };
+  });
+
+  scored.sort((a, b) => b.score - a.score);
+  return scored.filter(s => s.score > 0).slice(0, limit).map(s => s.model);
+}
+
+function cleanText(text) {
+  if (!text) return "";
+  return text.replace(/\*\*/g, "").replace(/\*/g, "").replace(/__/g, "").replace(/_/g, "");
+}
+
+function printModel(m) {
+  console.log(`
+🧠 ${cleanText(m.title)}
+Category: ${cleanText(m.category)}
+
+Problem:
+${cleanText(m.problem)}
+
+Question:
+${cleanText(m.question)}
+
+Explanation:
+${cleanText(m.explanation)}
+
+Example:
+${cleanText(m.example)}
+
+Action:
+👉 ${cleanText(m.action)}
+`);
+}
+
+async function consultGemini(query) {
+  let config = loadConfig();
+  
+  if (!config.GEMINI_API_KEY) {
+    console.log("ℹ️  No Gemini API Key found.");
+    const readline = require('readline');
+    const rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout
+    });
+
+    const answer = await new Promise(resolve => {
+      rl.question('👉 Do you want to set it up now to get AI consulting? (y/N) ', resolve);
+    });
+    rl.close();
+
+    if (answer.toLowerCase().startsWith('y')) {
+      await configure();
+      config = loadConfig(); // Reload config
+    } else {
+      console.log("👉 Falling back to local search (offline mode).\n");
+      const results = searchModels(query);
+      if (results.length > 0) {
+        console.log(`🔍 Finding suitable mental model for: "${query}"...\n`);
+        printModel(results[0]);
+      } else {
+        console.log(`❌ No mental model found for: "${query}"`);
+      }
+      return;
+    }
+  }
+
+  const matches = searchModels(query, 5); // Get top 5 candidates
+  
+  if (matches.length === 0) {
+     console.log(`❌ No relevant mental models found for: "${query}" locally to send to AI.`);
+     return;
+  }
+
+  console.log(`🤔 Thinking...`);
+
+  const genAI = new GoogleGenerativeAI(config.GEMINI_API_KEY);
+  const model = genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
+
+  const prompt = `
+  You are an expert consultant using specific mental models to help people.
+  User Problem: "${query}"
 
-🧠 Mental Models
-🗿 Stoicism
-⚙️ Clear thinking for creators & builders
+  Here are the most relevant mental models from your database:
+  ${JSON.stringify(matches.map(m => ({ title: m.title, explanation: m.explanation, action: m.action })))}
+
+  Task:
+  1. Analyze the user's problem.
+  2. Select the top 3 most relevant mental models from the list to offer a multi-perspective solution.
+  3. Synthesize a comprehensive answer. Reference the models by name (e.g., "Perspective 1: [Model Name]").
+  4. Provide a "Consolidated Action Plan" with 3-5 concise, actionable bullet points.
+
+  Format:
+  - Start with a direct, empathetic opening.
+  - Use emojis for structure.
+  - Do NOT use markdown formatting (no bolding **, no headers #). Use uppercase for emphasis or bullet points like "•".
+  - Keep the action steps short and punchy.
+  `;
+
+  try {
+    const result = await model.generateContent(prompt);
+    const response = await result.response;
+    console.log(response.text());
+  } catch (error) {
+    console.error("❌ AI Error:", error.message);
+    if (matches.length > 0) {
+      console.log("\n👇 Falling back to top local match:");
+      printModel(matches[0]);
+    }
+  }
+}
+
+async function main() {
+  if (!input) {
+    const figlet = require("figlet");
+    console.log(figlet.textSync("thelapyae", { horizontalLayout: "full" }));
+    console.log(`
+👋 Hi, I'm La Pyae (@thelapyae)
 
 Try:
   npx thelapyae random
-  npx thelapyae stoic
+  npx thelapyae list
+  npx thelapyae "I feel anxious"
+  npx thelapyae config  (Setup AI Key)
 `);
+    process.exit(0);
+  }
+
+  if (input === "config") {
+    await configure();
+  } else if (input === "random") {
+    randomModel();
+  } else if (input === "list") {
+    models.forEach(m => console.log(`- ${m.id} (${m.title})`));
+  } else {
+    // Try exact ID match first
+    const exactMatch = models.find(m => m.id === input);
+    if (exactMatch) {
+      printModel(exactMatch);
+    } else {
+      await consultGemini(input);
+    }
+  }
+}
+
+main();