prituscode 2.0.1 → 2.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prituscode",
-  "version": "2.0.1",
+  "version": "2.0.3",
   "description": "PRITUS CODE - Production-ready terminal AI coding agent CLI",
   "main": "prituscode.js",
   "bin": {

package/prituscode.js

@@ -38,8 +38,9 @@ loadEnv()
 const VERSION = '1.0.0'
 const DEFAULT_API_URL = process.env.PRITUS_API_URL || 'https://pritusai.netlify.app'
 const DEFAULT_MODEL = process.env.PRITUS_MODEL || 'pro'
-const AUTH_DIR = path.join(os.homedir(), '.prituscode')
-const AUTH_FILE = path.join(AUTH_DIR, 'auth.json')
+const CONFIG_DIR = path.join(os.homedir(), '.prituscode')
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json')
+const AUTH_FILE = path.join(CONFIG_DIR, 'auth.json')
 
 const MODELS = {
   1: { id: 'flash', label: 'Flash', desc: 'Fast & lightweight' },
@@ -48,6 +49,13 @@ const MODELS = {
   4: { id: 'ultra', label: 'Ultra Code', desc: 'Elite code generation' }
 }
 
+const THEMES = {
+  1: { id: 'dark', label: 'Dark', desc: 'Vibrant dark mode (Default)' },
+  2: { id: 'light', label: 'Light', desc: 'Clean light mode' },
+  3: { id: 'daltonized', label: 'Colorblind Friendly', desc: 'High contrast blue & yellow' },
+  4: { id: 'monochrome', label: 'Monochrome', desc: 'Sleek black & white' }
+}
+
 const AGENT_SYSTEM = `You are Pritus Code, an AI coding agent in a terminal. You CREATE and MODIFY real files.
 When asked to build something, respond in EXACT format:
 THINKING: <one sentence>
@@ -57,11 +65,33 @@ CREATE_FILE: <filename>
 \`\`\`
 `
 
-// --- AUTHENTICATION STORAGE HELPERS ---
+// --- CONFIG & AUTH STORAGE HELPERS ---
+function loadConfig() {
+  try {
+    if (fs.existsSync(CONFIG_FILE)) {
+      return JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'))
+    }
+  } catch (e) {
+    // ignore
+  }
+  return { theme: 'dark', model: DEFAULT_MODEL }
+}
+
+function saveConfig(cfg) {
+  try {
+    if (!fs.existsSync(CONFIG_DIR)) {
+      fs.mkdirSync(CONFIG_DIR, { recursive: true })
+    }
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(cfg, null, 2), 'utf8')
+  } catch (e) {
+    // ignore
+  }
+}
+
 function saveAuthSession(sessionData) {
   try {
-    if (!fs.existsSync(AUTH_DIR)) {
-      fs.mkdirSync(AUTH_DIR, { recursive: true })
+    if (!fs.existsSync(CONFIG_DIR)) {
+      fs.mkdirSync(CONFIG_DIR, { recursive: true })
     }
     fs.writeFileSync(AUTH_FILE, JSON.stringify(sessionData, null, 2), 'utf8')
     return true
@@ -104,7 +134,7 @@ function openBrowser(url) {
   }
 }
 
-// --- ANSI & FORMATTING HELPERS ---
+// --- ANSI & THEME HELPERS ---
 function stripAnsi(str) {
   return str.replace(/\x1B\[[0-9;]{0,}[A-Za-z]/g, '')
 }
@@ -117,20 +147,25 @@ function dim(str) {
   return '\x1b[2m' + str + '\x1b[22m'
 }
 
-function cyan(str) {
-  return '\x1b[36m' + str + '\x1b[39m'
-}
-
-function green(str) {
-  return '\x1b[32m' + str + '\x1b[39m'
+function colorPrimary(themeId, str) {
+  if (themeId === 'light') return '\x1b[34m' + str + '\x1b[39m'
+  if (themeId === 'daltonized') return '\x1b[34m' + str + '\x1b[39m'
+  if (themeId === 'monochrome') return '\x1b[37m' + str + '\x1b[39m'
+  return '\x1b[36m' + str + '\x1b[39m' // dark
 }
 
-function magenta(str) {
-  return '\x1b[35m' + str + '\x1b[39m'
+function colorSecondary(themeId, str) {
+  if (themeId === 'light') return '\x1b[35m' + str + '\x1b[39m'
+  if (themeId === 'daltonized') return '\x1b[33m' + str + '\x1b[39m'
+  if (themeId === 'monochrome') return '\x1b[90m' + str + '\x1b[39m'
+  return '\x1b[95m' + str + '\x1b[39m' // dark
 }
 
-function brightMagenta(str) {
-  return '\x1b[95m' + str + '\x1b[39m'
+function colorSuccess(themeId, str) {
+  if (themeId === 'light') return '\x1b[32m' + str + '\x1b[39m'
+  if (themeId === 'daltonized') return '\x1b[33m' + str + '\x1b[39m'
+  if (themeId === 'monochrome') return '\x1b[37m' + str + '\x1b[39m'
+  return '\x1b[32m' + str + '\x1b[39m' // dark
 }
 
 function ansi256(code, str) {
@@ -145,139 +180,28 @@ function formatPath(p) {
   return p
 }
 
-// --- LOGO & ASCII ART GENERATORS ---
-function getDiamondLogo() {
-  const lines = [
-    ansi256(39, '██'),
-    '  ' + ansi256(75, '██'),
-    '    ' + ansi256(99, '██'),
-    '    ' + ansi256(141, '██'),
-    '  ' + ansi256(177, '██'),
-    ansi256(207, '██'),
-    ansi256(207, '████████')
-  ]
-  return lines
-}
-
-const LETTERS = {
-  P: [
-    '██████╗ ',
-    '██╔══██╗',
-    '██████╔╝',
-    '██╔═══╝ ',
-    '██║     ',
-    '╚═╝     '
-  ],
-  R: [
-    '██████╗ ',
-    '██╔══██╗',
-    '██████╔╝',
-    '██╔██╗  ',
-    '██║╚██╗ ',
-    '╚═╝ ╚═╝ '
-  ],
-  I: [
-    '██╗',
-    '██║',
-    '██║',
-    '██║',
-    '██║',
-    '╚═╝'
-  ],
-  T: [
-    '████████╗',
-    '╚══██╔══╝',
-    '   ██║   ',
-    '   ██║   ',
-    '   ██║   ',
-    '   ╚═╝   '
-  ],
-  U: [
-    '██╗   ██╗',
-    '██║   ██║',
-    '██║   ██║',
-    '██║   ██║',
-    '╚██████╔╝',
-    ' ╚═════╝ '
-  ],
-  S: [
-    '███████╗',
-    '██╔════╝',
-    '███████╗',
-    '╚════██║',
-    '███████║',
-    '╚══════╝'
-  ],
-  C: [
-    '███████╗',
-    '██╔════╝',
-    '██║     ',
-    '██║     ',
-    '╚██████╗',
-    ' ╚═════╝'
-  ],
-  O: [
-    ' █████╗ ',
-    '██╔══██╗',
-    '██║  ██║',
-    '██║  ██║',
-    '╚██████╔╝',
-    ' ╚═════╝ '
-  ],
-  D: [
-    '██████╗ ',
-    '██╔══██╗',
-    '██║  ██║',
-    '██║  ██║',
-    '╚██████╗',
-    ' ╚═════╝ '
-  ],
-  E: [
-    '███████╗',
-    '██╔════╝',
-    '███████╗',
-    '██╔════╝',
-    '███████╝',
-    '        '
-  ]
-}
-
-function getBigBanner() {
-  const rows = []
-  for (let r = 0; r < 6; r++) {
-    const pritusLine = LETTERS.P[r] + ' ' + LETTERS.R[r] + ' ' + LETTERS.I[r] + ' ' + LETTERS.T[r] + ' ' + LETTERS.U[r] + ' ' + LETTERS.S[r]
-    const codeLine = LETTERS.C[r] + ' ' + LETTERS.O[r] + ' ' + LETTERS.D[r] + ' ' + LETTERS.E[r]
-    rows.push(pritusLine + '   ' + codeLine)
-  }
-  return rows
-}
-
-function renderHeader() {
-  const diamond = getDiamondLogo()
-  const banner = getBigBanner()
-
-  console.log('')
-  // Render Diamond Logo cleanly
-  for (let i = 0; i < diamond.length; i++) {
-    console.log(diamond[i])
-  }
+// --- LOGO & HEADER RENDERER ---
+function renderHeader(themeId) {
   console.log('')
-  // Render ASCII Banner stacked cleanly on its own lines
-  for (let r = 0; r < banner.length; r++) {
-    console.log(cyan(banner[r]))
-  }
-  console.log(brightMagenta(bold('PRITUS CODE v' + VERSION)))
+  // Sleek glowing diamond crystal icon + clean typography
+  const row1 = ansi256(39, '   ▲   ') + '  ' + bold(colorPrimary(themeId, 'P R I T U S   C O D E'))
+  const row2 = ansi256(99, ' ◄ ◈ ► ') + '  ' + dim('Terminal AI Coding Agent ') + colorSecondary(themeId, 'v' + VERSION)
+  const row3 = ansi256(207, '   ▼   ') + '  ' + ansi256(207, '━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━')
+
+  console.log(row1)
+  console.log(row2)
+  console.log(row3)
   console.log('')
 }
 
-function renderWelcomeBox(cwd, authSession) {
+function renderWelcomeBox(cwd, authSession, themeId) {
   const displayCwd = formatPath(cwd)
-  const userText = authSession && authSession.email ? 'Logged in as: ' + authSession.email : 'Not logged in (type /login to authenticate with Google)'
+  const userText = authSession && authSession.email ? 'Logged in as: ' + authSession.email : 'Not logged in'
 
   const lines = [
     'Welcome to Pritus Code!',
     userText,
-    '/help for help, /status for your current setup',
+    '/help for help, /status for config, /theme to change colors',
     'cwd: ' + displayCwd
   ]
 
@@ -292,39 +216,39 @@ function renderWelcomeBox(cwd, authSession) {
   const topBorder = '┌' + '─'.repeat(innerWidth) + '┐'
   const bottomBorder = '└' + '─'.repeat(innerWidth) + '┘'
 
-  console.log(cyan(topBorder))
+  console.log(colorPrimary(themeId, topBorder))
   
   const pad1 = ' '.repeat(innerWidth - lines[0].length - 2)
-  console.log(cyan('│ ') + bold(lines[0]) + pad1 + cyan(' │'))
+  console.log(colorPrimary(themeId, '│ ') + bold(lines[0]) + pad1 + colorPrimary(themeId, ' │'))
 
   const pad2 = ' '.repeat(innerWidth - lines[1].length - 2)
-  const formattedUserLine = authSession && authSession.email ? green(lines[1]) : dim(lines[1])
-  console.log(cyan('│ ') + formattedUserLine + pad2 + cyan(' │'))
+  const formattedUserLine = authSession && authSession.email ? colorSuccess(themeId, lines[1]) : dim(lines[1])
+  console.log(colorPrimary(themeId, '│ ') + formattedUserLine + pad2 + colorPrimary(themeId, ' │'))
 
   const pad3 = ' '.repeat(innerWidth - lines[2].length - 2)
-  console.log(cyan('│ ') + dim(lines[2]) + pad3 + cyan(' │'))
+  console.log(colorPrimary(themeId, '│ ') + dim(lines[2]) + pad3 + colorPrimary(themeId, ' │'))
 
   const pad4 = ' '.repeat(innerWidth - lines[3].length - 2)
-  console.log(cyan('│ ') + dim(lines[3]) + pad4 + cyan(' │'))
+  console.log(colorPrimary(themeId, '│ ') + dim(lines[3]) + pad4 + colorPrimary(themeId, ' │'))
 
-  console.log(cyan(bottomBorder))
+  console.log(colorPrimary(themeId, bottomBorder))
   console.log('')
 }
 
-function renderTips() {
+function renderTips(themeId) {
   console.log(bold('Tips for getting started:'))
   console.log('1. Ask Pritus to build something — it will create actual files in your project')
   console.log('2. Use @filename to include a file contents in your message')
   console.log('3. Be as specific as you would with another engineer for the best results')
-  console.log(green('✓') + ' Run /init to create a PRITUS.md with project instructions')
+  console.log(colorSuccess(themeId, '✓') + ' Run /init to create a PRITUS.md with project instructions')
   console.log('')
 }
 
-function renderStatusBar(currentModelId, cwd, authSession) {
+function renderStatusBar(currentModelId, cwd, authSession, themeId) {
   const displayCwd = formatPath(cwd)
   const selectedObj = getModelInfo(currentModelId)
   const authStatus = authSession && authSession.email ? authSession.email : 'guest'
-  const barText = 'workspace (' + displayCwd + ')  /model ' + selectedObj.id + '  user: ' + authStatus + '  ? for shortcuts'
+  const barText = 'workspace (' + displayCwd + ')  /model ' + selectedObj.id + '  theme: ' + themeId + '  user: ' + authStatus + '  ? for help'
   console.log(dim(barText))
   console.log('')
 }
@@ -339,11 +263,23 @@ function getModelInfo(keyOrId) {
   return MODELS[2]
 }
 
+function getThemeInfo(keyOrId) {
+  const str = String(keyOrId).toLowerCase()
+  for (const k in THEMES) {
+    if (String(k) === str || THEMES[k].id === str) {
+      return THEMES[k]
+    }
+  }
+  return THEMES[1]
+}
+
 // --- GOOGLE OAUTH LOGIN SERVER ---
+const DEFAULT_GOOGLE_AUTH_URL = 'https://accounts.google.com/v3/signin/accountchooser?client_id=428748183111-03pulj6olh5cp74t3f6s8p6cg5ksne0a.apps.googleusercontent.com&context_uri=https%3A%2F%2Fpritusai.netlify.app&redirect_uri=https%3A%2F%2Fpritus-ai-2c54c.firebaseapp.com%2F__%2Fauth%2Fhandler&response_type=code&scope=openid+https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fuserinfo.email+profile'
+
 function startGoogleOAuthFlow(apiUrl, onComplete) {
   const port = 8585
   const redirectUri = 'http://localhost:' + port + '/callback'
-  const loginUrl = apiUrl + '/api/auth/google?redirect_uri=' + encodeURIComponent(redirectUri)
+  const loginUrl = process.env.PRITUS_AUTH_URL || DEFAULT_GOOGLE_AUTH_URL
 
   const server = http.createServer((req, res) => {
     const reqUrl = new URL(req.url, 'http://localhost:' + port)
@@ -409,8 +345,104 @@ function startGoogleOAuthFlow(apiUrl, onComplete) {
   })
 }
 
+function enforceCompulsoryLogin(state, callback) {
+  if (state.authSession && state.authSession.email) {
+    return callback(state.authSession)
+  }
+
+  console.log(bold('\x1b[33m[Authentication Required]\x1b[39m'))
+  console.log('Pritus Code requires a free Google sign-in to continue.')
+  process.stdout.write('Press Enter to sign in with Google in your browser... ')
+
+  if (process.stdin.isTTY) {
+    process.stdin.setRawMode(true)
+    process.stdin.resume()
+    process.stdin.once('data', () => {
+      process.stdin.setRawMode(false)
+      process.stdin.pause()
+      console.log('')
+      startGoogleOAuthFlow(state.apiUrl, (err, session) => {
+        if (err || !session) {
+          console.log('\x1b[31mAuthentication failed. Pritus Code requires Google Login to proceed.\x1b[39m')
+          process.exit(1)
+        }
+        state.authSession = session
+        console.log(colorSuccess(state.currentTheme, '✓') + ' Signed in as ' + session.email + '\n')
+        callback(session)
+      })
+    })
+  } else {
+    const rlTemp = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout
+    })
+    rlTemp.question('', () => {
+      rlTemp.close()
+      startGoogleOAuthFlow(state.apiUrl, (err, session) => {
+        if (err || !session) {
+          console.log('\x1b[31mAuthentication failed. Pritus Code requires Google Login to proceed.\x1b[39m')
+          process.exit(1)
+        }
+        state.authSession = session
+        console.log(colorSuccess(state.currentTheme, '✓') + ' Signed in as ' + session.email + '\n')
+        callback(session)
+      })
+    })
+  }
+}
+
+// --- CLAUDE CODE STYLE THEME SELECTOR ---
+function promptThemeSelection(currentThemeId, callback) {
+  console.log('Select a color theme:')
+  for (const k in THEMES) {
+    const t = THEMES[k]
+    const isSelected = t.id === currentThemeId
+    const pointer = isSelected ? '▸ ' : '  '
+    const num = k + '  '
+    console.log(pointer + num + bold(t.label) + ' — ' + t.desc)
+  }
+  
+  const defaultThemeObj = getThemeInfo(currentThemeId)
+  process.stdout.write('Press 1-4, or Enter for ' + defaultThemeObj.label + ': ')
+
+  if (process.stdin.isTTY) {
+    process.stdin.setRawMode(true)
+    process.stdin.resume()
+    process.stdin.once('data', (data) => {
+      process.stdin.setRawMode(false)
+      process.stdin.pause()
+      const char = data.toString('utf8').trim()
+      console.log(char)
+
+      let choiceObj = defaultThemeObj
+      if (THEMES[char]) {
+        choiceObj = THEMES[char]
+      }
+      console.log(colorSuccess(choiceObj.id, '✓') + ' Applied ' + choiceObj.label + ' theme')
+      console.log('')
+      callback(choiceObj.id)
+    })
+  } else {
+    const rlTemp = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout
+    })
+    rlTemp.question('', (ans) => {
+      rlTemp.close()
+      const trimmed = ans.trim()
+      let choiceObj = defaultThemeObj
+      if (THEMES[trimmed]) {
+        choiceObj = THEMES[trimmed]
+      }
+      console.log(colorSuccess(choiceObj.id, '✓') + ' Applied ' + choiceObj.label + ' theme')
+      console.log('')
+      callback(choiceObj.id)
+    })
+  }
+}
+
 // --- MODEL SELECTOR ---
-function promptModelSelection(currentModelId, callback) {
+function promptModelSelection(currentModelId, themeId, callback) {
   console.log('Select a model:')
   for (const k in MODELS) {
     const m = MODELS[k]
@@ -436,7 +468,7 @@ function promptModelSelection(currentModelId, callback) {
       if (MODELS[char]) {
         choiceObj = MODELS[char]
       }
-      console.log(green('✓') + ' Switched to ' + choiceObj.label)
+      console.log(colorSuccess(themeId, '✓') + ' Switched to ' + choiceObj.label)
       console.log('')
       callback(choiceObj.id)
     })
@@ -452,7 +484,7 @@ function promptModelSelection(currentModelId, callback) {
       if (MODELS[trimmed]) {
         choiceObj = MODELS[trimmed]
       }
-      console.log(green('✓') + ' Switched to ' + choiceObj.label)
+      console.log(colorSuccess(themeId, '✓') + ' Switched to ' + choiceObj.label)
       console.log('')
       callback(choiceObj.id)
     })
@@ -460,7 +492,7 @@ function promptModelSelection(currentModelId, callback) {
 }
 
 // --- FILE PARSER & AGENT OPERATIONS ---
-function processFileOperations(aiResponse) {
+function processFileOperations(aiResponse, themeId) {
   let modifiedAny = false
   const createPattern = /CREATE_FILE:\s{0,}([^\n]+)\n```[a-z]{0,}\n([\s\S]+?)```/gi
   const modifyPattern = /MODIFY_FILE:\s{0,}([^\n]+)\n```[a-z]{0,}\n([\s\S]+?)```/gi
@@ -474,7 +506,7 @@ function processFileOperations(aiResponse) {
       fs.mkdirSync(path.dirname(targetPath), { recursive: true })
       fs.writeFileSync(targetPath, content, 'utf8')
       const bytes = Buffer.byteLength(content, 'utf8')
-      console.log(green('✓ Created ') + fileName + ' (' + bytes + ' bytes)')
+      console.log(colorSuccess(themeId, '✓ Created ') + fileName + ' (' + bytes + ' bytes)')
       modifiedAny = true
     } catch (err) {
       console.log('\x1b[31mError creating file ' + fileName + ': ' + err.message + '\x1b[39m')
@@ -489,7 +521,7 @@ function processFileOperations(aiResponse) {
       fs.mkdirSync(path.dirname(targetPath), { recursive: true })
       fs.writeFileSync(targetPath, content, 'utf8')
       const bytes = Buffer.byteLength(content, 'utf8')
-      console.log(green('✓ Modified ') + fileName + ' (' + bytes + ' bytes)')
+      console.log(colorSuccess(themeId, '✓ Modified ') + fileName + ' (' + bytes + ' bytes)')
       modifiedAny = true
     } catch (err) {
       console.log('\x1b[31mError modifying file ' + fileName + ': ' + err.message + '\x1b[39m')
@@ -550,12 +582,16 @@ function sendChatRequest(apiUrl, modelId, userPrompt, messagesHistory, authSessi
     return callback(new Error('Invalid URL'))
   }
 
+  const allMessages = (messagesHistory || []).concat([
+    { role: 'user', content: expandedPrompt }
+  ])
+
   const httpModule = parsedUrl.protocol === 'https:' ? require('https') : require('http')
   const payload = JSON.stringify({
     prompt: expandedPrompt,
     model: modelId,
     system: AGENT_SYSTEM,
-    messages: messagesHistory
+    messages: allMessages
   })
 
   const reqHeaders = {
@@ -620,6 +656,7 @@ function handleSlashCommand(input, state, rl) {
     case '?':
       console.log('')
       console.log(bold('PRITUS CODE Commands & Shortcuts:'))
+      console.log('  /theme        Change color theme (Dark, Light, Colorblind, Monochrome)')
       console.log('  /login        Sign in with Google authentication')
       console.log('  /logout       Sign out of current account')
       console.log('  /whoami       Display active user profile')
@@ -634,11 +671,28 @@ function handleSlashCommand(input, state, rl) {
       rl.prompt()
       return true
 
+    case '/theme':
+      if (arg) {
+        const info = getThemeInfo(arg)
+        state.currentTheme = info.id
+        saveConfig({ theme: state.currentTheme, model: state.currentModel })
+        console.log(colorSuccess(state.currentTheme, '✓') + ' Applied ' + info.label + ' theme')
+        console.log('')
+        rl.prompt()
+      } else {
+        promptThemeSelection(state.currentTheme, (newTheme) => {
+          state.currentTheme = newTheme
+          saveConfig({ theme: state.currentTheme, model: state.currentModel })
+          rl.prompt()
+        })
+      }
+      return true
+
     case '/login':
       startGoogleOAuthFlow(state.apiUrl, (err, session) => {
         if (!err && session) {
           state.authSession = session
-          console.log(green('✓ Logged in as ') + session.email + ' (' + session.name + ')')
+          console.log(colorSuccess(state.currentTheme, '✓ Logged in as ') + session.email + ' (' + session.name + ')')
         }
         console.log('')
         rl.prompt()
@@ -648,7 +702,7 @@ function handleSlashCommand(input, state, rl) {
     case '/logout':
       clearAuthSession()
       state.authSession = null
-      console.log(green('✓ Successfully logged out.'))
+      console.log(colorSuccess(state.currentTheme, '✓ Successfully logged out.'))
       console.log('')
       rl.prompt()
       return true
@@ -670,6 +724,7 @@ function handleSlashCommand(input, state, rl) {
       console.log('')
       console.log(bold('Current Setup:'))
       console.log('  Version: ' + VERSION)
+      console.log('  Theme:   ' + getThemeInfo(state.currentTheme).label + ' (' + state.currentTheme + ')')
       console.log('  Model:   ' + getModelInfo(state.currentModel).label + ' (' + state.currentModel + ')')
       console.log('  User:    ' + (state.authSession ? state.authSession.email : 'Not logged in'))
       console.log('  API URL: ' + state.apiUrl)
@@ -692,7 +747,7 @@ Describe your project and architectural guidelines here for Pritus Code.
 `
         try {
           fs.writeFileSync(docPath, docContent, 'utf8')
-          console.log(green('✓ Created PRITUS.md with project instructions'))
+          console.log(colorSuccess(state.currentTheme, '✓ Created PRITUS.md with project instructions'))
         } catch (err) {
           console.log('\x1b[31mError creating PRITUS.md: ' + err.message + '\x1b[39m')
         }
@@ -705,12 +760,14 @@ Describe your project and architectural guidelines here for Pritus Code.
       if (arg) {
         const info = getModelInfo(arg)
         state.currentModel = info.id
-        console.log(green('✓ Switched to ') + info.label)
+        saveConfig({ theme: state.currentTheme, model: state.currentModel })
+        console.log(colorSuccess(state.currentTheme, '✓ Switched to ') + info.label)
         console.log('')
         rl.prompt()
       } else {
-        promptModelSelection(state.currentModel, (newModel) => {
+        promptModelSelection(state.currentModel, state.currentTheme, (newModel) => {
           state.currentModel = newModel
+          saveConfig({ theme: state.currentTheme, model: state.currentModel })
           rl.prompt()
         })
       }
@@ -718,8 +775,8 @@ Describe your project and architectural guidelines here for Pritus Code.
 
     case '/clear':
       console.clear()
-      renderHeader()
-      renderStatusBar(state.currentModel, process.cwd(), state.authSession)
+      renderHeader(state.currentTheme)
+      renderStatusBar(state.currentModel, process.cwd(), state.authSession, state.currentTheme)
       rl.prompt()
       return true
 
@@ -779,7 +836,7 @@ function startRepl(state) {
       console.log(responseText)
       console.log('')
 
-      const opsExecuted = processFileOperations(responseText)
+      const opsExecuted = processFileOperations(responseText, state.currentTheme)
       if (opsExecuted) {
         console.log('')
       }
@@ -800,10 +857,12 @@ function startRepl(state) {
 // --- ENTRY POINT ---
 function main() {
   const args = process.argv.slice(2)
+  const userConfig = loadConfig()
 
   const state = {
     apiUrl: DEFAULT_API_URL,
-    currentModel: DEFAULT_MODEL,
+    currentModel: userConfig.model || DEFAULT_MODEL,
+    currentTheme: userConfig.theme || 'dark',
     authSession: loadAuthSession()
   }
 
@@ -830,7 +889,7 @@ function main() {
       if (err) {
         process.exit(1)
       }
-      console.log(green('✓ Logged in as ') + session.email + ' (' + session.name + ')')
+      console.log(colorSuccess(state.currentTheme, '✓ Logged in as ') + session.email + ' (' + session.name + ')')
       process.exit(0)
     })
     return
@@ -865,21 +924,26 @@ function main() {
         process.exit(1)
       }
       console.log(responseText)
-      processFileOperations(responseText)
+      processFileOperations(responseText, state.currentTheme)
       process.exit(0)
     })
     return
   }
 
   // Interactive mode startup
-  renderHeader()
-  renderWelcomeBox(process.cwd(), state.authSession)
-  renderTips()
-
-  promptModelSelection(state.currentModel, (selectedModel) => {
-    state.currentModel = selectedModel
-    renderStatusBar(state.currentModel, process.cwd(), state.authSession)
-    startRepl(state)
+  renderHeader(state.currentTheme)
+
+  // Compulsory Login Enforcement
+  enforceCompulsoryLogin(state, () => {
+    renderWelcomeBox(process.cwd(), state.authSession, state.currentTheme)
+    renderTips(state.currentTheme)
+
+    promptModelSelection(state.currentModel, state.currentTheme, (selectedModel) => {
+      state.currentModel = selectedModel
+      saveConfig({ theme: state.currentTheme, model: state.currentModel })
+      renderStatusBar(state.currentModel, process.cwd(), state.authSession, state.currentTheme)
+      startRepl(state)
+    })
   })
 }