pyproject.toml

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "interiorfusion"
version = "0.1.0"
description = "Single 2D image to high-quality editable 3D interior scene generation"
readme = "README.md"
license = {text = "MIT"}
requires-python = ">=3.10"
authors = [
    {name = "InteriorFusion Research Team"}
]
keywords = [
    "3d-generation",
    "interior-design",
    "computer-vision",
    "gaussian-splatting",
    "mesh-reconstruction",
    "diffusion-models",
    "pbr-materials",
    "scene-understanding",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
    "torch>=2.1.0",
    "torchvision>=0.16.0",
    "diffusers>=0.27.0",
    "transformers>=4.40.0",
    "accelerate>=0.28.0",
    "datasets>=2.17.0",
    "pillow>=10.0.0",
    "numpy>=1.24.0",
    "scipy>=1.11.0",
    "trimesh>=4.0.0",
    "pygltflib>=1.16.0",
    "open3d>=0.18.0",
    "opencv-python>=4.8.0",
    "scikit-image>=0.22.0",
    "imageio>=2.31.0",
    "imageio-ffmpeg>=0.4.9",
    "tqdm>=4.66.0",
    "einops>=0.7.0",
    "omegaconf>=2.3.0",
    "wandb>=0.16.0",
    "tensorboard>=2.15.0",
    "lpips>=0.1.4",
    "xatlas>=0.0.8",
    "fastapi>=0.110.0",
    "uvicorn>=0.27.0",
    "gradio>=4.20.0",
    "httpx>=0.27.0",
    "pydantic>=2.6.0",
    "rich>=13.7.0",
    "typer>=0.12.0",
    "safetensors>=0.4.2",
    "huggingface-hub>=0.22.0",
    "peft>=0.10.0",
    "bitsandbytes>=0.42.0",
    "kernels>=0.0.2",
    "trackio>=0.4.0",
    "segment-anything>=1.0",
    "rembg>=2.0.0",
    "kiui>=1.0.0",
    "nerfacc>=0.5.3",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
    "pytest-cov>=4.1.0",
    "black>=24.1.0",
    "ruff>=0.2.0",
    "mypy>=1.8.0",
    "pre-commit>=3.6.0",
]
gs = [
    "gsplat>=0.1.0",
    "nerfstudio>=1.0.0",
]
comfy = [
    "comfyui>=0.2.0",
]

[project.urls]
Homepage = "https://huggingface.co/stevee00/InteriorFusion"
Repository = "https://github.com/stevee00/InteriorFusion"
Documentation = "https://huggingface.co/spaces/stevee00/InteriorFusion-Docs"

[project.scripts]
interiorfusion-infer = "interiorfusion.cli:infer_cli"
interiorfusion-train = "interiorfusion.cli:train_cli"
interiorfusion-api = "interiorfusion.api.main:main"

[tool.setuptools.packages.find]
where = ["src"]

[tool.setuptools.package-data]
interiorfusion = ["configs/*.yaml", "configs/*.json"]