Daugreen/ai-interior-designer

AI Interior Designer - HuggingFace Spaces

Ứng dụng AI thiết kế nội thất sử dụng Stable Diffusion và ControlNet.

Tính năng

• —Virtual Staging: Tạo thiết kế nội thất từ ảnh căn phòng trống
• —Virtual Renovation: Thay thế đồ đạc cũ bằng đồ mới
• —Day to Dusk: Chuyển đổi ảnh từ ngày sang hoàng hôn
• —Item Removal: Xóa vật thể khỏi ảnh
• —Image Enhancement: Tăng cường chất lượng ảnh

API Endpoints

• —GET / - Trang chủ
• —GET /health - Kiểm tra sức khỏe
• —GET /api/health - Health check chi tiết
• —POST /api/upload-room - Upload ảnh căn phòng
• —POST /api/generate-design - Tạo thiết kế nội thất
• —POST /api/generate-renovation - Tạo virtual renovation
• —POST /api/day-to-dusk - Chuyển đổi day to dusk
• —POST /api/remove-items - Xóa vật thể
• —POST /api/enhance-image - Tăng cường chất lượng ảnh
• —GET /api/styles - Lấy danh sách styles
• —GET /api/result/<filename> - Lấy ảnh kết quả

Cách sử dụng

1. 1.Upload ảnh căn phòng qua endpoint /api/upload-room
2. 2.Chọn style thiết kế (modern, classic, scandinavian, industrial, luxury, chinese)
3. 3.Gọi endpoint /api/generate-design để tạo thiết kế
4. 4.Nhận kết quả dưới dạng base64 image

Models sử dụng

• —Stable Diffusion v1.4 với ControlNet Canny
• —Stable Diffusion Inpainting
• —YOLO v8 cho phát hiện đồ đạc
• —LaMa Cleaner cho xóa vật thể
• —GFPGAN cho tăng cường khuôn mặt
• —RealESRGAN cho tăng cường chất lượng ảnh

Cấu hình

• —Port: 7860
• —Thư mục tạm: /tmp/uploads, /tmp/results
• —Sử dụng CUDA nếu có sẵn, fallback về CPU

Deployment

Ứng dụng được deploy trên HuggingFace Spaces với Docker container.