shadowdocs 2.1.5 → 2.1.6

package/README.md

@@ -1,19 +1,19 @@
 # ShadowDocs
 ## Overview
-ShadowDocs is an AI-powered documentation generator for developer projects, designed to simplify the process of creating and maintaining high-quality documentation. The project aims to provide an efficient and effective solution for developers to generate documentation with ease. By leveraging AI technology, ShadowDocs automates the documentation process, saving developers time and effort.
+ShadowDocs is an AI-powered documentation generator designed to assist developers in creating high-quality project documentation. Its primary purpose is to streamline the documentation process, making it easier for developers to focus on their core work. By leveraging AI capabilities, ShadowDocs aims to reduce the time and effort required to generate and maintain project documentation.
 
 ## Version
-Current version: 2.0.0
+Current version: 2.1.5
 
 ## Description
-ShadowDocs is a module designed for developers who want to generate high-quality documentation for their projects without the hassle of manual writing. The project uses AI technology to analyze the codebase and generate documentation automatically. ShadowDocs is ideal for developers who want to focus on writing code rather than documentation. With its interactive and intuitive interface, ShadowDocs makes it easy for developers to generate documentation that is accurate, up-to-date, and easy to understand.
+ShadowDocs is a module designed for developer projects, providing an efficient way to generate documentation using AI. It is particularly useful for projects that require extensive documentation, as it can significantly reduce the workload associated with creating and updating documentation. With ShadowDocs, developers can ensure that their projects have accurate, up-to-date, and comprehensive documentation, which is essential for collaboration, maintenance, and user adoption.
 
 ## Features
 - AI-powered documentation generation
-- Automatic code analysis
-- Interactive documentation interface
-- Support for multiple programming languages
-- Customizable configuration options
+- Real-time documentation updates with `--watch` mode
+- Command-line interface (CLI) for easy usage
+- Type checking for ensuring code quality
+- Interactive mode for customized documentation generation
 
 ## Installation
 To install ShadowDocs, run the following command:
@@ -22,46 +22,55 @@ npm install shadowdocs
 ```
 
 ## Usage
-To use ShadowDocs, follow these steps:
+ShadowDocs can be used in various ways, depending on the specific requirements of your project. Here are a few examples:
+```bash
+# Build documentation
+npx shadowdocs build
+
+# Start documentation generation in watch mode
+npx shadowdocs dev
+
+# Start the CLI
+npx shadowdocs start
+```
+In your JavaScript or TypeScript files, you can also use ShadowDocs as a module:
 ```javascript
-// Import ShadowDocs
-const { generateDocs } = require('shadowdocs');
+const { generateDocumentation } = require('shadowdocs');
 
 // Generate documentation
-generateDocs({
-  // Project path
-  path: './my-project',
-  // Output path
-  output: './docs',
-});
+generateDocumentation('path/to/project');
 ```
 
 ## API Reference
-ShadowDocs exports the following functions:
-- `generateDocs(options)`: Generates documentation for a project.
-  - `options.path`: The path to the project.
-  - `options.output`: The output path for the generated documentation.
+ShadowDocs exports several functions and classes that can be used to customize and extend its functionality. Some of the key exports include:
+- `generateDocumentation`: Generates documentation for a given project
+- `configure`: Configures ShadowDocs with custom settings
+- `Analyzer`: A class for analyzing project code and generating documentation
 
 ## Configuration
-ShadowDocs provides the following configuration options:
-- `path`: The path to the project.
-- `output`: The output path for the generated documentation.
-- `language`: The programming language of the project.
+ShadowDocs provides several configuration options that can be used to customize its behavior. These options can be specified in a `config.ts` file or passed as command-line arguments. Some of the available options include:
+- `outputPath`: Specifies the output path for generated documentation
+- `template`: Specifies a custom template for generated documentation
+- `watch`: Enables or disables watch mode
 
 ## Development
-To set up the development environment, run the following commands:
+To set up a development environment for ShadowDocs, follow these steps:
 ```bash
+# Install dependencies
 npm install
+
+# Build and watch the code
 npm run dev
 ```
-This will start the development server and watch for changes to the code.
+This will start the development server, and you can begin making changes to the codebase.
 
 ## Testing
-To run tests, use the following command:
+To run tests for ShadowDocs, use the following command:
 ```bash
+# Run type checking
 npm run typecheck
 ```
-This will run the type checker and ensure that the code is correct.
+Note that ShadowDocs currently uses TypeScript for type checking, and there are no additional tests beyond type checking.
 
 ## License
 MIT License
@@ -88,13 +97,13 @@ SOFTWARE.
 
 ## Contributing
 Contribution Guidelines:
-- Fork the repository and create a feature branch.
-- Follow the existing code style and conventions.
-- Add tests if applicable to ensure that the code is correct.
-- Submit a pull request with a clear description of the changes.
+- Fork the ShadowDocs repository and create a new feature branch
+- Follow the existing code style and conventions
+- Add tests if applicable to ensure the quality of the code
+- Submit a pull request with a detailed description of the changes
 
 ## Security Policy
-If you discover a vulnerability in ShadowDocs, please report it to the maintainers immediately. You can report vulnerabilities by emailing the maintainers or by submitting an issue on the repository.
+If you discover a vulnerability in ShadowDocs, please report it to the maintainers via email or by submitting an issue on the GitHub repository. We take security seriously and will address vulnerabilities promptly.
 
 ## Code of Conduct
-We are committed to creating a welcoming and inclusive environment for all contributors. We expect all contributors to be respectful, inclusive, and professional. Harassment, discrimination, or any other form of unacceptable behavior will not be tolerated and will result in immediate suspension of privileges.
+We are committed to creating a welcoming and inclusive environment for all contributors. We expect all participants to be respectful, considerate, and professional. Harassment, discrimination, or other forms of unacceptable behavior will not be tolerated and may result in immediate suspension or removal from the project.

package/dist/services/ai.d.ts

@@ -1,4 +1,11 @@
 import { DocResult } from '../core/types.js';
+export interface GeneratedDocs {
+    content?: string;
+    readme: string;
+    contributing: string;
+    license: string;
+    security: string;
+}
 export declare function generateDocs(projectData: Record<string, unknown>, mode?: 'generate' | 'improve' | 'explain'): Promise<DocResult>;
 export declare function validateApiKey(apiKey: string): Promise<boolean>;
 //# sourceMappingURL=ai.d.ts.map

package/dist/services/ai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["../../src/services/ai.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AA6D7C,wBAAsB,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,GAAE,UAAU,GAAG,SAAS,GAAG,SAAsB,GAAG,OAAO,CAAC,SAAS,CAAC,CA8BlJ;AA0HD,wBAAsB,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CA2BrE"}
+{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["../../src/services/ai.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AA6D7C,MAAM,WAAW,aAAa;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,wBAAsB,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,GAAE,UAAU,GAAG,SAAS,GAAG,SAAsB,GAAG,OAAO,CAAC,SAAS,CAAC,CA+ClJ;AA0MD,wBAAsB,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CA2BrE"}

package/dist/services/ai.js

@@ -2,7 +2,7 @@ import axios from 'axios';
 import { loadApiKey } from '../core/config.js';
 const GROQ_API_URL = 'https://api.groq.com/openai/v1/chat/completions';
 const MODEL = 'llama-3.3-70b-versatile';
-const SYSTEM_PROMPT = `You are a senior open-source maintainer and technical documentation expert. Generate detailed, professional READMEs with proper sections, realistic content (no fake features), MIT license, contribution guidelines, and security policy.`;
+const SYSTEM_PROMPT = `You are a senior open-source maintainer and technical documentation expert with 15+ years of experience. You generate comprehensive, production-grade documentation that is practical, accurate, and developer-friendly. NEVER hallucinate features - only document what exists in the codebase.`;
 const MAX_RETRIES = 3;
 const BASE_DELAY = 1000;
 async function sleep(ms) {
@@ -49,117 +49,207 @@ export async function generateDocs(projectData, mode = 'generate') {
     if (!apiKey) {
         throw new Error('API key not configured. Run shadowdocs setup.');
     }
-    let userPrompt = '';
-    switch (mode) {
-        case 'generate':
-            userPrompt = buildGeneratePrompt(projectData);
-            break;
-        case 'improve':
-            userPrompt = buildImprovePrompt(projectData);
-            break;
-        case 'explain':
-            userPrompt = buildExplainPrompt(projectData);
-            break;
-        default:
-            userPrompt = buildGeneratePrompt(projectData);
-    }
-    const response = await withRetry(() => makeRequest(apiKey, userPrompt));
     if (mode === 'explain') {
+        const response = await withRetry(() => makeRequest(apiKey, buildExplainPrompt(projectData)));
         return { type: 'text', content: response };
     }
-    return { type: 'markdown', content: response };
+    if (mode === 'generate') {
+        const [readme, contributing, license, security] = await Promise.all([
+            withRetry(() => makeRequest(apiKey, buildReadmePrompt(projectData))),
+            withRetry(() => makeRequest(apiKey, buildContributingPrompt(projectData))),
+            withRetry(() => makeRequest(apiKey, buildLicensePrompt(projectData))),
+            withRetry(() => makeRequest(apiKey, buildSecurityPrompt(projectData)))
+        ]);
+        return {
+            type: 'markdown',
+            content: readme,
+            contributing,
+            license,
+            security
+        };
+    }
+    if (mode === 'improve') {
+        const [readme, contributing, license, security] = await Promise.all([
+            withRetry(() => makeRequest(apiKey, buildImproveReadmePrompt(projectData))),
+            withRetry(() => makeRequest(apiKey, buildImproveContributingPrompt(projectData))),
+            withRetry(() => makeRequest(apiKey, buildImproveLicensePrompt(projectData))),
+            withRetry(() => makeRequest(apiKey, buildImproveSecurityPrompt(projectData)))
+        ]);
+        return {
+            type: 'markdown',
+            content: readme,
+            contributing,
+            license,
+            security
+        };
+    }
+    throw new Error('Unknown mode');
+}
+function buildReadmePrompt(projectData) {
+    return `You are an expert technical writer. Generate a comprehensive, production-ready README.md for this project.
+
+Analyze the project structure and package.json thoroughly. Document ONLY what's actually present in the codebase.
+
+Requirements:
+1. Overview: What the project does, target audience, primary use case
+2. Features: List ACTUAL features from package.json scripts and dependencies - NO made-up features
+3. Installation: Exact commands for npm/pnpm/yarn
+4. Usage: Real code examples based on actual entry points and exports
+5. API: Document actual functions/endpoints/components found in the code
+6. Configuration: Actual config options if any
+7. Development: Real npm scripts from package.json
+8. Testing: How tests are run (from scripts)
+9. Project Structure: Actual directory layout
+10. Technologies: Real stack based on dependencies
+
+IMPORTANT: Use the project data provided and INFER realistic content. Don't leave placeholder text.
+
+Project Data:
+${JSON.stringify(projectData, null, 2)}
+
+Return ONLY complete README.md markdown.`;
 }
-function buildGeneratePrompt(projectData) {
-    return `Generate a detailed, professional README.md for this project with these exact sections:
+function buildContributingPrompt(projectData) {
+    return `Generate a professional CONTRIBUTING.md for this project.
 
-# Project Name
+Requirements:
+1. How to contribute (fork, clone, branch workflow)
+2. Development setup steps
+3. Code style guidelines (linting, formatting)
+4. Commit message conventions
+5. PR process and requirements
+6. Testing requirements
+7. Types of contributions welcome
+8. Issue templates or guidelines
+9. Recognition for contributors
+10. Code of conduct reference
 
-## Overview
-2-3 sentences about what this project does and its purpose.
+Base it on the project's package.json scripts and known best practices.
 
-## Version
-Current version: X.X.X
+Project Data:
+${JSON.stringify(projectData, null, 2)}
 
-## Description
-Detailed description of what this project does, who it's for, and why to use it.
+Return ONLY complete CONTRIBUTING.md markdown.`;
+}
+function buildLicensePrompt(projectData) {
+    return `Generate a complete LICENSE file. Use MIT License as default. Include:
 
-## Features
-- List real features (infer from package.json scripts and dependencies)
-- Do NOT add fake features
+1. Full MIT License text
+2. Copyright year (current year: ${new Date().getFullYear()})
+3. Project name: ${projectData.name || 'Project'}
+4. Author name if available in package.json
 
-## Installation
-Provide the exact npm/pnpm/yarn install command.
+Project Data:
+${JSON.stringify(projectData, null, 2)}
+
+Return ONLY complete LICENSE file content (plain text, not markdown).`;
+}
+function buildSecurityPrompt(projectData) {
+    return `Generate a professional SECURITY.md for this project.
 
-## Usage
-Show actual usage examples with code snippets.
+Requirements:
+1. How to report vulnerabilities
+2. Security policy scope
+3. Response timeline
+4. Disclosure guidelines
+5. Security update process
+6. Contact information
+7. Dependencies security info if applicable
 
-## API Reference
-If backend: document endpoints
-If frontend: document components/hooks
-If library: document exports/functions
+Project Data:
+${JSON.stringify(projectData, null, 2)}
 
-## Configuration
-Any config options available.
+Return ONLY complete SECURITY.md markdown.`;
+}
+function buildImproveReadmePrompt(projectData) {
+    return `Improve the existing README.md for this project.
 
-## Development
-How to set up dev environment:
-npm install
-npm run [script] - description
+Requirements:
+1. Fix unclear sections
+2. Add missing important sections
+3. Improve code examples to be more practical
+4. Make installation more clear
+5. Enhance usage examples with real code
+6. Add API documentation if missing
+7. Improve structure and readability
+8. Keep existing good content
+9. Add project structure visualization if missing
 
-## Testing
-How to run tests.
+EXISTING README:
+${projectData.existingReadme || 'No existing README'}
 
-## License
-MIT License - include full license text
+Project Data:
+${JSON.stringify(projectData, null, 2)}
 
-## Contributing
-Contribution Guidelines:
-- Fork and create feature branch
-- Follow existing code style
-- Add tests if applicable
-- Submit PR
+Return ONLY improved README.md markdown.`;
+}
+function buildImproveContributingPrompt(projectData) {
+    return `Improve the existing CONTRIBUTING.md for this project.
 
-## Security Policy
-How to report vulnerabilities.
+Requirements:
+1. Make contribution process clearer
+2. Add missing guidelines
+3. Improve development setup instructions
+4. Add testing requirements
+5. Update for current project state
 
-## Code of Conduct
-Be respectful, inclusive, no harassment. violations will result in immediate suspension.
+EXISTING CONTRIBUTING:
+${projectData.existingContributing || 'No existing CONTRIBUTING.md'}
 
 Project Data:
 ${JSON.stringify(projectData, null, 2)}
 
-Return ONLY complete markdown.`;
+Return ONLY improved CONTRIBUTING.md markdown.`;
 }
-function buildImprovePrompt(projectData) {
-    return `Improve the existing README.md for this project.
+function buildImproveLicensePrompt(projectData) {
+    return `Review and improve the LICENSE if needed.
+
+Requirements:
+1. Ensure correct copyright year
+2. Verify project name is correct
+3. Add any missing clauses
 
-Current README:
-${projectData.existingReadme}
+EXISTING LICENSE:
+${projectData.existingLicense || 'No existing LICENSE'}
 
 Project Data:
 ${JSON.stringify(projectData, null, 2)}
 
-Rules:
-- Keep improvements practical and factual
-- Do NOT add hallucinated features
-- Maintain existing structure
-- Make it more useful for developers
+Return ONLY complete LICENSE file content.`;
+}
+function buildImproveSecurityPrompt(projectData) {
+    return `Improve the existing SECURITY.md for this project.
+
+Requirements:
+1. Make reporting process clearer
+2. Add missing security information
+3. Update contact info if needed
+4. Improve response timeline clarity
+
+EXISTING SECURITY:
+${projectData.existingSecurity || 'No existing SECURITY.md'}
 
-Return ONLY the improved markdown.`;
+Project Data:
+${JSON.stringify(projectData, null, 2)}
+
+Return ONLY improved SECURITY.md markdown.`;
 }
 function buildExplainPrompt(projectData) {
-    return `Explain this project in simple terms for developers.
+    return `Explain this project in simple, practical terms for developers.
+
+Requirements:
+1. What the project does (one sentence)
+2. Who it's for
+3. Main features (top 3-5)
+4. How to use it (quick start)
+5. Why someone would use it
+6. Tech stack briefly
+
+Be conversational but professional. No markdown.
 
 Project Data:
 ${JSON.stringify(projectData, null, 2)}
 
-Rules:
-- Be concise
-- Explain what the project does
-- Explain how to use it
-- Mention key features
-- Keep it conversational
-
 Return as plain text explanation.`;
 }
 async function makeRequest(apiKey, userPrompt) {
@@ -169,7 +259,7 @@ async function makeRequest(apiKey, userPrompt) {
             { role: 'system', content: SYSTEM_PROMPT },
             { role: 'user', content: userPrompt }
         ],
-        temperature: 0.7,
+        temperature: 0.5,
         max_tokens: 8192
     }, {
         headers: {

package/dist/services/ai.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai.js","sourceRoot":"","sources":["../../src/services/ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAG/C,MAAM,YAAY,GAAG,iDAAiD,CAAC;AACvE,MAAM,KAAK,GAAG,yBAAyB,CAAC;AAExC,MAAM,aAAa,GAAG,4OAA4O,CAAC;AAEnQ,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,UAAU,GAAG,IAAI,CAAC;AAExB,KAAK,UAAU,KAAK,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,EAAoB,EACpB,aAAqB,WAAW,EAChC,YAAoB,UAAU;IAE9B,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QACvD,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAEhE,IAAI,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,MAAM,GAAG,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC;gBAEpC,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;oBACrC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;gBACrC,CAAC;gBAED,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;oBACnB,MAAM,UAAU,GAAG,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC;oBAC1D,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBAC1F,OAAO,CAAC,GAAG,CAAC,+BAA+B,KAAK,GAAC,IAAI,iBAAiB,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,CAAC,CAAC;oBACxG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;oBACnB,SAAS;gBACX,CAAC;gBAED,IAAI,MAAM,IAAI,MAAM,IAAI,GAAG,EAAE,CAAC;oBAC5B,MAAM,KAAK,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBAC/C,OAAO,CAAC,GAAG,CAAC,+BAA+B,KAAK,GAAC,IAAI,iBAAiB,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,CAAC,CAAC;oBACxG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;oBACnB,SAAS;gBACX,CAAC;YACH,CAAC;YAED,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,CAAC,OAAO,iBAAiB,KAAK,GAAC,IAAI,MAAM,CAAC,CAAC;gBAC5E,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;AAC/D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,WAAoC,EAAE,OAA2C,UAAU;IAC5H,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;IAElC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,UAAU;YACb,UAAU,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAC9C,MAAM;QACR,KAAK,SAAS;YACZ,UAAU,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM;QACR,KAAK,SAAS;YACZ,UAAU,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM;QACR;YACE,UAAU,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;IAExE,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AACjD,CAAC;AAED,SAAS,mBAAmB,CAAC,WAAoC;IAC/D,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAwDP,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;+BAEP,CAAC;AAChC,CAAC;AAED,SAAS,kBAAkB,CAAC,WAAoC;IAC9D,OAAO;;;EAGP,WAAW,CAAC,cAAc;;;EAG1B,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;mCAQH,CAAC;AACpC,CAAC;AAED,SAAS,kBAAkB,CAAC,WAAoC;IAC9D,OAAO;;;EAGP,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;;kCASJ,CAAC;AACnC,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,MAAc,EAAE,UAAkB;IAC3D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,YAAY,EACZ;QACE,KAAK,EAAE,KAAK;QACZ,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE;YAC1C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE;SACtC;QACD,WAAW,EAAE,GAAG;QAChB,UAAU,EAAE,IAAI;KACjB,EACD;QACE,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,eAAe,EAAE,UAAU,MAAM,EAAE;SACpC;QACD,OAAO,EAAE,MAAM;KAChB,CACF,CAAC;IAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;AAC1D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAc;IACjD,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,IAAI,CACd,YAAY,EACZ;YACE,KAAK,EAAE,KAAK;YACZ,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,8BAA8B,EAAE;gBAC3D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;aAChC;YACD,UAAU,EAAE,EAAE;SACf,EACD;YACE,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,MAAM,EAAE;aACpC;YACD,OAAO,EAAE,KAAK;SACf,CACF,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,CAAC,EAAE,CAAC;YAC9F,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}
+{"version":3,"file":"ai.js","sourceRoot":"","sources":["../../src/services/ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAG/C,MAAM,YAAY,GAAG,iDAAiD,CAAC;AACvE,MAAM,KAAK,GAAG,yBAAyB,CAAC;AAExC,MAAM,aAAa,GAAG,kSAAkS,CAAC;AAEzT,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,UAAU,GAAG,IAAI,CAAC;AAExB,KAAK,UAAU,KAAK,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,EAAoB,EACpB,aAAqB,WAAW,EAChC,YAAoB,UAAU;IAE9B,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QACvD,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAEhE,IAAI,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,MAAM,GAAG,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC;gBAEpC,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;oBACrC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;gBACrC,CAAC;gBAED,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;oBACnB,MAAM,UAAU,GAAG,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC;oBAC1D,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBAC1F,OAAO,CAAC,GAAG,CAAC,+BAA+B,KAAK,GAAC,IAAI,iBAAiB,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,CAAC,CAAC;oBACxG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;oBACnB,SAAS;gBACX,CAAC;gBAED,IAAI,MAAM,IAAI,MAAM,IAAI,GAAG,EAAE,CAAC;oBAC5B,MAAM,KAAK,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBAC/C,OAAO,CAAC,GAAG,CAAC,+BAA+B,KAAK,GAAC,IAAI,iBAAiB,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,CAAC,CAAC;oBACxG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;oBACnB,SAAS;gBACX,CAAC;YACH,CAAC;YAED,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC/C,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,CAAC,OAAO,iBAAiB,KAAK,GAAC,IAAI,MAAM,CAAC,CAAC;gBAC5E,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;AAC/D,CAAC;AAUD,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,WAAoC,EAAE,OAA2C,UAAU;IAC5H,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;IAElC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAC7F,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;QACxB,MAAM,CAAC,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAClE,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAAC;YACpE,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,uBAAuB,CAAC,WAAW,CAAC,CAAC,CAAC;YAC1E,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC;YACrE,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC,CAAC;SACvE,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,MAAM;YACf,YAAY;YACZ,OAAO;YACP,QAAQ;SACe,CAAC;IAC5B,CAAC;IAED,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,MAAM,CAAC,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAClE,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,wBAAwB,CAAC,WAAW,CAAC,CAAC,CAAC;YAC3E,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,8BAA8B,CAAC,WAAW,CAAC,CAAC,CAAC;YACjF,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,WAAW,CAAC,CAAC,CAAC;YAC5E,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,0BAA0B,CAAC,WAAW,CAAC,CAAC,CAAC;SAC9E,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,MAAM;YACf,YAAY;YACZ,OAAO;YACP,QAAQ;SACe,CAAC;IAC5B,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,iBAAiB,CAAC,WAAoC;IAC7D,OAAO;;;;;;;;;;;;;;;;;;;EAmBP,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;yCAEG,CAAC;AAC1C,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAoC;IACnE,OAAO;;;;;;;;;;;;;;;;;EAiBP,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;+CAES,CAAC;AAChD,CAAC;AAED,SAAS,kBAAkB,CAAC,WAAoC;IAC9D,OAAO;;;mCAG0B,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;mBACxC,WAAW,CAAC,IAAI,IAAI,SAAS;;;;EAI9C,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;sEAEgC,CAAC;AACvE,CAAC;AAED,SAAS,mBAAmB,CAAC,WAAoC;IAC/D,OAAO;;;;;;;;;;;;EAYP,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;2CAEK,CAAC;AAC5C,CAAC;AAED,SAAS,wBAAwB,CAAC,WAAoC;IACpE,OAAO;;;;;;;;;;;;;;EAcP,WAAW,CAAC,cAAc,IAAI,oBAAoB;;;EAGlD,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;yCAEG,CAAC;AAC1C,CAAC;AAED,SAAS,8BAA8B,CAAC,WAAoC;IAC1E,OAAO;;;;;;;;;;EAUP,WAAW,CAAC,oBAAoB,IAAI,6BAA6B;;;EAGjE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;+CAES,CAAC;AAChD,CAAC;AAED,SAAS,yBAAyB,CAAC,WAAoC;IACrE,OAAO;;;;;;;;EAQP,WAAW,CAAC,eAAe,IAAI,qBAAqB;;;EAGpD,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;2CAEK,CAAC;AAC5C,CAAC;AAED,SAAS,0BAA0B,CAAC,WAAoC;IACtE,OAAO;;;;;;;;;EASP,WAAW,CAAC,gBAAgB,IAAI,yBAAyB;;;EAGzD,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;2CAEK,CAAC;AAC5C,CAAC;AAED,SAAS,kBAAkB,CAAC,WAAoC;IAC9D,OAAO;;;;;;;;;;;;;EAaP,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;;kCAEJ,CAAC;AACnC,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,MAAc,EAAE,UAAkB;IAC3D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,YAAY,EACZ;QACE,KAAK,EAAE,KAAK;QACZ,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE;YAC1C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE;SACtC;QACD,WAAW,EAAE,GAAG;QAChB,UAAU,EAAE,IAAI;KACjB,EACD;QACE,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,eAAe,EAAE,UAAU,MAAM,EAAE;SACpC;QACD,OAAO,EAAE,MAAM;KAChB,CACF,CAAC;IAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;AAC1D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAc;IACjD,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,IAAI,CACd,YAAY,EACZ;YACE,KAAK,EAAE,KAAK;YACZ,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,8BAA8B,EAAE;gBAC3D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;aAChC;YACD,UAAU,EAAE,EAAE;SACf,EACD;YACE,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,MAAM,EAAE;aACpC;YACD,OAAO,EAAE,KAAK;SACf,CACF,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,CAAC,EAAE,CAAC;YAC9F,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/ui/components/Generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Generator.d.ts","sourceRoot":"","sources":["../../../src/ui/components/Generator.tsx"],"names":[],"mappings":"AASA,UAAU,cAAc;IACvB,IAAI,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;IACzC,MAAM,EAAE,MAAM,IAAI,CAAC;CACnB;AAQD,MAAM,CAAC,OAAO,UAAU,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,2CA+FjE"}
+{"version":3,"file":"Generator.d.ts","sourceRoot":"","sources":["../../../src/ui/components/Generator.tsx"],"names":[],"mappings":"AASA,UAAU,cAAc;IACvB,IAAI,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;IACzC,MAAM,EAAE,MAAM,IAAI,CAAC;CACnB;AAQD,MAAM,CAAC,OAAO,UAAU,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,2CAiHjE"}

package/dist/ui/components/Generator.js

@@ -8,8 +8,8 @@ import { analyzeProject, summarizeProject } from '../../services/analyzer.js';
 import fs from 'fs/promises';
 import path from 'path';
 const titles = {
-    generate: 'Generate README',
-    improve: 'Improve README',
+    generate: 'Generate Documentation',
+    improve: 'Improve Documentation',
     explain: 'Explain Project',
 };
 export default function Generator({ mode, onBack }) {
@@ -48,17 +48,29 @@ export default function Generator({ mode, onBack }) {
             setStatus('generating');
             spinner.text = 'Generating documentation...';
             const docResult = await generateDocs(summary, mode);
-            if (mode === 'generate') {
-                const readmePath = path.join(projectPath, 'README.md');
-                await fs.writeFile(readmePath, docResult.content);
-                setResult('README.md created successfully!');
+            if (mode === 'generate' || mode === 'improve') {
+                const files = [];
+                if (docResult.content) {
+                    await fs.writeFile(path.join(projectPath, 'README.md'), docResult.content);
+                    files.push('README.md');
+                }
+                if (docResult.contributing) {
+                    await fs.writeFile(path.join(projectPath, 'CONTRIBUTING.md'), docResult.contributing);
+                    files.push('CONTRIBUTING.md');
+                }
+                if (docResult.license) {
+                    await fs.writeFile(path.join(projectPath, 'LICENSE'), docResult.license);
+                    files.push('LICENSE');
+                }
+                if (docResult.security) {
+                    await fs.writeFile(path.join(projectPath, 'SECURITY.md'), docResult.security);
+                    files.push('SECURITY.md');
+                }
+                setResult(`${files.join(', ')} ${files.length === 1 ? 'created' : 'created'} successfully!`);
             }
             else if (mode === 'explain') {
                 setResult(docResult.content);
             }
-            else {
-                setResult('Documentation improved!');
-            }
             spinner.succeed('Done!');
             setStatus('done');
         }
@@ -68,6 +80,6 @@ export default function Generator({ mode, onBack }) {
             setStatus('error');
         }
     };
-    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { bold: true, color: "cyan", children: titles[mode] }), _jsx(Text, { dimColor: true, children: process.cwd() }), _jsx(Text, { children: "\n" }), status === 'checking' && _jsx(Text, { children: "Checking API key..." }), status === 'scanning' && _jsx(Text, { children: "Scanning project..." }), status === 'analyzing' && _jsx(Text, { children: "Analyzing project..." }), status === 'generating' && _jsx(Text, { children: "Generating documentation..." }), status === 'done' && (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: "green", children: ["\u2713 ", result] }), _jsx(Text, { children: "\n" }), _jsx(Text, { dimColor: true, children: "Press any key to continue..." })] })), status === 'error' && (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: "red", children: ["\u2717 ", error] }), _jsx(Text, { children: "\n" }), _jsx(Text, { dimColor: true, children: "Press any key to go back..." })] }))] }));
+    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { bold: true, color: "cyan", children: titles[mode] }), _jsx(Text, { dimColor: true, children: process.cwd() }), _jsx(Text, { children: "\n" }), status === 'checking' && _jsx(Text, { children: "Checking API key..." }), status === 'scanning' && _jsx(Text, { children: "Scanning project..." }), status === 'analyzing' && _jsx(Text, { children: "Analyzing project..." }), status === 'generating' && _jsx(Text, { children: "Generating documentation (README, CONTRIBUTING, LICENSE, SECURITY)..." }), status === 'done' && (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: "green", children: ["\u2713 ", result] }), _jsx(Text, { children: "\n" }), _jsx(Text, { dimColor: true, children: "Press any key to continue..." })] })), status === 'error' && (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: "red", children: ["\u2717 ", error] }), _jsx(Text, { children: "\n" }), _jsx(Text, { dimColor: true, children: "Press any key to go back..." })] }))] }));
 }
 //# sourceMappingURL=Generator.js.map

package/dist/ui/components/Generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"Generator.js","sourceRoot":"","sources":["../../../src/ui/components/Generator.tsx"],"names":[],"mappings":";AAAA,OAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,KAAK,CAAC;AAC1C,OAAO,GAAG,MAAM,KAAK,CAAC;AACtB,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9E,OAAO,EAAE,MAAM,aAAa,CAAC;AAC7B,OAAO,IAAI,MAAM,MAAM,CAAC;AAOxB,MAAM,MAAM,GAAG;IACd,QAAQ,EAAE,iBAAiB;IAC3B,OAAO,EAAE,gBAAgB;IACzB,OAAO,EAAE,iBAAiB;CAC1B,CAAC;AAEF,MAAM,CAAC,OAAO,UAAU,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAkB;IACjE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAA0E,UAAU,CAAC,CAAC;IAC1H,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IACxD,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAE1D,SAAS,CAAC,GAAG,EAAE;QACd,YAAY,EAAE,CAAC;IAChB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,QAAQ,CAAC,GAAG,EAAE;QACb,IAAI,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC7C,MAAM,EAAE,CAAC;QACV,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;QAC/B,MAAM,OAAO,GAAG,GAAG,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QAE1C,IAAI,CAAC;YACJ,SAAS,CAAC,UAAU,CAAC,CAAC;YACtB,OAAO,CAAC,IAAI,GAAG,qBAAqB,CAAC;YAErC,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACb,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;gBAC7D,QAAQ,CAAC,uBAAuB,CAAC,CAAC;gBAClC,SAAS,CAAC,OAAO,CAAC,CAAC;gBACnB,OAAO;YACR,CAAC;YAED,SAAS,CAAC,UAAU,CAAC,CAAC;YACtB,OAAO,CAAC,IAAI,GAAG,qBAAqB,CAAC;YAErC,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAClC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAE7B,SAAS,CAAC,WAAW,CAAC,CAAC;YACvB,OAAO,CAAC,IAAI,GAAG,sBAAsB,CAAC;YAEtC,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,OAAO,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAEvC,SAAS,CAAC,YAAY,CAAC,CAAC;YACxB,OAAO,CAAC,IAAI,GAAG,6BAA6B,CAAC;YAE7C,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,OAA6C,EAAE,IAAI,CAAC,CAAC;YAE1F,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACzB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;gBACvD,MAAM,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;gBAClD,SAAS,CAAC,iCAAiC,CAAC,CAAC;YAC9C,CAAC;iBAAM,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC/B,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC;iBAAM,CAAC;gBACP,SAAS,CAAC,yBAAyB,CAAC,CAAC;YACtC,CAAC;YAED,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACzB,SAAS,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YACnE,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YAC/D,SAAS,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CACN,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,EAAC,OAAO,EAAE,CAAC,aACrC,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,MAAM,YAAE,MAAM,CAAC,IAAI,CAAC,GAAQ,EAC7C,KAAC,IAAI,IAAC,QAAQ,kBAAE,OAAO,CAAC,GAAG,EAAE,GAAQ,EACrC,KAAC,IAAI,cAAE,IAAI,GAAQ,EAElB,MAAM,KAAK,UAAU,IAAI,KAAC,IAAI,sCAA2B,EACzD,MAAM,KAAK,UAAU,IAAI,KAAC,IAAI,sCAA2B,EACzD,MAAM,KAAK,WAAW,IAAI,KAAC,IAAI,uCAA4B,EAC3D,MAAM,KAAK,YAAY,IAAI,KAAC,IAAI,8CAAmC,EAEnE,MAAM,KAAK,MAAM,IAAI,CACrB,8BACC,MAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,OAAO,wBAAI,MAAM,IAAQ,EAC1C,KAAC,IAAI,cAAE,IAAI,GAAQ,EACnB,KAAC,IAAI,IAAC,QAAQ,mDAAoC,IAChD,CACH,EAEA,MAAM,KAAK,OAAO,IAAI,CACtB,8BACC,MAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,KAAK,wBAAI,KAAK,IAAQ,EACvC,KAAC,IAAI,cAAE,IAAI,GAAQ,EACnB,KAAC,IAAI,IAAC,QAAQ,kDAAmC,IAC/C,CACH,IACI,CACN,CAAC;AACH,CAAC"}
+{"version":3,"file":"Generator.js","sourceRoot":"","sources":["../../../src/ui/components/Generator.tsx"],"names":[],"mappings":";AAAA,OAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,KAAK,CAAC;AAC1C,OAAO,GAAG,MAAM,KAAK,CAAC;AACtB,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAiB,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9E,OAAO,EAAE,MAAM,aAAa,CAAC;AAC7B,OAAO,IAAI,MAAM,MAAM,CAAC;AAOxB,MAAM,MAAM,GAAG;IACd,QAAQ,EAAE,wBAAwB;IAClC,OAAO,EAAE,uBAAuB;IAChC,OAAO,EAAE,iBAAiB;CAC1B,CAAC;AAEF,MAAM,CAAC,OAAO,UAAU,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAkB;IACjE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAA0E,UAAU,CAAC,CAAC;IAC1H,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IACxD,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAE1D,SAAS,CAAC,GAAG,EAAE;QACd,YAAY,EAAE,CAAC;IAChB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,QAAQ,CAAC,GAAG,EAAE;QACb,IAAI,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YAC7C,MAAM,EAAE,CAAC;QACV,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;QAC/B,MAAM,OAAO,GAAG,GAAG,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QAE1C,IAAI,CAAC;YACJ,SAAS,CAAC,UAAU,CAAC,CAAC;YACtB,OAAO,CAAC,IAAI,GAAG,qBAAqB,CAAC;YAErC,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACb,OAAO,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;gBAC7D,QAAQ,CAAC,uBAAuB,CAAC,CAAC;gBAClC,SAAS,CAAC,OAAO,CAAC,CAAC;gBACnB,OAAO;YACR,CAAC;YAED,SAAS,CAAC,UAAU,CAAC,CAAC;YACtB,OAAO,CAAC,IAAI,GAAG,qBAAqB,CAAC;YAErC,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAClC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAE7B,SAAS,CAAC,WAAW,CAAC,CAAC;YACvB,OAAO,CAAC,IAAI,GAAG,sBAAsB,CAAC;YAEtC,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,OAAO,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAEvC,SAAS,CAAC,YAAY,CAAC,CAAC;YACxB,OAAO,CAAC,IAAI,GAAG,6BAA6B,CAAC;YAE7C,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,OAA6C,EAAE,IAAI,CAA6B,CAAC;YAEtH,IAAI,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC/C,MAAM,KAAK,GAAa,EAAE,CAAC;gBAE3B,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;oBACvB,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;oBAC3E,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBACzB,CAAC;gBAED,IAAI,SAAS,CAAC,YAAY,EAAE,CAAC;oBAC5B,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,iBAAiB,CAAC,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;oBACtF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBAC/B,CAAC;gBAED,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;oBACvB,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;oBACzE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACvB,CAAC;gBAED,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;oBACxB,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC;oBAC9E,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC3B,CAAC;gBAED,SAAS,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,gBAAgB,CAAC,CAAC;YAC9F,CAAC;iBAAM,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC/B,SAAS,CAAC,SAAS,CAAC,OAAiB,CAAC,CAAC;YACxC,CAAC;YAED,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACzB,SAAS,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YACnE,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YAC/D,SAAS,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CACN,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,EAAC,OAAO,EAAE,CAAC,aACrC,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,MAAM,YAAE,MAAM,CAAC,IAAI,CAAC,GAAQ,EAC7C,KAAC,IAAI,IAAC,QAAQ,kBAAE,OAAO,CAAC,GAAG,EAAE,GAAQ,EACrC,KAAC,IAAI,cAAE,IAAI,GAAQ,EAElB,MAAM,KAAK,UAAU,IAAI,KAAC,IAAI,sCAA2B,EACzD,MAAM,KAAK,UAAU,IAAI,KAAC,IAAI,sCAA2B,EACzD,MAAM,KAAK,WAAW,IAAI,KAAC,IAAI,uCAA4B,EAC3D,MAAM,KAAK,YAAY,IAAI,KAAC,IAAI,wFAA6E,EAE7G,MAAM,KAAK,MAAM,IAAI,CACrB,8BACC,MAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,OAAO,wBAAI,MAAM,IAAQ,EAC1C,KAAC,IAAI,cAAE,IAAI,GAAQ,EACnB,KAAC,IAAI,IAAC,QAAQ,mDAAoC,IAChD,CACH,EAEA,MAAM,KAAK,OAAO,IAAI,CACtB,8BACC,MAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,KAAK,wBAAI,KAAK,IAAQ,EACvC,KAAC,IAAI,cAAE,IAAI,GAAQ,EACnB,KAAC,IAAI,IAAC,QAAQ,kDAAmC,IAC/C,CACH,IACI,CACN,CAAC;AACH,CAAC"}

package/dist/ui/components/Menu.js

@@ -2,8 +2,8 @@ import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-run
 import { useState } from 'react';
 import { Box, Text, useInput } from 'ink';
 const menuItems = [
-    { key: 'g', label: 'Generate README', desc: 'Create new documentation' },
-    { key: 'i', label: 'Improve README', desc: 'Enhance existing docs' },
+    { key: 'g', label: 'Generate Docs', desc: 'Create README, CONTRIBUTING, LICENSE, SECURITY' },
+    { key: 'i', label: 'Improve Docs', desc: 'Enhance existing documentation' },
     { key: 'e', label: 'Explain Project', desc: 'Terminal explanation' },
     { key: 's', label: 'Settings', desc: 'Configure API key' },
     { key: 'h', label: 'Help', desc: 'Usage guide' },
@@ -28,6 +28,6 @@ export default function Menu({ onSelect }) {
             }
         }
     });
-    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { bold: true, color: "cyan", children: "\u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2557   \u2588\u2588\u2557\u2588\u2588\u2557     \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D \u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2551   \u2588\u2588\u2551\u2588\u2588\u2551     \u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D \u2588\u2588\u2551  \u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2551   \u2588\u2588\u2551\u2588\u2588\u2551     \u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2551   \u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2551   \u2588\u2588\u2551\u2588\u2588\u2551     \u2588\u2588\u2554\u2550\u2550\u255D  \u255A\u2550\u2550\u2550\u2550\u2588\u2588\u2551 \u255A\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2551  \u2588\u2588\u2551\u255A\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2551 \u255A\u2550\u2550\u2550\u2550\u2550\u255D \u255A\u2550\u255D  \u255A\u2550\u255D\u255A\u2550\u255D  \u255A\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D" }), _jsx(Text, { dimColor: true, children: "AI-Powered Documentation Generator" }), _jsx(Text, { children: "\n" }), _jsx(Text, { bold: true, children: "Commands" }), _jsx(Text, { dimColor: true, children: "\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500" }), menuItems.map((item, index) => (_jsx(Box, { flexDirection: "column", children: index === selectedIndex ? (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: "red", children: [' ', "\u25B6 [", item.key, "] ", item.label] }), _jsxs(Text, { dimColor: true, children: ["  ", item.desc] })] })) : (_jsxs(Text, { children: ["  [", item.key, "] ", item.label] })) }, item.key))), _jsx(Text, { children: "\n" }), _jsx(Text, { dimColor: true, children: "Use w/s or arrow keys to navigate, Enter or key to select, q to quit" }), _jsx(Text, { dimColor: true, children: "Version 2.1.4" })] }));
+    return (_jsxs(Box, { flexDirection: "column", padding: 1, children: [_jsx(Text, { bold: true, color: "cyan", children: "\u2588\u2588\u2557    \u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2557   \u2588\u2588\u2588\u2557 \u2588\u2588\u2551    \u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2551 \u2588\u2588\u2551 \u2588\u2557 \u2588\u2588\u2551\u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2554\u2588\u2588\u2588\u2588\u2554\u2588\u2588\u2551 \u2588\u2588\u2551\u2588\u2588\u2588\u2557\u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u255D  \u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u255D  \u2588\u2588\u2554\u2550\u2550\u255D  \u2588\u2588\u2551\u255A\u2588\u2588\u2554\u255D\u2588\u2588\u2551 \u255A\u2588\u2588\u2588\u2554\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2551 \u255A\u2550\u255D \u2588\u2588\u2551 \u255A\u2550\u2550\u255D\u255A\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u255D  \u255A\u2550\u255D\u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u255D     \u255A\u2550\u255D" }), _jsx(Text, { dimColor: true, children: "AI-Powered Documentation Generator" }), _jsx(Text, { children: "\n" }), _jsx(Text, { bold: true, children: "Commands" }), _jsx(Text, { dimColor: true, children: "\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500" }), menuItems.map((item, index) => (_jsx(Box, { flexDirection: "column", children: index === selectedIndex ? (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: "red", children: [' ', "\u25B6 [", item.key, "] ", item.label] }), _jsxs(Text, { dimColor: true, children: ["  ", item.desc] })] })) : (_jsxs(Text, { children: ["  [", item.key, "] ", item.label] })) }, item.key))), _jsx(Text, { children: "\n" }), _jsx(Text, { dimColor: true, children: "Use w/s or arrow keys to navigate, Enter or key to select, q to quit" }), _jsx(Text, { dimColor: true, children: "Version 2.1.6" })] }));
 }
 //# sourceMappingURL=Menu.js.map

package/dist/ui/components/Menu.js.map

@@ -1 +1 @@
-{"version":3,"file":"Menu.js","sourceRoot":"","sources":["../../../src/ui/components/Menu.tsx"],"names":[],"mappings":";AAAA,OAAc,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,KAAK,CAAC;AAY1C,MAAM,SAAS,GAAe;IAC7B,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,iBAAiB,EAAE,IAAI,EAAE,0BAA0B,EAAE;IACxE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,gBAAgB,EAAE,IAAI,EAAE,uBAAuB,EAAE;IACpE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,iBAAiB,EAAE,IAAI,EAAE,sBAAsB,EAAE;IACpE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,mBAAmB,EAAE;IAC1D,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;IAChD,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE;CACrD,CAAC;AAEF,MAAM,CAAC,OAAO,UAAU,IAAI,CAAC,EAAE,QAAQ,EAAa;IACnD,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IAEtD,QAAQ,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACvB,IAAI,KAAK,KAAK,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAClC,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC;aAAM,IAAI,KAAK,KAAK,GAAG,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAC3C,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC;aAAM,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACP,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;YAClE,IAAI,IAAI,EAAE,CAAC;gBACV,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACpB,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,OAAO,CACN,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,EAAC,OAAO,EAAE,CAAC,aACrC,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,MAAM,0xDAShB,EAEP,KAAC,IAAI,IAAC,QAAQ,yDAA0C,EACxD,KAAC,IAAI,cAAE,IAAI,GAAQ,EAEnB,KAAC,IAAI,IAAC,IAAI,+BAAgB,EAC1B,KAAC,IAAI,IAAC,QAAQ,6EAAiB,EAE9B,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAC/B,KAAC,GAAG,IAAgB,aAAa,EAAC,QAAQ,YACxC,KAAK,KAAK,aAAa,CAAC,CAAC,CAAC,CAC1B,8BACC,MAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,KAAK,aACpB,GAAG,cACA,IAAI,CAAC,GAAG,QAAI,IAAI,CAAC,KAAK,IACpB,EACP,MAAC,IAAI,IAAC,QAAQ,yBAAI,IAAI,CAAC,IAAI,IAAQ,IACjC,CACH,CAAC,CAAC,CAAC,CACH,MAAC,IAAI,sBAAK,IAAI,CAAC,GAAG,QAAI,IAAI,CAAC,KAAK,IAAQ,CACxC,IAXQ,IAAI,CAAC,GAAG,CAYZ,CACN,CAAC,EAEF,KAAC,IAAI,cAAE,IAAI,GAAQ,EACnB,KAAC,IAAI,IAAC,QAAQ,2FAA4E,EAC1F,KAAC,IAAI,IAAC,QAAQ,oCAAqB,IAC9B,CACN,CAAC;AACH,CAAC"}
+{"version":3,"file":"Menu.js","sourceRoot":"","sources":["../../../src/ui/components/Menu.tsx"],"names":[],"mappings":";AAAA,OAAc,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,KAAK,CAAC;AAY1C,MAAM,SAAS,GAAe;IAC7B,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,gDAAgD,EAAE;IAC5F,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,gCAAgC,EAAE;IAC3E,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,iBAAiB,EAAE,IAAI,EAAE,sBAAsB,EAAE;IACpE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,mBAAmB,EAAE;IAC1D,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;IAChD,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE;CACrD,CAAC;AAEF,MAAM,CAAC,OAAO,UAAU,IAAI,CAAC,EAAE,QAAQ,EAAa;IACnD,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IAEtD,QAAQ,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACvB,IAAI,KAAK,KAAK,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAClC,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC;aAAM,IAAI,KAAK,KAAK,GAAG,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAC3C,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC;aAAM,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACP,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;YAClE,IAAI,IAAI,EAAE,CAAC;gBACV,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACpB,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,OAAO,CACN,MAAC,GAAG,IAAC,aAAa,EAAC,QAAQ,EAAC,OAAO,EAAE,CAAC,aACrC,KAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,MAAM,isDAShB,EAEP,KAAC,IAAI,IAAC,QAAQ,yDAA0C,EACxD,KAAC,IAAI,cAAE,IAAI,GAAQ,EAEnB,KAAC,IAAI,IAAC,IAAI,+BAAgB,EAC1B,KAAC,IAAI,IAAC,QAAQ,6EAAiB,EAE9B,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAC/B,KAAC,GAAG,IAAgB,aAAa,EAAC,QAAQ,YACxC,KAAK,KAAK,aAAa,CAAC,CAAC,CAAC,CAC1B,8BACC,MAAC,IAAI,IAAC,IAAI,QAAC,KAAK,EAAC,KAAK,aACpB,GAAG,cACA,IAAI,CAAC,GAAG,QAAI,IAAI,CAAC,KAAK,IACpB,EACP,MAAC,IAAI,IAAC,QAAQ,yBAAI,IAAI,CAAC,IAAI,IAAQ,IACjC,CACH,CAAC,CAAC,CAAC,CACH,MAAC,IAAI,sBAAK,IAAI,CAAC,GAAG,QAAI,IAAI,CAAC,KAAK,IAAQ,CACxC,IAXQ,IAAI,CAAC,GAAG,CAYZ,CACN,CAAC,EAEF,KAAC,IAAI,cAAE,IAAI,GAAQ,EACnB,KAAC,IAAI,IAAC,QAAQ,2FAA4E,EAC1F,KAAC,IAAI,IAAC,QAAQ,oCAAqB,IAC9B,CACN,CAAC;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shadowdocs",
-  "version": "2.1.5",
+  "version": "2.1.6",
   "description": "AI-powered documentation generator for developer projects",
   "type": "module",
   "main": "dist/interactive.js",

package/src/services/ai.ts

@@ -5,7 +5,7 @@ import { DocResult } from '../core/types.js';
 const GROQ_API_URL = 'https://api.groq.com/openai/v1/chat/completions';
 const MODEL = 'llama-3.3-70b-versatile';
 
-const SYSTEM_PROMPT = `You are a senior open-source maintainer and technical documentation expert. Generate detailed, professional READMEs with proper sections, realistic content (no fake features), MIT license, contribution guidelines, and security policy.`;
+const SYSTEM_PROMPT = `You are a senior open-source maintainer and technical documentation expert with 15+ years of experience. You generate comprehensive, production-grade documentation that is practical, accurate, and developer-friendly. NEVER hallucinate features - only document what exists in the codebase.`;
 
 const MAX_RETRIES = 3;
 const BASE_DELAY = 1000;
@@ -61,6 +61,14 @@ async function withRetry<T>(
   throw lastError || new Error('Request failed after retries');
 }
 
+export interface GeneratedDocs {
+  content?: string;
+  readme: string;
+  contributing: string;
+  license: string;
+  security: string;
+}
+
 export async function generateDocs(projectData: Record<string, unknown>, mode: 'generate' | 'improve' | 'explain' = 'generate'): Promise<DocResult> {
   const apiKey = await loadApiKey();
   
@@ -68,124 +76,221 @@ export async function generateDocs(projectData: Record<string, unknown>, mode: '
     throw new Error('API key not configured. Run shadowdocs setup.');
   }
 
-  let userPrompt = '';
-
-  switch (mode) {
-    case 'generate':
-      userPrompt = buildGeneratePrompt(projectData);
-      break;
-    case 'improve':
-      userPrompt = buildImprovePrompt(projectData);
-      break;
-    case 'explain':
-      userPrompt = buildExplainPrompt(projectData);
-      break;
-    default:
-      userPrompt = buildGeneratePrompt(projectData);
-  }
-
-  const response = await withRetry(() => makeRequest(apiKey, userPrompt));
-  
   if (mode === 'explain') {
+    const response = await withRetry(() => makeRequest(apiKey, buildExplainPrompt(projectData)));
     return { type: 'text', content: response };
   }
-  
-  return { type: 'markdown', content: response };
+
+  if (mode === 'generate') {
+    const [readme, contributing, license, security] = await Promise.all([
+      withRetry(() => makeRequest(apiKey, buildReadmePrompt(projectData))),
+      withRetry(() => makeRequest(apiKey, buildContributingPrompt(projectData))),
+      withRetry(() => makeRequest(apiKey, buildLicensePrompt(projectData))),
+      withRetry(() => makeRequest(apiKey, buildSecurityPrompt(projectData)))
+    ]);
+
+    return {
+      type: 'markdown',
+      content: readme,
+      contributing,
+      license,
+      security
+    } as unknown as DocResult;
+  }
+
+  if (mode === 'improve') {
+    const [readme, contributing, license, security] = await Promise.all([
+      withRetry(() => makeRequest(apiKey, buildImproveReadmePrompt(projectData))),
+      withRetry(() => makeRequest(apiKey, buildImproveContributingPrompt(projectData))),
+      withRetry(() => makeRequest(apiKey, buildImproveLicensePrompt(projectData))),
+      withRetry(() => makeRequest(apiKey, buildImproveSecurityPrompt(projectData)))
+    ]);
+
+    return {
+      type: 'markdown',
+      content: readme,
+      contributing,
+      license,
+      security
+    } as unknown as DocResult;
+  }
+
+  throw new Error('Unknown mode');
 }
 
-function buildGeneratePrompt(projectData: Record<string, unknown>): string {
-  return `Generate a detailed, professional README.md for this project with these exact sections:
+function buildReadmePrompt(projectData: Record<string, unknown>): string {
+  return `You are an expert technical writer. Generate a comprehensive, production-ready README.md for this project.
+
+Analyze the project structure and package.json thoroughly. Document ONLY what's actually present in the codebase.
 
-# Project Name
+Requirements:
+1. Overview: What the project does, target audience, primary use case
+2. Features: List ACTUAL features from package.json scripts and dependencies - NO made-up features
+3. Installation: Exact commands for npm/pnpm/yarn
+4. Usage: Real code examples based on actual entry points and exports
+5. API: Document actual functions/endpoints/components found in the code
+6. Configuration: Actual config options if any
+7. Development: Real npm scripts from package.json
+8. Testing: How tests are run (from scripts)
+9. Project Structure: Actual directory layout
+10. Technologies: Real stack based on dependencies
 
-## Overview
-2-3 sentences about what this project does and its purpose.
+IMPORTANT: Use the project data provided and INFER realistic content. Don't leave placeholder text.
 
-## Version
-Current version: X.X.X
+Project Data:
+${JSON.stringify(projectData, null, 2)}
 
-## Description
-Detailed description of what this project does, who it's for, and why to use it.
+Return ONLY complete README.md markdown.`;
+}
 
-## Features
-- List real features (infer from package.json scripts and dependencies)
-- Do NOT add fake features
+function buildContributingPrompt(projectData: Record<string, unknown>): string {
+  return `Generate a professional CONTRIBUTING.md for this project.
 
-## Installation
-Provide the exact npm/pnpm/yarn install command.
+Requirements:
+1. How to contribute (fork, clone, branch workflow)
+2. Development setup steps
+3. Code style guidelines (linting, formatting)
+4. Commit message conventions
+5. PR process and requirements
+6. Testing requirements
+7. Types of contributions welcome
+8. Issue templates or guidelines
+9. Recognition for contributors
+10. Code of conduct reference
 
-## Usage
-Show actual usage examples with code snippets.
+Base it on the project's package.json scripts and known best practices.
 
-## API Reference
-If backend: document endpoints
-If frontend: document components/hooks
-If library: document exports/functions
+Project Data:
+${JSON.stringify(projectData, null, 2)}
 
-## Configuration
-Any config options available.
+Return ONLY complete CONTRIBUTING.md markdown.`;
+}
 
-## Development
-How to set up dev environment:
-npm install
-npm run [script] - description
+function buildLicensePrompt(projectData: Record<string, unknown>): string {
+  return `Generate a complete LICENSE file. Use MIT License as default. Include:
 
-## Testing
-How to run tests.
+1. Full MIT License text
+2. Copyright year (current year: ${new Date().getFullYear()})
+3. Project name: ${projectData.name || 'Project'}
+4. Author name if available in package.json
 
-## License
-MIT License - include full license text
+Project Data:
+${JSON.stringify(projectData, null, 2)}
 
-## Contributing
-Contribution Guidelines:
-- Fork and create feature branch
-- Follow existing code style
-- Add tests if applicable
-- Submit PR
+Return ONLY complete LICENSE file content (plain text, not markdown).`;
+}
 
-## Security Policy
-How to report vulnerabilities.
+function buildSecurityPrompt(projectData: Record<string, unknown>): string {
+  return `Generate a professional SECURITY.md for this project.
 
-## Code of Conduct
-Be respectful, inclusive, no harassment. violations will result in immediate suspension.
+Requirements:
+1. How to report vulnerabilities
+2. Security policy scope
+3. Response timeline
+4. Disclosure guidelines
+5. Security update process
+6. Contact information
+7. Dependencies security info if applicable
 
 Project Data:
 ${JSON.stringify(projectData, null, 2)}
 
-Return ONLY complete markdown.`;
+Return ONLY complete SECURITY.md markdown.`;
 }
 
-function buildImprovePrompt(projectData: Record<string, unknown>): string {
+function buildImproveReadmePrompt(projectData: Record<string, unknown>): string {
   return `Improve the existing README.md for this project.
 
-Current README:
-${projectData.existingReadme}
+Requirements:
+1. Fix unclear sections
+2. Add missing important sections
+3. Improve code examples to be more practical
+4. Make installation more clear
+5. Enhance usage examples with real code
+6. Add API documentation if missing
+7. Improve structure and readability
+8. Keep existing good content
+9. Add project structure visualization if missing
+
+EXISTING README:
+${projectData.existingReadme || 'No existing README'}
+
+Project Data:
+${JSON.stringify(projectData, null, 2)}
+
+Return ONLY improved README.md markdown.`;
+}
+
+function buildImproveContributingPrompt(projectData: Record<string, unknown>): string {
+  return `Improve the existing CONTRIBUTING.md for this project.
+
+Requirements:
+1. Make contribution process clearer
+2. Add missing guidelines
+3. Improve development setup instructions
+4. Add testing requirements
+5. Update for current project state
+
+EXISTING CONTRIBUTING:
+${projectData.existingContributing || 'No existing CONTRIBUTING.md'}
+
+Project Data:
+${JSON.stringify(projectData, null, 2)}
+
+Return ONLY improved CONTRIBUTING.md markdown.`;
+}
+
+function buildImproveLicensePrompt(projectData: Record<string, unknown>): string {
+  return `Review and improve the LICENSE if needed.
+
+Requirements:
+1. Ensure correct copyright year
+2. Verify project name is correct
+3. Add any missing clauses
+
+EXISTING LICENSE:
+${projectData.existingLicense || 'No existing LICENSE'}
 
 Project Data:
 ${JSON.stringify(projectData, null, 2)}
 
-Rules:
-- Keep improvements practical and factual
-- Do NOT add hallucinated features
-- Maintain existing structure
-- Make it more useful for developers
+Return ONLY complete LICENSE file content.`;
+}
+
+function buildImproveSecurityPrompt(projectData: Record<string, unknown>): string {
+  return `Improve the existing SECURITY.md for this project.
+
+Requirements:
+1. Make reporting process clearer
+2. Add missing security information
+3. Update contact info if needed
+4. Improve response timeline clarity
+
+EXISTING SECURITY:
+${projectData.existingSecurity || 'No existing SECURITY.md'}
 
-Return ONLY the improved markdown.`;
+Project Data:
+${JSON.stringify(projectData, null, 2)}
+
+Return ONLY improved SECURITY.md markdown.`;
 }
 
 function buildExplainPrompt(projectData: Record<string, unknown>): string {
-  return `Explain this project in simple terms for developers.
+  return `Explain this project in simple, practical terms for developers.
+
+Requirements:
+1. What the project does (one sentence)
+2. Who it's for
+3. Main features (top 3-5)
+4. How to use it (quick start)
+5. Why someone would use it
+6. Tech stack briefly
+
+Be conversational but professional. No markdown.
 
 Project Data:
 ${JSON.stringify(projectData, null, 2)}
 
-Rules:
-- Be concise
-- Explain what the project does
-- Explain how to use it
-- Mention key features
-- Keep it conversational
-
 Return as plain text explanation.`;
 }
 
@@ -198,7 +303,7 @@ async function makeRequest(apiKey: string, userPrompt: string): Promise<string>
         { role: 'system', content: SYSTEM_PROMPT },
         { role: 'user', content: userPrompt }
       ],
-      temperature: 0.7,
+      temperature: 0.5,
       max_tokens: 8192
     },
     {

package/src/ui/components/Generator.tsx

@@ -2,7 +2,7 @@ import React, { useState, useEffect } from 'react';
 import { Box, Text, useInput } from 'ink';
 import ora from 'ora';
 import { loadApiKey } from '../../core/config.js';
-import { generateDocs } from '../../services/ai.js';
+import { generateDocs, GeneratedDocs } from '../../services/ai.js';
 import { analyzeProject, summarizeProject } from '../../services/analyzer.js';
 import fs from 'fs/promises';
 import path from 'path';
@@ -13,8 +13,8 @@ interface GeneratorProps {
 }
 
 const titles = {
-	generate: 'Generate README',
-	improve: 'Improve README',
+	generate: 'Generate Documentation',
+	improve: 'Improve Documentation',
 	explain: 'Explain Project',
 };
 
@@ -64,16 +64,34 @@ export default function Generator({ mode, onBack }: GeneratorProps) {
 			setStatus('generating');
 			spinner.text = 'Generating documentation...';
 
-			const docResult = await generateDocs(summary as unknown as Record<string, unknown>, mode);
+			const docResult = await generateDocs(summary as unknown as Record<string, unknown>, mode) as unknown as GeneratedDocs;
 
-			if (mode === 'generate') {
-				const readmePath = path.join(projectPath, 'README.md');
-				await fs.writeFile(readmePath, docResult.content);
-				setResult('README.md created successfully!');
+			if (mode === 'generate' || mode === 'improve') {
+				const files: string[] = [];
+
+				if (docResult.content) {
+					await fs.writeFile(path.join(projectPath, 'README.md'), docResult.content);
+					files.push('README.md');
+				}
+
+				if (docResult.contributing) {
+					await fs.writeFile(path.join(projectPath, 'CONTRIBUTING.md'), docResult.contributing);
+					files.push('CONTRIBUTING.md');
+				}
+
+				if (docResult.license) {
+					await fs.writeFile(path.join(projectPath, 'LICENSE'), docResult.license);
+					files.push('LICENSE');
+				}
+
+				if (docResult.security) {
+					await fs.writeFile(path.join(projectPath, 'SECURITY.md'), docResult.security);
+					files.push('SECURITY.md');
+				}
+
+				setResult(`${files.join(', ')} ${files.length === 1 ? 'created' : 'created'} successfully!`);
 			} else if (mode === 'explain') {
-				setResult(docResult.content);
-			} else {
-				setResult('Documentation improved!');
+				setResult(docResult.content as string);
 			}
 
 			spinner.succeed('Done!');
@@ -94,7 +112,7 @@ export default function Generator({ mode, onBack }: GeneratorProps) {
 			{status === 'checking' && <Text>Checking API key...</Text>}
 			{status === 'scanning' && <Text>Scanning project...</Text>}
 			{status === 'analyzing' && <Text>Analyzing project...</Text>}
-			{status === 'generating' && <Text>Generating documentation...</Text>}
+			{status === 'generating' && <Text>Generating documentation (README, CONTRIBUTING, LICENSE, SECURITY)...</Text>}
 
 			{status === 'done' && (
 				<>

package/src/ui/components/Menu.tsx

@@ -12,8 +12,8 @@ interface MenuProps {
 }
 
 const menuItems: MenuItem[] = [
-	{ key: 'g', label: 'Generate README', desc: 'Create new documentation' },
-	{ key: 'i', label: 'Improve README', desc: 'Enhance existing docs' },
+	{ key: 'g', label: 'Generate Docs', desc: 'Create README, CONTRIBUTING, LICENSE, SECURITY' },
+	{ key: 'i', label: 'Improve Docs', desc: 'Enhance existing documentation' },
 	{ key: 'e', label: 'Explain Project', desc: 'Terminal explanation' },
 	{ key: 's', label: 'Settings', desc: 'Configure API key' },
 	{ key: 'h', label: 'Help', desc: 'Usage guide' },
@@ -42,12 +42,12 @@ export default function Menu({ onSelect }: MenuProps) {
 		<Box flexDirection="column" padding={1}>
 			<Text bold color="cyan">
 				
-  ██████╗ ██████╗ ██████╗ ██╗   ██╗██╗     ███████╗███████╗
- ██╔════╝ ██╔══██╗██╔══██╗██║   ██║██║     ██╔════╝██╔════╝
- ██║  ███╗██████╔╝██████╔╝██║   ██║██║     █████╗  ███████╗
- ██║   ██║██╔══██╗██╔══██╗██║   ██║██║     ██╔══╝  ╚════██║
- ╚██████╔╝██║  ██║██║  ██║╚██████╔╝███████╗███████╗███████║
-  ╚═════╝ ╚═╝  ╚═╝╚═╝  ╚═╝ ╚═════╝ ╚══════╝╚══════╝╚══════╝
+  ██╗    ██╗███████╗██████╗ ███████╗███████╗███╗   ███╗
+  ██║    ██║██╔════╝██╔══██╗██╔════╝██╔════╝████╗ ████║
+  ██║ █╗ ██║█████╗  ██████╔╝█████╗  █████╗  ██╔████╔██║
+  ██║███╗██║██╔══╝  ██╔══██╗██╔══╝  ██╔══╝  ██║╚██╔╝██║
+  ╚███╔███╔╝███████╗██║  ██║███████╗███████╗██║ ╚═╝ ██║
+   ╚══╝╚══╝ ╚══════╝╚═╝  ╚═╝╚══════╝╚══════╝╚═╝     ╚═╝
  
 			</Text>
 
@@ -75,7 +75,7 @@ export default function Menu({ onSelect }: MenuProps) {
 
 			<Text>{"\n"}</Text>
 			<Text dimColor>Use w/s or arrow keys to navigate, Enter or key to select, q to quit</Text>
-			<Text dimColor>Version 2.1.4</Text>
+			<Text dimColor>Version 2.1.6</Text>
 		</Box>
 	);
 }