Merge pull request #207 from vietanhdev/feature/dark-light-theme

Implement dark/light theme

Author: vietanhdev

README.md

@@ -143,3 +143,4 @@ If you want to contribute to **AnyLabeling**, please read [Contribution Guidelin
 - Labeling UI built with ideas and components from [LabelImg](https://github.com/heartexlabs/labelImg), [LabelMe](https://github.com/wkentaro/labelme).
 - Auto-labeling with [Segment Anything Models](https://segment-anything.com/), [MobileSAM](https://github.com/ChaoningZhang/MobileSAM).
 - Auto-labeling with [YOLOv8](https://github.com/ultralytics/ultralytics).
+- Icons from FlatIcon: [DinosoftLabs](https://www.flaticon.com/free-icons/sun "sun icons"), [Freepik](https://www.flaticon.com/free-icons/moon "moon icons"), [Vectoricons](https://www.flaticon.com/free-icons/system "system icons"), [HideMaru](https://www.flaticon.com/free-icons/ungroup "ungroup icons").

anylabeling/app.py

@@ -22,6 +22,7 @@
 from anylabeling.views.labeling.logger import logger
 from anylabeling.views.labeling.utils import new_icon
 from anylabeling.resources import resources
+from anylabeling.styles import AppTheme
 
 __all__ = ["resources"]
 
@@ -111,6 +112,13 @@ def main():
         help="epsilon to find nearest vertex on canvas",
         default=argparse.SUPPRESS,
     )
+    # Add theme argument
+    parser.add_argument(
+        "--theme",
+        choices=["system", "light", "dark"],
+        help="set application theme (default: system)",
+        default="system",
+    )
     args = parser.parse_args()
 
     logger.setLevel(getattr(logging, args.logger_level.upper()))
@@ -141,6 +149,7 @@ def main():
     filename = config_from_args.pop("filename")
     output = config_from_args.pop("output")
     config_file_or_yaml = config_from_args.pop("config")
+    theme = config_from_args.pop("theme")
     anylabeling_config.current_config_file = config_file_or_yaml
     config = get_config(config_file_or_yaml, config_from_args)
 
@@ -176,6 +185,19 @@ def main():
     app = QtWidgets.QApplication(sys.argv)
     app.processEvents()
 
+    # Apply theme
+    if theme != "system":
+        # Override system theme detection
+        os.environ["DARK_MODE"] = "1" if theme == "dark" else "0"
+    else:
+        # Check if theme is in config
+        config_theme = config.get("theme", "system")
+        if config_theme != "system":
+            os.environ["DARK_MODE"] = "1" if config_theme == "dark" else "0"
+
+    # Apply our modern theme
+    AppTheme.apply_theme(app)
+
     app.setApplicationName(__appname__)
     app.setWindowIcon(new_icon("icon"))
     if loaded_language:

anylabeling/config.py

@@ -21,6 +21,10 @@ def update_dict(target_dict, new_dict, validate_item=None):
     for key, value in new_dict.items():
         if validate_item:
             validate_item(key, value)
+        # Special handling for recognized new keys
+        if key not in target_dict and key in ["theme"]:
+            target_dict[key] = value
+            continue
         if key not in target_dict:
             logger.warning("Skipping unexpected key in config: %s", key)
             continue
@@ -59,6 +63,8 @@ def validate_config_item(key, value):
         raise ValueError(f"Unexpected value for config key 'shape_color': {value}")
     if key == "labels" and value is not None and len(value) != len(set(value)):
         raise ValueError(f"Duplicates are detected for config key 'labels': {value}")
+    if key == "theme" and value not in ["system", "light", "dark"]:
+        raise ValueError(f"Unexpected value for config key 'theme': {value}")
 
 
 def get_config(config_file_or_yaml=None, config_from_args=None):

anylabeling/configs/anylabeling_config.yaml

@@ -1,4 +1,5 @@
 language: en_US
+theme: system
 auto_save: true
 display_label_popup: true
 store_data: false