StableDesign2 / README.md
anbucur
Added functionality
4e4b650
---
title: StableDesign
emoji: 🏆
colorFrom: yellow
colorTo: gray
sdk: gradio
sdk_version: 4.26.0
app_file: app.py
pinned: false
license: mit
---
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
# Interior Design Assistant
An AI-powered interior design assistant that helps transform room photos with customizable design variations.
## Features
- Upload room photos and generate design variations
- Customize room type, style, color schemes
- Adjust floor and wall finishes
- Add wall decorations (art, mirrors, sconces)
- Control generation parameters (quality, creativity, etc.)
- Save results to Google Drive
- Production and test modes
## Requirements
- Python 3.8 or higher
- CUDA-capable GPU with 8GB+ VRAM (recommended)
- CPU-only mode supported but slower
## Installation
1. Clone the repository:
```bash
git clone [your-repo-url]
cd StableDesign2
```
2. Create and activate a virtual environment:
```bash
python -m venv venv
source venv/bin/activate # Linux/Mac
# or
.\venv\Scripts\activate # Windows
```
3. Install dependencies:
```bash
pip install -r requirements.txt
```
4. Set up Google Drive integration (optional):
- Create a project in Google Cloud Console
- Enable the Google Drive API
- Create OAuth 2.0 credentials
- Download credentials and save as `credentials.json` in the project root
## Usage
### Production Mode
```bash
python app.py
```
### Test Mode (for development)
```bash
python app.py --test
```
The interface will be available at `http://localhost:7860`
## Configuration
### Model Settings
- Quality Steps: 20-100 (default: 50)
- Design Freedom: 1-20 (default: 7.5)
- Change Amount: 0.1-1.0 (default: 0.75)
- Number of Variations: 1-4
### Design Options
- Room Types: 16 options
- Design Styles: 20 options
- Color Schemes: 20 options
- Floor & Wall Options: Multiple materials, colors, patterns
- Wall Decorations: Art, mirrors, sconces, shelves, plants
## Error Handling
The application includes comprehensive error handling:
- Input validation
- Model generation fallbacks
- Google Drive upload retries
- Detailed error logging
## Development
### Running Tests
```bash
python -m pytest
```
### Code Style
```bash
black .
flake8
isort .
```
## License
[Your License]
## Credits
- Built with [Gradio](https://gradio.app/)
- Powered by [Stable Diffusion](https://stability.ai/)