deploying to gradio

README.md

@@ -1,3 +1,9 @@
+---
+title: stable-diffusion-image-generator
+app_file: src/sdgen/main.py
+sdk: gradio
+sdk_version: 3.50.2
+---
 # 🎨 Stable Diffusion Image Generator
 
 AI system built using **Stable Diffusion (HuggingFace Diffusers)** and a modern **Gradio UI**.

requirements.txt

@@ -35,6 +35,4 @@ realesrgan-ncnn-py==2.0.0
 # DEVELOPMENT TOOLS
 black==24.3.0
 ruff==0.3.5
-pre-commit==3.7.0
-
--e .
+pre-commit==3.7.0

src/sdgen/config/__init__.py

@@ -1,7 +1,7 @@
 """Configuration exports for the sdgen package.
 
 This module re-exports commonly used configuration paths and settings
-so they can be imported directly from `sdgen.config`.
+so they can be imported directly from `src.sdgen.config`.
 """
 
 from __future__ import annotations

src/sdgen/config/paths.py

@@ -1,4 +1,4 @@
-"""Path configuration for sdgen.
+"""Path configuration for src.sdgen.
 
 All filesystem paths are resolved relative to the project root.
 The project root is detected by walking upward until a marker

src/sdgen/config/settings.py

@@ -1,4 +1,4 @@
-"""Application runtime settings for sdgen.
+"""Application runtime settings for src.sdgen.
 
 AppSettings reads configuration values from environment variables at
 process start and exposes them as strongly typed attributes.

src/sdgen/main.py

@@ -9,11 +9,11 @@ from __future__ import annotations
 import torch
 from dotenv import load_dotenv
 
-from sdgen.config import AppSettings
-from sdgen.sd.img2img import prepare_img2img_pipeline
-from sdgen.sd.pipeline import load_pipeline, warmup_pipeline
-from sdgen.ui import build_ui
-from sdgen.utils.logger import get_logger
+from src.sdgen.config import AppSettings
+from src.sdgen.sd.img2img import prepare_img2img_pipeline
+from src.sdgen.sd.pipeline import load_pipeline, warmup_pipeline
+from src.sdgen.ui import build_ui
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 load_dotenv()

src/sdgen/sd/generator.py

@@ -8,9 +8,9 @@ from typing import Tuple
 import torch
 from PIL import Image
 
-from sdgen.sd.models import GenerationMetadata, Txt2ImgConfig
-from sdgen.utils.common import validate_resolution
-from sdgen.utils.logger import get_logger
+from src.sdgen.sd.models import GenerationMetadata, Txt2ImgConfig
+from src.sdgen.utils.common import validate_resolution
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/sd/img2img.py

@@ -8,9 +8,9 @@ import torch
 from diffusers import StableDiffusionImg2ImgPipeline
 from PIL import Image
 
-from sdgen.sd.models import GenerationMetadata, Img2ImgConfig
-from sdgen.utils.common import validate_resolution
-from sdgen.utils.logger import get_logger
+from src.sdgen.sd.models import GenerationMetadata, Img2ImgConfig
+from src.sdgen.utils.common import validate_resolution
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/sd/pipeline.py

@@ -12,7 +12,7 @@ from diffusers import (
     StableDiffusionPipeline,
 )
 
-from sdgen.utils.logger import get_logger
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/ui/__init__.py

@@ -1,5 +1,5 @@
 from __future__ import annotations
 
-from sdgen.ui.layout import build_ui
+from src.sdgen.ui.layout import build_ui
 
 __all__ = ["build_ui"]

src/sdgen/ui/layout.py

@@ -6,20 +6,20 @@ from typing import Any, Dict, Tuple
 
 import gradio as gr
 
-from sdgen.sd.generator import generate_image
-from sdgen.sd.img2img import generate_img2img
-from sdgen.sd.models import Img2ImgConfig, Txt2ImgConfig
-from sdgen.ui.tabs import (
+from src.sdgen.sd.generator import generate_image
+from src.sdgen.sd.img2img import generate_img2img
+from src.sdgen.sd.models import Img2ImgConfig, Txt2ImgConfig
+from src.sdgen.ui.tabs import (
     build_history_tab,
     build_img2img_tab,
     build_presets_tab,
     build_txt2img_tab,
     build_upscaler_tab,
 )
-from sdgen.upscaler.upscaler import Upscaler
-from sdgen.utils.common import pretty_json, to_pil
-from sdgen.utils.history import save_history_entry
-from sdgen.utils.logger import get_logger
+from src.sdgen.upscaler.upscaler import Upscaler
+from src.sdgen.utils.common import pretty_json, to_pil
+from src.sdgen.utils.history import save_history_entry
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/ui/tabs/history_tab.py

@@ -7,13 +7,13 @@ from typing import Any, Dict, List, Optional, Tuple
 import gradio as gr
 from PIL import Image
 
-from sdgen.utils.common import short_prompt
-from sdgen.utils.history import (
+from src.sdgen.utils.common import short_prompt
+from src.sdgen.utils.history import (
     delete_history_entry,
     list_history,
     load_entry,
 )
-from sdgen.utils.logger import get_logger
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/ui/tabs/presets_tab.py

@@ -6,9 +6,9 @@ from typing import Any, Tuple
 
 import gradio as gr
 
-from sdgen.presets.styles import get_preset, list_presets
-from sdgen.ui.tabs.img2img_tab import Img2ImgControls
-from sdgen.ui.tabs.txt2img_tab import Txt2ImgControls
+from src.sdgen.presets.styles import get_preset, list_presets
+from src.sdgen.ui.tabs.img2img_tab import Img2ImgControls
+from src.sdgen.ui.tabs.txt2img_tab import Txt2ImgControls
 
 
 def apply_preset(preset_name: Any) -> Tuple[Any, ...]:

src/sdgen/upscaler/realesrgan.py

@@ -12,7 +12,7 @@ from typing import Final
 from PIL import Image
 from realesrgan_ncnn_py import Realesrgan
 
-from sdgen.utils.logger import get_logger
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/upscaler/upscaler.py

@@ -14,8 +14,8 @@ from typing import Optional
 
 from PIL import Image
 
-from sdgen.upscaler.realesrgan import NCNNUpscaler
-from sdgen.utils.logger import get_logger
+from src.sdgen.upscaler.realesrgan import NCNNUpscaler
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/utils/history.py

@@ -18,14 +18,14 @@ from typing import Any, Dict, List, Optional, Tuple
 
 from PIL import Image
 
-from sdgen.config import (
+from src.sdgen.config import (
     HISTORY_ENTRIES_DIR,
     HISTORY_FULL_DIR,
     HISTORY_ROOT,
     HISTORY_THUMBS_DIR,
 )
-from sdgen.sd.models import GenerationMetadata, HistorySummary
-from sdgen.utils.logger import get_logger
+from src.sdgen.sd.models import GenerationMetadata, HistorySummary
+from src.sdgen.utils.logger import get_logger
 
 logger = get_logger(__name__)
 

src/sdgen/utils/logger.py

@@ -10,7 +10,7 @@ import logging
 from logging import Handler, Logger
 from logging.handlers import RotatingFileHandler
 
-from sdgen.config import LOGS_ROOT
+from src.sdgen.config import LOGS_ROOT
 
 # Ensure logs directory exists
 LOGS_ROOT.mkdir(parents=True, exist_ok=True)