Update processing/two_stage/__init__.py

processing/two_stage/__init__.py

@@ -1,5 +1,18 @@
-# /home/user/app/processing/two_stage/__init__.py
-# package marker for the two_stage subpackage — no side-effects
-from .two_stage_processor import TwoStageProcessor, CHROMA_PRESETS
+# processing/two_stage/__init__.py
 
-__all__ = ["TwoStageProcessor", "CHROMA_PRESETS"]
+from .two_stage_processor import TwoStageProcessor
+
+# Import CHROMA_PRESETS only if it exists, otherwise define a fallback
+try:
+    from .two_stage_processor import CHROMA_PRESETS
+except ImportError:
+    # Fallback CHROMA_PRESETS if not defined in two_stage_processor
+    CHROMA_PRESETS = {
+        "office": {"name": "Professional Office", "color": (240, 240, 240)},
+        "blue": {"name": "Studio Blue", "color": (255, 200, 100)},
+        "studio": {"name": "Neutral Studio", "color": (200, 200, 200)},
+        "white": {"name": "Pure White", "color": (255, 255, 255)},
+        "black": {"name": "Deep Black", "color": (0, 0, 0)}
+    }
+
+__all__ = ["TwoStageProcessor", "CHROMA_PRESETS"]