Upload 2 files

57/models.txt

@@ -0,0 +1,62 @@
+# Mapping of folder names to optimization target descriptions
+OPTIMIZATION_TARGETS = {
+    "0": "OPTIMIZATION_TARGET_UNKNOWN",
+    "1": "OPTIMIZATION_TARGET_PAINFUL_PAGE_LOAD",
+    "2": "OPTIMIZATION_TARGET_LANGUAGE_DETECTION",
+    "3": "OPTIMIZATION_TARGET_PAGE_TOPICS",
+    "4": "OPTIMIZATION_TARGET_SEGMENTATION_NEW_TAB",
+    "5": "OPTIMIZATION_TARGET_SEGMENTATION_SHARE",
+    "6": "OPTIMIZATION_TARGET_SEGMENTATION_VOICE",
+    "7": "OPTIMIZATION_TARGET_MODEL_VALIDATION",
+    "8": "OPTIMIZATION_TARGET_PAGE_ENTITIES",
+    "9": "OPTIMIZATION_TARGET_NOTIFICATION_PERMISSION_PREDICTIONS",
+    "10": "OPTIMIZATION_TARGET_SEGMENTATION_DUMMY",
+    "11": "OPTIMIZATION_TARGET_SEGMENTATION_CHROME_START_ANDROID",
+    "12": "OPTIMIZATION_TARGET_SEGMENTATION_QUERY_TILES",
+    "13": "OPTIMIZATION_TARGET_PAGE_VISIBILITY",
+    "15": "OPTIMIZATION_TARGET_PAGE_TOPICS_V2",
+    "16": "OPTIMIZATION_TARGET_SEGMENTATION_CHROME_LOW_USER_ENGAGEMENT",
+    "17": "OPTIMIZATION_TARGET_SEGMENTATION_FEED_USER",
+    "18": "OPTIMIZATION_TARGET_CONTEXTUAL_PAGE_ACTION_PRICE_TRACKING",
+    "19": "OPTIMIZATION_TARGET_TEXT_CLASSIFIER",
+    "20": "OPTIMIZATION_TARGET_GEOLOCATION_PERMISSION_PREDICTIONS",
+    "21": "OPTIMIZATION_TARGET_SEGMENTATION_SHOPPING_USER",
+    "22": "OPTIMIZATION_TARGET_SEGMENTATION_CHROME_START_ANDROID_V2",
+    "23": "OPTIMIZATION_TARGET_SEGMENTATION_SEARCH_USER",
+    "24": "OPTIMIZATION_TARGET_OMNIBOX_ON_DEVICE_TAIL_SUGGEST",
+    "25": "OPTIMIZATION_TARGET_CLIENT_SIDE_PHISHING",
+    "26": "OPTIMIZATION_TARGET_OMNIBOX_URL_SCORING",
+    "27": "OPTIMIZATION_TARGET_SEGMENTATION_DEVICE_SWITCHER",
+    "28": "OPTIMIZATION_TARGET_SEGMENTATION_ADAPTIVE_TOOLBAR",
+    "29": "OPTIMIZATION_TARGET_SEGMENTATION_TABLET_PRODUCTIVITY_USER",
+    "30": "OPTIMIZATION_TARGET_CLIENT_SIDE_PHISHING_IMAGE_EMBEDDER",
+    "31": "OPTIMIZATION_TARGET_NEW_TAB_PAGE_HISTORY_CLUSTERS_MODULE_RANKING",
+    "32": "OPTIMIZATION_TARGET_WEB_APP_INSTALLATION_PROMO",
+    "33": "OPTIMIZATION_TARGET_TEXT_EMBEDDER",
+    "34": "OPTIMIZATION_TARGET_VISUAL_SEARCH_CLASSIFICATION",
+    "35": "OPTIMIZATION_TARGET_SEGMENTATION_BOTTOM_TOOLBAR",
+    "36": "OPTIMIZATION_TARGET_AUTOFILL_FIELD_CLASSIFICATION",
+    "37": "OPTIMIZATION_TARGET_SEGMENTATION_IOS_MODULE_RANKER",
+    "38": "OPTIMIZATION_TARGET_SEGMENTATION_DESKTOP_NTP_MODULE",
+    "39": "OPTIMIZATION_TARGET_PRELOADING_HEURISTICS",
+    "40": "OPTIMIZATION_TARGET_TEXT_SAFETY",
+    "41": "OPTIMIZATION_TARGET_SEGMENTATION_ANDROID_HOME_MODULE_RANKER",
+    "42": "OPTIMIZATION_TARGET_COMPOSE",
+    "43": "OPTIMIZATION_TARGET_PASSAGE_EMBEDDER",
+    "44": "OPTIMIZATION_TARGET_PHRASE_SEGMENTATION",
+    "45": "OPTIMIZATION_TARGET_SEGMENTATION_COMPOSE_PROMOTION",
+    "46": "OPTIMIZATION_TARGET_URL_VISIT_RESUMPTION_RANKER",
+    "47": "OPTIMIZATION_TARGET_CAMERA_BACKGROUND_SEGMENTATION",
+    "48": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_HISTORY_SEARCH",
+    "49": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_PROMPT_API",
+    "50": "OPTIMIZATION_TARGET_SEGMENTATION_METRICS_CLUSTERING",
+    "51": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_SUMMARIZE",
+    "52": "OPTIMIZATION_TARGET_PASSWORD_MANAGER_FORM_CLASSIFICATION",
+    "53": "OPTIMIZATION_TARGET_NOTIFICATION_CONTENT_DETECTION",
+    "54": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_HISTORY_QUERY_INTENT",
+    "55": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_SCAM_DETECTION",
+    "56": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_PERMISSIONS_AI",
+    "57": "OPTIMIZATION_TARGET_EXPERIMENTAL_EMBEDDER",
+    "58": "OPTIMIZATION_TARGET_SEGMENTATION_FEDCM_USER",
+    "59": "OPTIMIZATION_TARGET_MODEL_EXECUTION_FEATURE_WRITING_ASSISTANCE_API"
+}

57/watcher.py

@@ -0,0 +1,135 @@
+#!/usr/bin/env python3
+import time
+import sys
+import os
+import json
+from datetime import datetime
+from watchdog.observers import Observer
+from watchdog.events import FileSystemEventHandler
+
+# ANSI color codes
+class Colors:
+    RED = '\033[91m'
+    GREEN = '\033[92m'
+    ORANGE = '\033[93m'
+    RESET = '\033[0m'
+
+# Load optimization target mappings from models.txt
+def load_optimization_targets(filename="models.txt"):
+    with open(filename, "r") as file:
+        content = file.read()
+        start_index = content.find("{")
+        end_index = content.rfind("}") + 1
+        json_data = content[start_index:end_index]
+        return {k: v.replace("OPTIMIZATION_TARGET_", "") for k, v in json.loads(json_data).items()}
+
+OPTIMIZATION_TARGETS = load_optimization_targets()
+
+class MyHandler(FileSystemEventHandler):
+    def __init__(self, log_file, base_path):
+        self.log_file = log_file
+        self.base_path = base_path
+        self.suppress_log = log_file  # Avoid infinite loop
+
+    def get_target_descriptions(self, event_path):
+        """ Extracts all matching optimization targets from the path. """
+        folder_names = set(event_path.split(os.sep))  # Get all path components
+        matched_targets = [
+            (folder, OPTIMIZATION_TARGETS[folder])
+            for folder in folder_names if folder in OPTIMIZATION_TARGETS
+        ]
+        return matched_targets if matched_targets else [("Unknown", "[Unknown]")]
+
+    def format_path(self, path):
+        """ Returns relative path from base directory. """
+        return f".{path.replace(self.base_path, '')}" if path.startswith(self.base_path) else path
+
+    def get_event_color(self, event_type):
+        """ Returns the appropriate color for each event type """
+        if "DELETED" in event_type:
+            return Colors.RED
+        elif "NEW" in event_type:
+            return Colors.GREEN
+        else:
+            return Colors.ORANGE
+
+    def log_event(self, event_type, event_path):
+        if os.path.abspath(event_path) == os.path.abspath(self.suppress_log):
+            return  # Ignore log file modifications
+
+        matched_targets = self.get_target_descriptions(event_path)
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        formatted_path = self.format_path(event_path)
+        color = self.get_event_color(event_type)
+
+        # Log each event separately in a stacked format
+        log_entries = []
+        colored_entries = []
+        
+        for folder, target in matched_targets:
+            base_entry = (
+                f"------------------------------\n"
+                f"Timestamp: {timestamp}\n"
+                f"Event: {event_type}\n"
+                f"Folder: {folder}\n"
+                f"Optimization Target: {target}\n"
+                f"Path: {formatted_path}\n"
+                f"------------------------------\n"
+            )
+            
+            # Create colored version for console
+            colored_entry = (
+                f"------------------------------\n"
+                f"Timestamp: {timestamp}\n"
+                f"Event: {color}{event_type}{Colors.RESET}\n"
+                f"Folder: {folder}\n"
+                f"Optimization Target: {target}\n"
+                f"Path: {formatted_path}\n"
+                f"------------------------------\n"
+            )
+            
+            log_entries.append(base_entry)
+            colored_entries.append(colored_entry)
+
+        log_message = "\n".join(log_entries)
+        colored_message = "\n".join(colored_entries)
+
+        print(colored_message)  # Print colored version to console
+        with open(self.log_file, "a") as f:
+            f.write(log_message + "\n")  # Write uncolored version to file
+
+    def on_created(self, event):
+        what = "NEW FOLDER" if event.is_directory else "NEW FILE"
+        self.log_event(what, event.src_path)
+
+    def on_deleted(self, event):
+        what = "FOLDER DELETED" if event.is_directory else "FILE DELETED"
+        self.log_event(what, event.src_path)
+
+    def on_modified(self, event):
+        what = "FOLDER MODIFIED" if event.is_directory else "FILE MODIFIED"
+        self.log_event(what, event.src_path)
+
+    def on_moved(self, event):
+        what = "FOLDER MOVED" if event.is_directory else "FILE MOVED"
+        self.log_event(f"{what}: from {self.format_path(event.src_path)} to {self.format_path(event.dest_path)}", event.src_path)
+
+if __name__ == "__main__":
+    path = sys.argv[1] if len(sys.argv) > 1 else os.getcwd()
+    log_file = "log.txt"
+
+    event_handler = MyHandler(log_file, path)
+    observer = Observer()
+    observer.schedule(event_handler, path, recursive=True)
+    observer.start()
+    
+    print(f"Monitoring folder: {os.path.abspath(path)}")
+    with open(log_file, "a") as f:
+        f.write(f"Monitoring folder: {os.path.abspath(path)}\n")
+    
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        observer.stop()
+    observer.join()