ollamadiffuser 1.0.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

ollamadiffuser-1.1.1.dist-info/METADATA

@@ -0,0 +1,470 @@
+Metadata-Version: 2.4
+Name: ollamadiffuser
+Version: 1.1.1
+Summary: 🎨 Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX.1, and LoRA support
+Home-page: https://github.com/ollamadiffuser/ollamadiffuser
+Author: OllamaDiffuser Team
+Author-email: OllamaDiffuser Team <ollamadiffuser@gmail.com>
+License: MIT
+Project-URL: Homepage, https://www.ollamadiffuser.com/
+Project-URL: Website, https://www.ollamadiffuser.com/
+Project-URL: Repository, https://github.com/ollamadiffuser/ollamadiffuser
+Project-URL: Issues, https://github.com/ollamadiffuser/ollamadiffuser/issues
+Project-URL: Documentation, https://www.ollamadiffuser.com/
+Project-URL: Bug Reports, https://github.com/ollamadiffuser/ollamadiffuser/issues
+Project-URL: Feature Requests, https://github.com/ollamadiffuser/ollamadiffuser/issues
+Project-URL: Source Code, https://github.com/ollamadiffuser/ollamadiffuser
+Keywords: diffusion,image-generation,ai,machine-learning,lora,ollama,stable-diffusion,flux,local-ai,controlnet,web-ui,cli
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: End Users/Desktop
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Multimedia :: Graphics
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Environment :: Console
+Classifier: Environment :: Web Environment
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.1.0
+Requires-Dist: diffusers>=0.26.0
+Requires-Dist: transformers>=4.35.0
+Requires-Dist: accelerate>=0.25.0
+Requires-Dist: fastapi>=0.104.0
+Requires-Dist: uvicorn>=0.23.0
+Requires-Dist: huggingface-hub>=0.16.0
+Requires-Dist: Pillow>=9.0.0
+Requires-Dist: click>=8.0.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: protobuf>=3.20.0
+Requires-Dist: sentencepiece>=0.1.99
+Requires-Dist: safetensors>=0.3.0
+Requires-Dist: python-multipart>=0.0.0
+Requires-Dist: psutil>=5.9.0
+Requires-Dist: jinja2>=3.0.0
+Requires-Dist: peft>=0.10.0
+Requires-Dist: controlnet-aux>=0.0.7
+Requires-Dist: opencv-python>=4.8.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: isort>=5.12.0; extra == "dev"
+Requires-Dist: flake8>=6.0.0; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# OllamaDiffuser 🎨
+
+[![PyPI version](https://badge.fury.io/py/ollamadiffuser.svg)](https://badge.fury.io/py/ollamadiffuser)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+
+
+## Local AI Image Generation with OllamaDiffuser
+
+**OllamaDiffuser** simplifies local deployment of **Stable Diffusion**, **FLUX.1**, and other AI image generation models. An intuitive **local SD** tool inspired by **Ollama's** simplicity - perfect for **local diffuser** workflows with CLI, web UI, and LoRA support.
+
+🌐 **Website**: [ollamadiffuser.com](https://www.ollamadiffuser.com/) | 📦 **PyPI**: [pypi.org/project/ollamadiffuser](https://pypi.org/project/ollamadiffuser/)
+
+---
+
+## ✨ Features
+
+- **🚀 Fast Startup**: Instant application launch with lazy loading architecture
+- **🎛️ ControlNet Support**: Precise image generation control with 10+ control types
+- **🔄 LoRA Integration**: Dynamic LoRA loading and management
+- **🌐 Multiple Interfaces**: CLI, Python API, Web UI, and REST API
+- **📦 Model Management**: Easy installation and switching between models
+- **⚡ Performance Optimized**: Memory-efficient with GPU acceleration
+- **🎨 Professional Results**: High-quality image generation with fine-tuned control
+
+## 🚀 Quick Start
+
+### Option 1: Install from PyPI (Recommended)
+```bash
+# Install from PyPI
+pip install ollamadiffuser
+
+# Pull and run a model (4-command setup)
+ollamadiffuser pull flux.1-schnell
+ollamadiffuser run flux.1-schnell
+
+# Generate via API
+curl -X POST http://localhost:8000/api/generate \
+  -H "Content-Type: application/json" \
+  -d '{"prompt": "A beautiful sunset"}' \
+  --output image.png
+```
+
+### Option 2: Development Installation
+```bash
+# Clone the repository
+git clone https://github.com/ollamadiffuser/ollamadiffuser.git
+cd ollamadiffuser
+
+# Install dependencies
+pip install -e .
+```
+
+### Basic Usage
+```bash
+# Install a model
+ollamadiffuser pull stable-diffusion-1.5
+
+# Run the model (loads and starts API server)
+ollamadiffuser run stable-diffusion-1.5
+
+# Generate an image via API
+curl -X POST http://localhost:8000/api/generate \
+  -H "Content-Type: application/json" \
+  -d '{"prompt": "a beautiful sunset over mountains"}' \
+  --output image.png
+
+# Start web interface
+ollamadiffuser --mode ui
+
+open http://localhost:8001 in your browser
+```
+
+### ControlNet Quick Start
+```bash
+# Install ControlNet model
+ollamadiffuser pull controlnet-canny-sd15
+
+# Run ControlNet model (loads and starts API server)
+ollamadiffuser run controlnet-canny-sd15
+
+# Generate with control image
+curl -X POST http://localhost:8000/api/generate/controlnet \
+  -F "prompt=a beautiful landscape" \
+  -F "control_image=@your_image.jpg"
+```
+
+---
+
+## 🎯 Supported Models
+
+Choose from a variety of state-of-the-art image generation models:
+
+| Model | License | Quality | Speed | Commercial Use |
+|-------|---------|---------|-------|----------------|
+| **FLUX.1-schnell** | Apache 2.0 | High | **4 steps** (12x faster) | ✅ Commercial OK |
+| **FLUX.1-dev** | Non-commercial | High | 50 steps | ❌ Non-commercial |
+| **Stable Diffusion 3.5** | CreativeML | Medium | 28 steps | ⚠️ Check License |
+| **Stable Diffusion 1.5** | CreativeML | Fast | Lightweight | ⚠️ Check License |
+
+### Why Choose FLUX.1-schnell?
+- **Apache 2.0 license** - Perfect for commercial use
+- **4-step generation** - Lightning fast results  
+- **Commercial OK** - Use in your business
+
+---
+
+## 🎛️ ControlNet Features
+
+### ⚡ Lazy Loading Architecture
+**New in v1.1.0**: ControlNet preprocessors use intelligent lazy loading:
+
+- **Instant Startup**: `ollamadiffuser --help` runs immediately without downloading models
+- **On-Demand Loading**: Preprocessors initialize only when actually needed
+- **Automatic Initialization**: Seamless loading when uploading control images
+- **User Control**: Manual initialization available for pre-loading
+
+### Available Control Types
+- **Canny Edge Detection**: Structural control with edge maps
+- **Depth Estimation**: 3D structure control with depth maps
+- **OpenPose**: Human pose and body position control
+- **Scribble/Sketch**: Artistic control with hand-drawn inputs
+- **Advanced Types**: HED, MLSD, Normal, Lineart, Anime Lineart, Content Shuffle
+
+### ControlNet Models
+```bash
+# SD 1.5 ControlNet Models
+ollamadiffuser pull controlnet-canny-sd15
+ollamadiffuser pull controlnet-depth-sd15
+ollamadiffuser pull controlnet-openpose-sd15
+ollamadiffuser pull controlnet-scribble-sd15
+
+# SDXL ControlNet Models
+ollamadiffuser pull controlnet-canny-sdxl
+ollamadiffuser pull controlnet-depth-sdxl
+```
+
+## 🔄 LoRA Support
+
+### Dynamic LoRA Management
+```bash
+# Download LoRA from Hugging Face
+ollamadiffuser lora pull "openfree/flux-chatgpt-ghibli-lora"
+
+# Load LoRA with custom strength
+ollamadiffuser lora load ghibli --scale 1.2
+
+# Unload LoRA
+ollamadiffuser lora unload
+```
+
+### Web UI LoRA Integration
+- **Easy Download**: Enter Hugging Face repository ID
+- **Strength Control**: Adjust LoRA influence with sliders
+- **Real-time Loading**: Load/unload LoRAs without restarting
+- **Alias Support**: Create custom names for your LoRAs
+
+## 🌐 Multiple Interfaces
+
+### Command Line Interface
+```bash
+# Pull and run a model
+ollamadiffuser pull stable-diffusion-1.5
+ollamadiffuser run stable-diffusion-1.5
+
+# In another terminal, generate images via API
+curl -X POST http://localhost:8000/api/generate \
+  -H "Content-Type: application/json" \
+  -d '{
+    "prompt": "a futuristic cityscape",
+    "negative_prompt": "blurry, low quality",
+    "num_inference_steps": 30,
+    "guidance_scale": 7.5,
+    "width": 1024,
+    "height": 1024
+  }' \
+  --output image.png
+```
+
+### Web UI
+```bash
+# Start web interface
+ollamadiffuser --mode ui
+Open http://localhost:8001
+```
+
+Features:
+- **Responsive Design**: Works on desktop and mobile
+- **Real-time Status**: Model and LoRA loading indicators
+- **ControlNet Integration**: File upload with preprocessing
+- **Parameter Controls**: Intuitive sliders and inputs
+
+### REST API
+```bash
+# Start API server
+ollamadiffuser --mode api
+
+ollamadiffuser load stable-diffusion-1.5
+
+# Generate image
+curl -X POST http://localhost:8000/api/generate \
+  -H "Content-Type: application/json" \
+  -d '{"prompt": "a beautiful landscape", "width": 1024, "height": 1024}'
+```
+
+### Python API
+```python
+from ollamadiffuser.core.models.manager import model_manager
+
+# Load model
+success = model_manager.load_model("stable-diffusion-1.5")
+if success:
+    engine = model_manager.loaded_model
+    
+    # Generate image
+    image = engine.generate_image(
+        prompt="a beautiful sunset",
+        width=1024,
+        height=1024
+    )
+    image.save("output.jpg")
+else:
+    print("Failed to load model")
+```
+
+## 📦 Supported Models
+
+### Base Models
+- **Stable Diffusion 1.5**: Classic, reliable, fast
+- **Stable Diffusion XL**: High-resolution, detailed
+- **Stable Diffusion 3**: Latest architecture
+- **FLUX.1**: State-of-the-art quality
+
+### ControlNet Models
+- **SD 1.5 ControlNet**: 4 control types (canny, depth, openpose, scribble)
+- **SDXL ControlNet**: 2 control types (canny, depth)
+
+### LoRA Support
+- **Hugging Face Integration**: Direct download from HF Hub
+- **Local LoRA Files**: Support for local .safetensors files
+- **Dynamic Loading**: Load/unload without model restart
+- **Strength Control**: Adjustable influence (0.1-2.0)
+
+## ⚙️ Configuration
+
+### Model Configuration
+Models are automatically configured with optimal settings:
+- **Memory Optimization**: Attention slicing, CPU offloading
+- **Device Detection**: Automatic CUDA/MPS/CPU selection
+- **Precision Handling**: FP16/BF16 support for efficiency
+- **Safety Features**: NSFW filter bypass for creative freedom
+
+## 🔧 Advanced Usage
+
+### ControlNet Parameters
+```python
+# Fine-tune ControlNet behavior
+image = engine.generate_image(
+    prompt="architectural masterpiece",
+    control_image=control_img,
+    controlnet_conditioning_scale=1.2,  # Strength (0.0-2.0)
+    control_guidance_start=0.0,         # When to start (0.0-1.0)
+    control_guidance_end=1.0            # When to end (0.0-1.0)
+)
+```
+
+### Batch Processing
+```python
+from ollamadiffuser.core.utils.controlnet_preprocessors import controlnet_preprocessor
+
+# Pre-initialize for faster processing
+controlnet_preprocessor.initialize()
+
+# Process multiple images
+prompt = "beautiful landscape"  # Define the prompt
+for i, image_path in enumerate(image_list):
+    control_img = controlnet_preprocessor.preprocess(image_path, "canny")
+    result = engine.generate_image(prompt, control_image=control_img)
+    result.save(f"output_{i}.jpg")
+```
+
+### API Integration
+```python
+import requests
+
+# Initialize ControlNet preprocessors
+response = requests.post("http://localhost:8000/api/controlnet/initialize")
+
+# Check available preprocessors
+response = requests.get("http://localhost:8000/api/controlnet/preprocessors")
+print(response.json()["available_types"])
+
+# Generate with file upload
+with open("control.jpg", "rb") as f:
+    response = requests.post(
+        "http://localhost:8000/api/generate/controlnet",
+        data={"prompt": "beautiful landscape"},
+        files={"control_image": f}
+    )
+```
+
+## 📚 Documentation & Guides
+
+- **[ControlNet Guide](CONTROLNET_GUIDE.md)**: Comprehensive ControlNet usage and examples
+- **[Website Documentation](https://www.ollamadiffuser.com/)**: Complete tutorials and guides
+
+## 🚀 Performance & Hardware
+
+### Minimum Requirements
+- **RAM**: 8GB system RAM
+- **Storage**: 10GB free space
+- **Python**: 3.8+
+
+### Recommended Hardware
+- **GPU**: 8GB+ VRAM (NVIDIA/AMD)
+- **RAM**: 16GB+ system RAM
+- **Storage**: SSD with 50GB+ free space
+
+### Supported Platforms
+- **CUDA**: NVIDIA GPUs (recommended)
+- **MPS**: Apple Silicon (M1/M2/M3)
+- **CPU**: All platforms (slower but functional)
+
+## 🔧 Troubleshooting
+
+### Common Issues
+
+#### Slow Startup
+If you experience slow startup, ensure you're using the latest version with lazy loading:
+```bash
+git pull origin main
+pip install -e .
+```
+
+#### ControlNet Not Working
+```bash
+# Check preprocessor status
+python -c "
+from ollamadiffuser.core.utils.controlnet_preprocessors import controlnet_preprocessor
+print('Available:', controlnet_preprocessor.is_available())
+print('Initialized:', controlnet_preprocessor.is_initialized())
+"
+
+# Manual initialization
+curl -X POST http://localhost:8000/api/controlnet/initialize
+```
+
+#### Memory Issues
+```bash
+# Use smaller image sizes via API
+curl -X POST http://localhost:8000/api/generate \
+  -H "Content-Type: application/json" \
+  -d '{"prompt": "test", "width": 512, "height": 512}' \
+  --output test.png
+
+# CPU offloading is automatic
+# Close other applications to free memory
+# Use basic preprocessors instead of advanced ones
+```
+
+### Debug Mode
+```bash
+# Enable verbose logging
+ollamadiffuser --verbose run model-name
+```
+
+## 🤝 Contributing
+
+We welcome contributions! Please check the GitHub repository for contribution guidelines.
+
+## 🤝 Community & Support
+
+### Quick Actions
+
+- **🐛 [Report a Bug](https://github.com/ollamadiffuser/ollamadiffuser/issues)** - Found an issue? Let us know
+- **💡 [Feature Request](https://github.com/ollamadiffuser/ollamadiffuser/issues)** - Have an idea? Share it with us  
+- **💬 [Join Discussions](https://github.com/ollamadiffuser/ollamadiffuser/discussions)** - Community discussion
+- **⭐ [Star on GitHub](https://github.com/ollamadiffuser/ollamadiffuser)** - Show your support
+
+### Community Driven
+
+OllamaDiffuser is an open-source project that thrives on community feedback. Every suggestion, bug report, and contribution helps make it better for everyone.
+
+**Open Source** • **Community Driven** • **Actively Maintained**
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+## 🙏 Acknowledgments
+
+- **Stability AI**: For Stable Diffusion models
+- **Hugging Face**: For model hosting and diffusers library
+- **ControlNet Team**: For ControlNet architecture
+- **Community**: For feedback and contributions
+
+## 📞 Support
+
+- **Issues**: [GitHub Issues](https://github.com/ollamadiffuser/ollamadiffuser/issues)
+- **Discussions**: [GitHub Discussions](https://github.com/ollamadiffuser/ollamadiffuser/discussions)
+
+---
+
+**Ready to get started?** Install from PyPI: `pip install ollamadiffuser` or visit [ollamadiffuser.com](https://www.ollamadiffuser.com/) 🎨✨ 

{ollamadiffuser-1.0.0.dist-info → ollamadiffuser-1.1.1.dist-info}/RECORD

@@ -1,26 +1,27 @@
 ollamadiffuser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ollamadiffuser/__main__.py,sha256=-3WVAex0P3PgfIUtOhVof9nF4Zor8mvMNTK5nKFlrHU,1841
 ollamadiffuser/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser/api/server.py,sha256=b0syFJryp6WFe47HMmRcmEa1NOG4hz54Asm8LolsGeA,10561
+ollamadiffuser/api/server.py,sha256=4-3gT8W1404bxvJ7y9htvKbd2yxrrbtAUvT7shOlJss,17679
 ollamadiffuser/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser/cli/main.py,sha256=ufZCk8iU6Vcaa0US8JaBDuYQJAFWHwrZ83wCaX7Z930,24640
+ollamadiffuser/cli/main.py,sha256=Ck_JGfD1o2eSR8dNcK8Bw0ZRIeBym4eCSND2ADDe-Ao,38722
 ollamadiffuser/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ollamadiffuser/core/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ollamadiffuser/core/config/settings.py,sha256=VhI1vLGmOAQ7-XtyHrT5KoMpcGeGt-Mij-9NxX_ZKsI,4881
 ollamadiffuser/core/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser/core/inference/engine.py,sha256=_j3m2rETeM36PI7zORzBKlMS9I-8crdtqSUtL7Yi2lI,48969
+ollamadiffuser/core/inference/engine.py,sha256=ky76lAjWexlrgmHSZZILa3FPQP7xx2WQ0_DdW34M9Xk,57808
 ollamadiffuser/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser/core/models/manager.py,sha256=mumzPLpq0ZNrTL8TV2t5tcStNmVYlfG8RzSev7raR5w,18299
+ollamadiffuser/core/models/manager.py,sha256=vO1Az_aO5lZKMgSyK_6j2wT5nzPMowZgKhcH2mQVLkI,24139
 ollamadiffuser/core/utils/__init__.py,sha256=ZdXZWX1hfDnnV6OmRD6UStNljDJIQ892da2CtC-zdDw,31
-ollamadiffuser/core/utils/download_utils.py,sha256=giABX3tWTGkFdScVbA589GVr4CsQHUPPZYbbBrdefhA,13667
+ollamadiffuser/core/utils/controlnet_preprocessors.py,sha256=JcLxvnuYAJdQb9EM1mAgyye1nARHwyjFdWzyl7yh7So,12684
+ollamadiffuser/core/utils/download_utils.py,sha256=DvCt-cjH6WSBJniJT112b4a9AUzlwOYhQtPuEfISmtM,20961
 ollamadiffuser/core/utils/lora_manager.py,sha256=SrZydPSGJqCS_Vek35bEdG2Q51qCOLZmPvnNzUjjIN0,14328
 ollamadiffuser/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser/ui/web.py,sha256=q47bUdHydVZ0xw5qqS4dqGqrf3XWbZRiBAO7oRB8tXE,11647
-ollamadiffuser/ui/templates/index.html,sha256=OpblXpIsVTGe-pXtaupupV_QZ34plZKX9XJsL7Gnbpc,18449
+ollamadiffuser/ui/web.py,sha256=xZ5Ja47B-51LRyadfC-gW_aE_B3D571RgpQX0RDzVxM,15290
+ollamadiffuser/ui/templates/index.html,sha256=qTQVFxiTbeZ90O-iNqWC_4pYP6yyIs2z6U69VJPqAB4,38176
 ollamadiffuser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser-1.0.0.dist-info/licenses/LICENSE,sha256=cnGL9l2P510Uk3TCnv62kot6vAfdSawhOZh7Y-oYoIE,1071
-ollamadiffuser-1.0.0.dist-info/METADATA,sha256=3z3kKud6T-AgPYqwBNc4buDLGyQX_O0Gc5jL4YnQRf4,14690
-ollamadiffuser-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ollamadiffuser-1.0.0.dist-info/entry_points.txt,sha256=tHXXO3N0GSnIobDe_eSOLfHPjjVFjeTg2Fd-APoD6sY,64
-ollamadiffuser-1.0.0.dist-info/top_level.txt,sha256=97wOGgTCxDE765Nr_o7B4Kwr_M_jy8fCCeQ81sMKlC4,15
-ollamadiffuser-1.0.0.dist-info/RECORD,,
+ollamadiffuser-1.1.1.dist-info/licenses/LICENSE,sha256=cnGL9l2P510Uk3TCnv62kot6vAfdSawhOZh7Y-oYoIE,1071
+ollamadiffuser-1.1.1.dist-info/METADATA,sha256=ZQOmzyft8wVpr3oZ6WczxfjUIu0G_m7_rzJ_g68hawY,15353
+ollamadiffuser-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ollamadiffuser-1.1.1.dist-info/entry_points.txt,sha256=tHXXO3N0GSnIobDe_eSOLfHPjjVFjeTg2Fd-APoD6sY,64
+ollamadiffuser-1.1.1.dist-info/top_level.txt,sha256=97wOGgTCxDE765Nr_o7B4Kwr_M_jy8fCCeQ81sMKlC4,15
+ollamadiffuser-1.1.1.dist-info/RECORD,,