Update smoke test, run script, and Makefile

- smoke_test.py now uses GenerationResult and displays seed/LoRA status
- run.sh simplified with proper bash error handling
- Makefile cleaned up with standard phony targets

Makefile

@@ -1,33 +1,13 @@
-.PHONY: help install run test clean
-
-help:
-	@echo ">� BrickStyle-Gen - Makefile Commands"
-	@echo ""
-	@echo "  make install    - Install dependencies"
-	@echo "  make run        - Run the Gradio app"
-	@echo "  make test       - Run smoke test"
-	@echo "  make clean      - Clean generated files and cache"
-	@echo ""
+.PHONY: install run smoke clean
 
 install:
-	@echo "=� Installing dependencies..."
 	pip install -r requirements.txt
-	@echo " Installation complete"
 
 run:
-	@echo "=� Launching BrickStyle-Gen..."
-	@echo "   App will be available at: http://localhost:7860"
-	@echo ""
 	python app.py
 
-test:
-	@echo ">� Running smoke test..."
+smoke:
 	python smoke_test.py
 
 clean:
-	@echo ">� Cleaning up..."
-	rm -rf __pycache__ pipelines/__pycache__ research/__pycache__
-	rm -rf out/*.png out/*.jpg
-	rm -rf .pytest_cache
-	rm -rf *.pyc
-	@echo " Cleanup complete"
+	rm -rf __pycache__ */__pycache__ .pytest_cache out/smoke.png

run.sh

@@ -1,39 +1,11 @@
-#!/bin/bash
-# BrickStyle-Gen Launcher Script
+#!/usr/bin/env bash
+set -euo pipefail
 
-set -e
-
-echo ">� BrickStyle-Gen Launcher"
-echo "=========================="
-echo ""
-
-# Check if virtual environment exists
-if [ ! -d "venv" ]; then
-    echo "Virtual environment not found. Creating..."
-    python3 -m venv venv
-    echo " Virtual environment created"
-fi
-
-# Activate virtual environment
-echo "Activating virtual environment..."
-source venv/bin/activate
-
-# Check if dependencies are installed
-if ! python -c "import gradio" 2>/dev/null; then
-    echo "Dependencies not installed. Installing..."
-    pip install -r requirements.txt
-    echo " Dependencies installed"
+if [[ -f ".venv/bin/activate" ]]; then
+  source .venv/bin/activate
 fi
 
-# Create output directory
-mkdir -p out
-
-# Launch app
-echo ""
-echo "=� Launching BrickStyle-Gen..."
-echo "   App will be available at: http://localhost:7860"
-echo ""
-echo "   Press Ctrl+C to stop"
-echo ""
+export GRADIO_SERVER_NAME=${GRADIO_SERVER_NAME:-0.0.0.0}
+export GRADIO_SERVER_PORT=${GRADIO_SERVER_PORT:-7860}
 
 python app.py

smoke_test.py

@@ -1,99 +1,40 @@
-#!/usr/bin/env python3
-"""
-Smoke Test for BrickStyle-Gen
-Quick validation that the pipeline can generate images.
+"""Minimal smoke test for the BrickStyle-Gen text pipeline."""
 
-Usage:
-    python smoke_test.py
-
-This will:
-1. Load the SDXL + LoRA model
-2. Generate one 768x768 test image
-3. Save to out/smoke_test.png
-4. Report timing and success/failure
-"""
-
-import os
-import time
 from pathlib import Path
 
-print("=" * 60)
-print(">� BrickStyle-Gen Smoke Test")
-print("=" * 60)
-print()
+from pipelines.sdxl_lego_lora import load_model
 
-# Create output directory
-out_dir = Path("out")
-out_dir.mkdir(exist_ok=True)
 
-print("Step 1/3: Importing dependencies...")
-try:
-    from pipelines.sdxl_lego_lora import load_model
-    import torch
-    print(f"   PyTorch {torch.__version__}")
-    print(f"   Device available: {torch.cuda.is_available() and 'CUDA' or torch.backends.mps.is_available() and 'MPS' or 'CPU'}")
-except ImportError as e:
-    print(f"  L Import failed: {e}")
-    print("\n  Please install dependencies:")
-    print("  pip install -r requirements.txt")
-    exit(1)
+def main() -> None:
+    print("=" * 60)
+    print("BrickStyle-Gen Smoke Test")
+    print("=" * 60)
+    print()
 
-print()
-print("Step 2/3: Loading model (this may take a few minutes on first run)...")
-start_load = time.time()
-try:
     generator = load_model()
-    load_time = time.time() - start_load
-    print(f"   Model loaded in {load_time:.1f}s")
-except Exception as e:
-    print(f"  L Model loading failed: {e}")
-    exit(1)
-
-print()
-print("Step 3/3: Generating test image...")
-print("  Prompt: 'a tiny brick castle on a hill'")
-print("  Settings: 768x768, 25 steps, seed=42")
-print()
-
-start_gen = time.time()
-try:
-    images = generator.generate(
-        prompt="a tiny brick castle on a hill",
-        negative_prompt="text, watermark, deformed",
-        num_inference_steps=25,
-        guidance_scale=7.5,
-        seed=42,
-        width=768,
-        height=768,
+    result = generator.generate(
+        prompt="brick-style micro city skyline, studs visible",
+        negative_prompt="text, watermark, logo",
+        num_inference_steps=10,
+        guidance_scale=7.0,
+        width=512,
+        height=512,
         batch_size=1,
+        lora_scale=1.0,
     )
 
-    gen_time = time.time() - start_gen
-
-    # Save image
-    output_path = out_dir / "smoke_test.png"
-    images[0].save(output_path)
+    out_dir = Path("out")
+    out_dir.mkdir(exist_ok=True)
+    output_path = out_dir / "smoke.png"
+    result.images[0].save(output_path)
 
-    print(f"   Generation complete in {gen_time:.1f}s")
-    print(f"   Image saved to: {output_path}")
+    print()
+    print("=" * 60)
+    print(f"Smoke test image saved to {output_path}")
+    print(f"Seed: {result.seed}")
+    print(f"LoRA applied: {result.lora_applied}")
+    print("=" * 60)
 
-except Exception as e:
-    print(f"  L Generation failed: {e}")
-    import traceback
-    traceback.print_exc()
-    exit(1)
 
-print()
-print("=" * 60)
-print(" SMOKE TEST PASSED")
-print("=" * 60)
-print()
-print(f"Total time: {time.time() - start_load:.1f}s")
-print(f"Model load: {load_time:.1f}s")
-print(f"Generation: {gen_time:.1f}s")
-print()
-print("Next steps:")
-print("  1. Open out/smoke_test.png to verify brick-style")
-print("  2. Run the full app: python app.py")
-print("  3. Visit http://localhost:7860")
-print()
+if __name__ == "__main__":
+    main()