@devangkumar/dvai 1.0.0 → 1.0.1

package/package.json

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

package/src/index.js

@@ -1,5 +1,6 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import { Command } from "commander";
+import chalk from "chalk";
 import ora from "ora";
 import inquirer from "inquirer";
 
@@ -12,14 +13,17 @@ const program = new Command();
 
 program
   .name("myai")
-  .description("AI Assistant")
-  .version("1.0.0");
+  .description("A stunning AI assistant in your terminal")
+  .version("1.0.1");
 
+/**
+ * Main AI function to get response
+ */
 async function getAIResponse(prompt, chatHistory = []) {
   const spinner = ora({
-    text: "",
-    spinner: "dots",
-    isSilent: true // Keep it quiet
+    text: chalk.cyan("AI is thinking..."),
+    color: "cyan",
+    spinner: "dots"
   }).start();
 
   try {
@@ -37,7 +41,7 @@ async function getAIResponse(prompt, chatHistory = []) {
     }
   } catch (error) {
     spinner.stop();
-    console.error("Error:", error.message);
+    console.error(chalk.red("\n✖ AI Error:"), error.message);
     return null;
   }
 }
@@ -46,15 +50,18 @@ async function getAIResponse(prompt, chatHistory = []) {
  * Handle Single Question
  */
 program
-  .argument("[prompt]", "Prompt to send to AI")
+  .argument("[prompt]", "The question you want to ask AI")
   .action(async (prompt) => {
     if (!prompt) {
+      displayWelcome();
       return;
     }
 
     const response = await getAIResponse(prompt);
     if (response) {
-      process.stdout.write(response + "\n");
+      console.log(chalk.blue.bold("\nAI Response:"));
+      console.log(chalk.white(response));
+      console.log("\n");
     }
   });
 
@@ -63,8 +70,14 @@ program
  */
 program
   .command("chat")
-  .description("Start chat session")
+  .description("Start an interactive 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) {
@@ -72,22 +85,36 @@ program
         {
           type: "input",
           name: "userInput",
-          message: ">",
+          message: chalk.green("? You:"),
+          validate: (input) => input.trim() !== "" || "Please enter a prompt."
         }
       ]);
 
-      const cmd = userInput.toLowerCase();
-      if (cmd === "exit" || cmd === "quit" || cmd === "bye") break;
+      if (userInput.toLowerCase() === "exit" || userInput.toLowerCase() === "quit") {
+        console.log(chalk.yellow("\nGoodbye! 👋\n"));
+        break;
+      }
 
       const response = await getAIResponse(userInput, chatHistory);
       
       if (response) {
-        process.stdout.write(response + "\n\n");
+        console.log(chalk.blue.bold("\nAI:"));
+        console.log(chalk.white(response));
+        console.log("\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();