enhance-this 0.1.0__tar.gz

enhance_this-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 hariharen9
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

enhance_this-0.1.0/PKG-INFO

@@ -0,0 +1,167 @@
+Metadata-Version: 2.4
+Name: enhance-this
+Version: 0.1.0
+Summary: A CLI tool to enhance prompts using Ollama AI.
+Author-email: Your Name <you@example.com>
+Project-URL: Homepage, https://github.com/your-username/enhance-this
+Project-URL: Bug Tracker, https://github.com/your-username/enhance-this/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: click>=8.0.0
+Requires-Dist: requests>=2.25.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: pyperclip>=1.8.0
+Requires-Dist: rich>=10.0.0
+Requires-Dist: importlib-resources; python_version < "3.9"
+Dynamic: license-file
+
+# enhance-this ✨
+
+[![PyPI - Version](https://img.shields.io/pypi/v/enhance-this?style=for-the-badge)](https://pypi.org/project/enhance-this/)
+[![npm - Version](https://img.shields.io/npm/v/enhance-this?style=for-the-badge)](https://www.npmjs.com/package/enhance-this)
+[![Homebrew - Version](https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fformulae.brew.sh%2Fapi%2Fformula%2Fenhance-this.json&query=%24.versions.stable&label=homebrew&style=for-the-badge)](https://formulae.brew.sh/formula/enhance-this)
+[![License](https://img.shields.io/github/license/hariharen9/enhance-this?style=for-the-badge)](LICENSE)
+
+**`enhance-this`** is a powerful, fast, and reliable CLI tool designed to elevate your prompts. It takes a simple idea, enhances it using local Ollama AI models, and delivers a comprehensive, structured prompt ready to generate superior AI responses. The enhanced prompt is streamed directly to your terminal and automatically copied to your clipboard for immediate use.
+
+## 🚀 Overview
+
+The core mission of `enhance-this` is to bridge the gap between a simple user query and a high-quality, detailed prompt that AI models can understand and act upon effectively. By leveraging a sophisticated and customizable templating system, the power of local language models via Ollama, and a rich terminal interface, it transforms basic inputs into professional-grade prompts.
+
+## ✨ Features
+
+-   **Live Streaming Output**: Get instant feedback as the enhanced prompt is generated token by token, providing a dynamic user experience.
+-   **Powerful Prompt Enhancement**: Utilizes a system of style-based templates (`detailed`, `concise`, `creative`, `technical`) to convert simple prompts into comprehensive, actionable ones.
+-   **Fully Customizable Templates**: Extend the tool with your own unique enhancement styles by simply adding paths to your custom template files in `config.yaml`.
+-   **Diff View**: Use the `--diff` flag to see a clear, color-coded comparison between your original prompt and the enhanced version, highlighting the changes.
+-   **Full Ollama Integration**: Seamlessly connects to your local Ollama instance (`http://localhost:11434`) to manage and interact with AI models directly on your machine.
+-   **Intelligent Model Management**:
+    -   Automatically detects if the Ollama service is running.
+    -   Lists all available local models with `enhance --list-models`.
+    -   Auto-selects an optimal model if one is not specified.
+    -   Features resilient retry logic for network requests, ensuring robust communication.
+-   **Automatic Setup**: A simple `enhance --auto-setup` command downloads a recommended model (e.g., `llama3.1:8b`) if no models are found locally, with intelligent fallbacks.
+-   **Rich Terminal UI**: Employs the `rich` library for beautiful, color-coded output, elegant markdown rendering, and informative progress bars during operations.
+-   **Cross-Platform Clipboard**: Automatically copies the final enhanced prompt to your system's clipboard, ready for pasting into your AI interface. Works seamlessly across macOS, Linux, and Windows.
+-   **Highly Configurable**: Control everything from the Ollama host and default styles to temperature and maximum token length via a simple YAML configuration file.
+-   **Robust Testing**: Includes a comprehensive testing suite covering unit, integration, and end-to-end scenarios to ensure reliability and correctness.
+
+## 📦 Installation
+
+`enhance-this` is designed for easy installation across multiple platforms.
+
+**PyPI (Python Package Index)**
+
+```bash
+pip install enhance-this
+```
+
+**NPM (Node.js Package Manager)**
+
+```bash
+npm install -g enhance-this
+```
+
+**Homebrew (macOS & Linux)**
+
+```bash
+brew install hariharen9/tap/enhance-this
+```
+
+**For Local Development**
+
+If you want to contribute or develop locally, clone the repository and install in editable mode:
+
+```bash
+git clone https://github.com/hariharen9/enhance-this.git
+cd enhance-this
+pip install -e .
+```
+
+## 💡 Usage
+
+First, ensure you have [Ollama](https://ollama.com/) installed and running. If you don't have any models downloaded, run:
+
+```bash
+enhance --auto-setup
+```
+
+Then, you can start enhancing your prompts:
+
+**Basic Enhancement:**
+
+```bash
+enhance "write a blog post about AI"
+```
+
+**Using the Diff View:**
+
+```bash
+enhance "review my code" --diff
+```
+
+**Using a Custom Style:**
+
+(First, define your custom style in `~/.enhance-this/config.yaml`)
+
+```bash
+enhance "a logo for a coffee shop" -s my-logo-style
+```
+
+### Command-Line Options
+
+| Option | Short | Description |
+|---|---|---|
+| `<prompt>` | | The initial prompt to enhance. |
+| `--model <MODEL>` | `-m` | Ollama model to use. |
+| `--temperature <T>` | `-t` | Temperature for generation (0.0-2.0). |
+| `--length <LENGTH>` | `-l` | Max tokens for the enhanced prompt. |
+| `--style <STYLE>` | `-s` | Enhancement style (e.g., `detailed`, `creative`). |
+| `--diff` | | Show a diff between the original and enhanced prompt. |
+| `--output <FILE>` | `-o` | Save the enhanced prompt to a file. |
+| `--no-copy` | `-n` | Disable automatic copying to the clipboard. |
+| `--verbose` | `-v` | Enable verbose output. |
+| `--list-models` | | List all available Ollama models. |
+| `--download-model <MODEL>`| | Download a specific model from Ollama. |
+| `--auto-setup` | | Automatically download a recommended model. |
+| `--version` | | Show the application version. |
+| `--help` | `-h` | Show the help message. |
+
+## 🧪 Testing
+
+The project includes a comprehensive test suite to ensure reliability and correctness.
+
+**Running Tests Locally**
+
+To run all tests, navigate to the project root and execute:
+
+```bash
+pytest tests/
+```
+
+**Test Strategy**
+
+-   **Unit Tests**: Verify individual functions and components in isolation.
+-   **Integration Tests**: Ensure different modules and external services (like Ollama, with proper mocking or live checks) work together correctly.
+-   **End-to-End Tests**: Simulate real-world usage scenarios through the CLI to confirm the entire workflow functions as expected.
+
+## 📚 Documentation
+
+-   **[Configuration Guide](./docs/CONFIGURATION.md)**: Learn how to customize every aspect of the tool, including setting up custom templates.
+-   **[Troubleshooting Guide](./docs/TROUBLESHOOTING.md)**: Find solutions to common problems and error messages.
+-   **[Examples](./examples)**: Explore more real-world usage examples and prompt enhancement scenarios.
+
+## 🤝 Contributing
+
+Contributions are welcome! Please feel free to open issues, submit pull requests, or suggest new features. See the [CONTRIBUTING.md](CONTRIBUTING.md) (coming soon) for more details.
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+---
+*This README was generated and enhanced by Gemini based on the project's development specification and implemented features.*

enhance_this-0.1.0/README.md

@@ -0,0 +1,146 @@
+# enhance-this ✨
+
+[![PyPI - Version](https://img.shields.io/pypi/v/enhance-this?style=for-the-badge)](https://pypi.org/project/enhance-this/)
+[![npm - Version](https://img.shields.io/npm/v/enhance-this?style=for-the-badge)](https://www.npmjs.com/package/enhance-this)
+[![Homebrew - Version](https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fformulae.brew.sh%2Fapi%2Fformula%2Fenhance-this.json&query=%24.versions.stable&label=homebrew&style=for-the-badge)](https://formulae.brew.sh/formula/enhance-this)
+[![License](https://img.shields.io/github/license/hariharen9/enhance-this?style=for-the-badge)](LICENSE)
+
+**`enhance-this`** is a powerful, fast, and reliable CLI tool designed to elevate your prompts. It takes a simple idea, enhances it using local Ollama AI models, and delivers a comprehensive, structured prompt ready to generate superior AI responses. The enhanced prompt is streamed directly to your terminal and automatically copied to your clipboard for immediate use.
+
+## 🚀 Overview
+
+The core mission of `enhance-this` is to bridge the gap between a simple user query and a high-quality, detailed prompt that AI models can understand and act upon effectively. By leveraging a sophisticated and customizable templating system, the power of local language models via Ollama, and a rich terminal interface, it transforms basic inputs into professional-grade prompts.
+
+## ✨ Features
+
+-   **Live Streaming Output**: Get instant feedback as the enhanced prompt is generated token by token, providing a dynamic user experience.
+-   **Powerful Prompt Enhancement**: Utilizes a system of style-based templates (`detailed`, `concise`, `creative`, `technical`) to convert simple prompts into comprehensive, actionable ones.
+-   **Fully Customizable Templates**: Extend the tool with your own unique enhancement styles by simply adding paths to your custom template files in `config.yaml`.
+-   **Diff View**: Use the `--diff` flag to see a clear, color-coded comparison between your original prompt and the enhanced version, highlighting the changes.
+-   **Full Ollama Integration**: Seamlessly connects to your local Ollama instance (`http://localhost:11434`) to manage and interact with AI models directly on your machine.
+-   **Intelligent Model Management**:
+    -   Automatically detects if the Ollama service is running.
+    -   Lists all available local models with `enhance --list-models`.
+    -   Auto-selects an optimal model if one is not specified.
+    -   Features resilient retry logic for network requests, ensuring robust communication.
+-   **Automatic Setup**: A simple `enhance --auto-setup` command downloads a recommended model (e.g., `llama3.1:8b`) if no models are found locally, with intelligent fallbacks.
+-   **Rich Terminal UI**: Employs the `rich` library for beautiful, color-coded output, elegant markdown rendering, and informative progress bars during operations.
+-   **Cross-Platform Clipboard**: Automatically copies the final enhanced prompt to your system's clipboard, ready for pasting into your AI interface. Works seamlessly across macOS, Linux, and Windows.
+-   **Highly Configurable**: Control everything from the Ollama host and default styles to temperature and maximum token length via a simple YAML configuration file.
+-   **Robust Testing**: Includes a comprehensive testing suite covering unit, integration, and end-to-end scenarios to ensure reliability and correctness.
+
+## 📦 Installation
+
+`enhance-this` is designed for easy installation across multiple platforms.
+
+**PyPI (Python Package Index)**
+
+```bash
+pip install enhance-this
+```
+
+**NPM (Node.js Package Manager)**
+
+```bash
+npm install -g enhance-this
+```
+
+**Homebrew (macOS & Linux)**
+
+```bash
+brew install hariharen9/tap/enhance-this
+```
+
+**For Local Development**
+
+If you want to contribute or develop locally, clone the repository and install in editable mode:
+
+```bash
+git clone https://github.com/hariharen9/enhance-this.git
+cd enhance-this
+pip install -e .
+```
+
+## 💡 Usage
+
+First, ensure you have [Ollama](https://ollama.com/) installed and running. If you don't have any models downloaded, run:
+
+```bash
+enhance --auto-setup
+```
+
+Then, you can start enhancing your prompts:
+
+**Basic Enhancement:**
+
+```bash
+enhance "write a blog post about AI"
+```
+
+**Using the Diff View:**
+
+```bash
+enhance "review my code" --diff
+```
+
+**Using a Custom Style:**
+
+(First, define your custom style in `~/.enhance-this/config.yaml`)
+
+```bash
+enhance "a logo for a coffee shop" -s my-logo-style
+```
+
+### Command-Line Options
+
+| Option | Short | Description |
+|---|---|---|
+| `<prompt>` | | The initial prompt to enhance. |
+| `--model <MODEL>` | `-m` | Ollama model to use. |
+| `--temperature <T>` | `-t` | Temperature for generation (0.0-2.0). |
+| `--length <LENGTH>` | `-l` | Max tokens for the enhanced prompt. |
+| `--style <STYLE>` | `-s` | Enhancement style (e.g., `detailed`, `creative`). |
+| `--diff` | | Show a diff between the original and enhanced prompt. |
+| `--output <FILE>` | `-o` | Save the enhanced prompt to a file. |
+| `--no-copy` | `-n` | Disable automatic copying to the clipboard. |
+| `--verbose` | `-v` | Enable verbose output. |
+| `--list-models` | | List all available Ollama models. |
+| `--download-model <MODEL>`| | Download a specific model from Ollama. |
+| `--auto-setup` | | Automatically download a recommended model. |
+| `--version` | | Show the application version. |
+| `--help` | `-h` | Show the help message. |
+
+## 🧪 Testing
+
+The project includes a comprehensive test suite to ensure reliability and correctness.
+
+**Running Tests Locally**
+
+To run all tests, navigate to the project root and execute:
+
+```bash
+pytest tests/
+```
+
+**Test Strategy**
+
+-   **Unit Tests**: Verify individual functions and components in isolation.
+-   **Integration Tests**: Ensure different modules and external services (like Ollama, with proper mocking or live checks) work together correctly.
+-   **End-to-End Tests**: Simulate real-world usage scenarios through the CLI to confirm the entire workflow functions as expected.
+
+## 📚 Documentation
+
+-   **[Configuration Guide](./docs/CONFIGURATION.md)**: Learn how to customize every aspect of the tool, including setting up custom templates.
+-   **[Troubleshooting Guide](./docs/TROUBLESHOOTING.md)**: Find solutions to common problems and error messages.
+-   **[Examples](./examples)**: Explore more real-world usage examples and prompt enhancement scenarios.
+
+## 🤝 Contributing
+
+Contributions are welcome! Please feel free to open issues, submit pull requests, or suggest new features. See the [CONTRIBUTING.md](CONTRIBUTING.md) (coming soon) for more details.
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+---
+*This README was generated and enhanced by Gemini based on the project's development specification and implemented features.*

enhance_this-0.1.0/enhance_this/cli.py

@@ -0,0 +1,174 @@
+import click
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.live import Live
+import sys
+import difflib
+
+from .config import load_config, create_default_config_if_not_exists
+from .ollama_client import OllamaClient
+from .enhancer import PromptEnhancer
+from .clipboard import copy_to_clipboard
+
+@click.command()
+@click.argument('prompt', required=False)
+@click.option('-m', '--model', 'model_name', help='Ollama model to use (auto-selects optimal if not specified)')
+@click.option('-t', '--temperature', type=click.FloatRange(0.0, 2.0), help='Temperature for generation (0.0-2.0)')
+@click.option('-l', '--length', 'max_tokens', type=int, help='Max tokens for enhancement')
+@click.option('-c', '--config', 'config_path', type=click.Path(), help='Configuration file path')
+@click.option('-v', '--verbose', is_flag=True, help='Enable verbose output')
+@click.option('-n', '--no-copy', is_flag=True, help="Don't copy to clipboard")
+@click.option('-o', '--output', 'output_file', type=click.File('w'), help='Save enhanced prompt to file')
+@click.option('-s', '--style', type=click.Choice(['detailed', 'concise', 'creative', 'technical']), help='Enhancement style')
+@click.option('--diff', is_flag=True, help='Show a diff between the original and enhanced prompt')
+@click.option('--list-models', is_flag=True, help='List available Ollama models')
+@click.option('--download-model', 'download_model_name', help='Download specific model from Ollama')
+@click.option('--auto-setup', is_flag=True, help='Automatically setup Ollama with optimal model')
+@click.option('--version', is_flag=True, help='Show version information')
+@click.help_option('-h', '--help')
+def main(prompt, model_name, temperature, max_tokens, config_path, verbose, no_copy, output_file, style, diff, list_models, download_model_name, auto_setup, version):
+    """
+    Enhances a simple prompt using Ollama AI models, displays the enhanced version,
+    and automatically copies it to the clipboard.
+    """
+    if version:
+        from importlib.metadata import version as get_version
+        pkg_version = get_version("enhance-this")
+        click.echo(f"enhance-this version {pkg_version}")
+        return
+
+    create_default_config_if_not_exists()
+    config = load_config(config_path)
+    
+    console = Console()
+    client = OllamaClient(host=config['ollama_host'], timeout=config['timeout'])
+
+    if not client.is_running():
+        console.print("[red]✖[/red] Ollama service is not running or is unreachable.")
+        console.print("Please start Ollama and try again.")
+        sys.exit(1)
+
+    if list_models:
+        models = client.list_models()
+        if models:
+            console.print("[bold green]Available Ollama models:[/bold green]")
+            for model in models:
+                console.print(f"- {model}")
+        else:
+            console.print("[yellow]No Ollama models found.[/yellow]")
+        return
+
+    if download_model_name:
+        console.print(f"Starting download for '{download_model_name}'...")
+        client.download_model(download_model_name)
+        return
+        
+    available_models = client.list_models()
+
+    if auto_setup or not available_models:
+        if not available_models:
+            console.print("[yellow]No models found. Starting auto-setup.[/yellow]")
+        else:
+            console.print("Starting auto-setup...")
+        
+        recommended_models = ["llama3.1:8b", "llama3", "mistral"]
+        for model_to_try in recommended_models:
+            if model_to_try not in available_models:
+                console.print(f"Downloading recommended model: {model_to_try}")
+                if client.download_model(model_to_try):
+                    available_models.append(model_to_try)
+                    break 
+            else:
+                console.print(f"Recommended model '{model_to_try}' is already available.")
+                break
+        else:
+            console.print("[red]✖[/red] Auto-setup failed. Could not download a recommended model.")
+            sys.exit(1)
+
+        if auto_setup:
+             return
+
+    if not prompt:
+        ctx = click.get_current_context()
+        click.echo(ctx.get_help())
+        ctx.exit()
+
+    if model_name:
+        if model_name not in available_models:
+            console.print(f"[red]✖[/red] Model '{model_name}' not found. Available models:")
+            for model in available_models:
+                console.print(f"- {model}")
+            sys.exit(1)
+        final_model = model_name
+    else:
+        if not available_models:
+            console.print("[red]✖[/red] No models available. Please download a model first, e.g.:")
+            console.print("`enhance --download-model llama3.1:8b` or run `enhance --auto-setup`")
+            sys.exit(1)
+        final_model = available_models[0]
+        if verbose:
+            console.print(f"No model specified. Using first available model: [cyan]{final_model}[/cyan]")
+
+    final_style = style or config.get('default_style', 'detailed')
+    final_temperature = temperature if temperature is not None else config.get('default_temperature', 0.7)
+    final_max_tokens = max_tokens or config.get('max_tokens', 2000)
+    auto_copy_enabled = not no_copy and config.get('auto_copy', True)
+
+    enhancer = PromptEnhancer(config.get('enhancement_templates'))
+    system_prompt = enhancer.enhance(prompt, final_style)
+
+    if verbose:
+        console.print("\n[bold]System Prompt:[/bold]")
+        console.print(Panel(system_prompt, title="System Prompt", border_style="dim"))
+
+    enhanced_prompt = ""
+    console.print("\n[bold magenta]✨ Enhanced Prompt ✨[/bold magenta]")
+
+    status_obj = console.status("[bold green]Enhancing prompt...")
+    status_obj.start()
+
+    try:
+        stream_generator = client.generate_stream(final_model, system_prompt, final_temperature, final_max_tokens)
+
+        # Stop the status as soon as the generator is ready
+        status_obj.stop()
+
+        with Live(console=console, auto_refresh=False) as live:
+            for chunk in stream_generator:
+                enhanced_prompt += chunk
+                live.update(Markdown(enhanced_prompt), refresh=True)
+    finally:
+        status_obj.stop()
+
+    if enhanced_prompt:
+        if diff:
+            console.print("\n[bold yellow]↔️  Diff View ↔️[/bold yellow]")
+            diff_result = difflib.unified_diff(
+                prompt.splitlines(keepends=True),
+                enhanced_prompt.splitlines(keepends=True),
+                fromfile='Original',
+                tofile='Enhanced',
+            )
+            for line in diff_result:
+                if line.startswith('+'):
+                    console.print(f"[green]{line}[/green]", end="")
+                elif line.startswith('-'):
+                    console.print(f"[red]{line}[/red]", end="")
+                elif line.startswith('@'):
+                    console.print(f"[dim]{line}[/dim]", end="")
+                else:
+                    console.print(line, end="")
+
+        if output_file:
+            output_file.write(enhanced_prompt)
+            console.print(f"\n[green]✔[/green] Saved to {output_file.name}")
+
+        if auto_copy_enabled:
+            copy_to_clipboard(enhanced_prompt)
+    else:
+        console.print("\n[red]✖[/red] Failed to generate enhanced prompt.")
+        sys.exit(1)
+
+if __name__ == '__main__':
+    main()

enhance_this-0.1.0/enhance_this/clipboard.py

@@ -0,0 +1,12 @@
+import pyperclip
+from rich.console import Console
+
+console = Console()
+
+def copy_to_clipboard(text: str):
+    """Copies the given text to the clipboard."""
+    try:
+        pyperclip.copy(text)
+        console.print("[green]✔[/green] Enhanced prompt copied to clipboard.")
+    except pyperclip.PyperclipException:
+        console.print("[yellow]⚠[/yellow] Could not copy to clipboard. `xclip` or `xsel` may be required on Linux.")

enhance_this-0.1.0/enhance_this/config.py

@@ -0,0 +1,53 @@
+import os
+import yaml
+from pathlib import Path
+from typing import Dict, Any, Optional
+
+DEFAULT_CONFIG = {
+    "default_temperature": 0.7,
+    "default_style": "detailed",
+    "ollama_host": "http://localhost:11434",
+    "timeout": 30,
+    "max_tokens": 2000,
+    "auto_copy": True,
+    "display_colors": True,
+    "auto_download_model": True,
+    "enhancement_templates": {},
+}
+
+def get_config_path(config_path_str: Optional[str] = None) -> Path:
+    if config_path_str:
+        return Path(config_path_str)
+    return Path.home() / ".enhance-this" / "config.yaml"
+
+def load_config(config_path_str: Optional[str] = None) -> Dict[str, Any]:
+    config_path = get_config_path(config_path_str)
+    if not config_path.exists():
+        return DEFAULT_CONFIG
+    
+    try:
+        with open(config_path, 'r') as f:
+            user_config = yaml.safe_load(f)
+        
+        config = DEFAULT_CONFIG.copy()
+        if user_config:
+            config.update(user_config)
+        return config
+    except (yaml.YAMLError, IOError):
+        return DEFAULT_CONFIG
+
+def ensure_config_dir_exists():
+    config_path = get_config_path()
+    config_path.parent.mkdir(parents=True, exist_ok=True)
+
+def create_default_config_if_not_exists():
+    ensure_config_dir_exists()
+    config_path = get_config_path()
+    if not config_path.exists():
+        config_to_write = DEFAULT_CONFIG.copy()
+        # Example for user
+        config_to_write['enhancement_templates'] = {
+            'my_style': '/path/to/your/custom_template.txt'
+        }
+        with open(config_path, 'w') as f:
+            yaml.dump(config_to_write, f, default_flow_style=False, sort_keys=False)

enhance_this-0.1.0/enhance_this/enhancer.py

@@ -0,0 +1,58 @@
+import importlib.resources
+from typing import Dict, Optional
+from pathlib import Path
+from rich.console import Console
+
+console = Console()
+
+def load_templates(custom_template_paths: Optional[Dict[str, str]] = None) -> Dict[str, str]:
+    templates = {}
+    package = 'enhance_this.templates'
+    
+    # Load built-in templates
+    try:
+        from importlib.resources import files
+    except ImportError:
+        from importlib_resources import files
+
+    try:
+        template_dir = files(package)
+        for style in ["detailed", "concise", "creative", "technical"]:
+            template_file = template_dir / f"{style}.txt"
+            if template_file.is_file():
+                 templates[style] = template_file.read_text(encoding='utf-8')
+    except Exception:
+        for style in ["detailed", "concise", "creative", "technical"]:
+            try:
+                with importlib.resources.open_text(package, f"{style}.txt") as f:
+                    templates[style] = f.read()
+            except FileNotFoundError:
+                pass
+
+    # Load custom templates from config
+    if custom_template_paths:
+        for style, path_str in custom_template_paths.items():
+            if not isinstance(path_str, str) or not path_str.strip():
+                continue # Skip if path is not a valid string
+            try:
+                path = Path(path_str).expanduser()
+                if path.is_file():
+                    templates[style] = path.read_text(encoding='utf-8')
+                else:
+                    console.print(f"[yellow]⚠[/yellow] Custom template for style '{style}' not found at: {path_str}")
+            except Exception as e:
+                console.print(f"[red]✖[/red] Error loading custom template for style '{style}': {e}")
+
+    return templates
+
+class PromptEnhancer:
+    def __init__(self, custom_template_paths: Optional[Dict[str, str]] = None):
+        self.templates = load_templates(custom_template_paths)
+
+    def enhance(self, user_prompt: str, style: str) -> str:
+        if style not in self.templates:
+            available_styles = list(self.templates.keys())
+            raise ValueError(f"Unknown style: '{style}'. Available styles: {available_styles}")
+        
+        template = self.templates[style]
+        return template.format(user_prompt=user_prompt)