@devangkumar/dvai 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devangkumar/dvai",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "A powerful AI assistant for your terminal",
   "type": "module",
   "main": "src/index.js",

package/src/index.js

@@ -1,6 +1,5 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import { Command } from "commander";
-import chalk from "chalk";
 import ora from "ora";
 import inquirer from "inquirer";
 
@@ -13,17 +12,14 @@ const program = new Command();
 
 program
   .name("myai")
-  .description("A stunning AI assistant in your terminal")
-  .version("1.0.1");
+  .description("AI Assistant")
+  .version("1.0.2");
 
-/**
- * Main AI function to get response
- */
 async function getAIResponse(prompt, chatHistory = []) {
   const spinner = ora({
-    text: chalk.cyan("AI is thinking..."),
-    color: "cyan",
-    spinner: "dots"
+    text: "",
+    spinner: "dots",
+    isSilent: true 
   }).start();
 
   try {
@@ -41,7 +37,7 @@ async function getAIResponse(prompt, chatHistory = []) {
     }
   } catch (error) {
     spinner.stop();
-    console.error(chalk.red("\n✖ AI Error:"), error.message);
+    console.error("Error:", error.message);
     return null;
   }
 }
@@ -50,18 +46,15 @@ async function getAIResponse(prompt, chatHistory = []) {
  * Handle Single Question
  */
 program
-  .argument("[prompt]", "The question you want to ask AI")
+  .argument("[prompt]", "Prompt to send to AI")
   .action(async (prompt) => {
     if (!prompt) {
-      displayWelcome();
       return;
     }
 
     const response = await getAIResponse(prompt);
     if (response) {
-      console.log(chalk.blue.bold("\nAI Response:"));
-      console.log(chalk.white(response));
-      console.log("\n");
+      process.stdout.write(response + "\n");
     }
   });
 
@@ -70,14 +63,8 @@ program
  */
 program
   .command("chat")
-  .description("Start an interactive chat session")
+  .description("Start chat session")
   .action(async () => {
-    console.clear();
-    console.log(chalk.cyan.bold("╔════════════════════════════════════════════╗"));
-    console.log(chalk.cyan.bold("║          WELCOME TO MYAI TERMINAL          ║"));
-    console.log(chalk.cyan.bold("╚════════════════════════════════════════════╝"));
-    console.log(chalk.dim("Type 'exit' or 'quit' to end the session.\n"));
-
     let chatHistory = [];
     
     while (true) {
@@ -85,36 +72,22 @@ program
         {
           type: "input",
           name: "userInput",
-          message: chalk.green("? You:"),
-          validate: (input) => input.trim() !== "" || "Please enter a prompt."
+          message: ">",
         }
       ]);
 
-      if (userInput.toLowerCase() === "exit" || userInput.toLowerCase() === "quit") {
-        console.log(chalk.yellow("\nGoodbye! 👋\n"));
-        break;
-      }
+      const cmd = userInput.toLowerCase();
+      if (cmd === "exit" || cmd === "quit" || cmd === "bye") break;
 
       const response = await getAIResponse(userInput, chatHistory);
       
       if (response) {
-        console.log(chalk.blue.bold("\nAI:"));
-        console.log(chalk.white(response));
-        console.log("\n");
+        process.stdout.write(response + "\n\n");
         
-        // Update history
         chatHistory.push({ role: "user", parts: [{ text: userInput }] });
         chatHistory.push({ role: "model", parts: [{ text: response }] });
       }
     }
   });
 
-function displayWelcome() {
-    console.log(chalk.cyan.bold("\n✨ MyAI Terminal Assistant ✨"));
-    console.log(chalk.white("Usage:"));
-    console.log(chalk.dim("  myai \"your question\"  ") + chalk.white("- Ask a quick question"));
-    console.log(chalk.dim("  myai chat            ") + chalk.white("- Start interactive session"));
-    console.log(chalk.dim("  myai --help          ") + chalk.white("- Show help\n"));
-}
-
 program.parse();