Hugging Face's logo

Spaces:
fastrabbit
/
StableDesign2
Running

App Files Community
StableDesign2
File size: 2,391 Bytes 
22ede16
 
 
 
 
 
6369e62
22ede16
 
 
 
 
 
4e4b650
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
---
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/)