• Decreased cache time

• Increased album art quality (40->80)

README.md

@@ -8,7 +8,7 @@ Simply run the application, and a local web server will run in the background. B
 [http://127.0.0.1:5000/now-playing/](http://127.0.0.1:5000/now-playing/)
 
 A ready-to-use "Now Playing" widget utilizing SMTC Bridge is available to try here:<br>
-**[https://widgets.nutty.gg/now-playing/settings/](https://widgets.nutty.gg/now-playing/settings/)**
+**[IF YOU FOUND THIS PAGE EARLY, DON'T SHARE IT WITH ANYONE YET OR I'LL FUCKING CUM](https://widgets.nutty.gg/now-playing/settings/)**
 
 ## Quick Start
 1. Download the latest `.exe` from the [Releases page](releases).

smtc-bridge.pyw

@@ -1,5 +1,5 @@
 # Versioning
-APP_VERSION = "0.0.4"
+APP_VERSION = "0.0.5"
 DEVELOPER = "nutty"
 
 
@@ -63,6 +63,7 @@ try:
     import webbrowser
     import configparser
     import time
+    import platform
     from PIL import Image
     from flask import Flask, jsonify
     from flask_cors import CORS
@@ -159,28 +160,66 @@ try:
     ### CORE FUNCTIONS ###
     ######################
 
+    # Cache variables for the SMTC manager and rate limiting
+    smtc_manager = None
+    last_execution_time = 0.0
+    last_payload = None
+    album_art_cache = {}
+
     async def get_all_media_info():            
         try:
+            # We will cache the last execution time and payload to avoid redundant parsing if requests flood in faster than 0.5 seconds.
+            current_time = time.time()
+            global smtc_manager, last_execution_time, last_payload, album_art_cache
+
+            # If requests flood in faster than 0.5 seconds, return the cached result 
+            # to completely spare the CPU from redundant parsing.
+            if (current_time - last_execution_time) < 0.5 and last_payload:
+                # print("Using cached media info.")
+                return last_payload
+            # else:
+            #     print("Fetching fresh media info.")
+            
+            last_execution_time = current_time
+
             # Instantiate the SMTC manager -> This allows use to "talk" to the Windows Media API
-            manager = await SMTC.request_async()
+            # Reuse the manager if we already have it, otherwise request it once
+            if not smtc_manager:
+                print("Instantiating new SMTC manager...")
+                smtc_manager = await SMTC.request_async()
+
+            manager = smtc_manager
             
             # If it returns null, then no media is playing, or something fucked up and I have no
             # idea what to do, so just return an empty session list
             if not manager:
                 return {"current_session_id": None, "sessions": []}
 
-            # This is the session for the current media player -> Whatever Windows deems is "in focus"
-            # will be the current session.
+            # current_focused:  This is the session for the current media player -> Whatever Windows deems is "in focus"
+            #                   will be the current session.
+            # all_sessions:     We will also get all sessions, not just the current session.
+            #                   This will provide the client with all the necessary info if they want to target just one application.
             # We will store the SourceAppUserModelId, which we all add to the final payload.
             # For all available properties/methods/events, see the official docs:
             # https://learn.microsoft.com/en-us/uwp/api/windows.media.control.globalsystemmediatransportcontrolssession?view=winrt-28000
-            current_focused = manager.get_current_session()
+            
+            # Try to pull sessions using the cached manager
+            try:
+                current_focused = manager.get_current_session()
+                all_sessions = manager.get_sessions()
+            except Exception:
+                # If the COM context dropped or invalidated, reset it and retry once
+                print("Instantiating new SMTC manager...")
+                smtc_manager = await SMTC.request_async()
+                manager = smtc_manager
+                if not manager:
+                    return {"current_session_id": None, "sessions": []}
+                current_focused = manager.get_current_session()
+                all_sessions = manager.get_sessions()
+
             current_session_id = current_focused.source_app_user_model_id if current_focused else None
 
-            # We will also get all sessions, not just the current session.
-            # This will provide the client with all the necessary info if they want to target just
-            # one application.
-            all_sessions = manager.get_sessions()
+            # We will store all the session info in a list of dictionaries, which we will return as JSON
             sessions_list = []
 
             # We will not iterate over all the sessions, and grab all the available info
@@ -271,51 +310,52 @@ try:
                     try:
                         stream_ref = raw_media.thumbnail
                         stream = await stream_ref.open_read_async()
-                        reader = DataReader(stream.get_input_stream_at(0))
-                        await reader.load_async(stream.size)
-                        buffer = bytearray(stream.size)
-                        reader.read_bytes(buffer)
-                        
-                        # Generate a safe filename based on the app_id
-                        safe_app_id = "".join(c for c in app_id if c.isalnum() or c in ('_', '-'))
-                        thumb_filename = f"{safe_app_id}.jpg"
-                        thumb_path = os.path.join(THUMB_DIR, thumb_filename)
-                        
-                        # Hash the raw image bytes to check if the artwork actually changed
-                        import hashlib
-                        img_hash = hashlib.md5(buffer).hexdigest()
-                        
-                        # Track hashes in memory to avoid writing to disk if nothing changed
-                        if not hasattr(get_all_media_info, "cache"):
-                            get_all_media_info.cache = {}
-                        
-                        # Check if we already processed this exact artwork for this app
-                        if get_all_media_info.cache.get(safe_app_id) == img_hash and os.path.exists(thumb_path):
-                            # Artwork hasn't changed, reuse existing file and version timestamp
-                            pass
-                        else:
-                            # Process and save with Pillow only when bytes actually change
-                            img = Image.open(io.BytesIO(buffer))
-                            if img.mode in ("RGBA", "P"):
-                                img = img.convert("RGB")
 
-                            # Save at full resolution, but use faster compression settings
-                            img.save(
-                                thumb_path, 
-                                "JPEG", 
-                                quality=40,          # Sweet spot for small size / high visual fidelity
-                                subsampling=2,       # Faster compression algorithm for low-end CPUs
-                                optimize=False       # Skips the extra CPU pass
-                            )
+                        # Only proceed if there's actual data to read
+                        if stream.size > 0:
+                            reader = DataReader(stream.get_input_stream_at(0))
+                            await reader.load_async(stream.size)
+                            buffer = bytearray(stream.size)
+                            reader.read_bytes(buffer)
                             
-                            # Update cache hash
-                            get_all_media_info.cache[safe_app_id] = img_hash
-                        
-                        # Use file modification time as the version token so browsers cache it aggressively
-                        thumb_version = int(os.path.getmtime(thumb_path) * 1000)
-                        
-                        # Construct the URL with the version query parameter
-                        thumb_url = f"http://{HOST}:{PORT}/artwork/{safe_app_id}?v={thumb_version}"
+                            # Generate a safe filename based on the app_id
+                            safe_app_id = "".join(c for c in app_id if c.isalnum() or c in ('_', '-'))
+                            thumb_filename = f"{safe_app_id}.jpg"
+                            thumb_path = os.path.join(THUMB_DIR, thumb_filename)
+                            
+                            # Hash the raw image bytes to check if the artwork actually changed
+                            import hashlib
+                            img_hash = hashlib.md5(buffer).hexdigest()
+                            
+                            # Check if we already processed this exact artwork for this app
+                            if album_art_cache.get(safe_app_id) == img_hash and os.path.exists(thumb_path):
+                                # Artwork hasn't changed, reuse existing file and version timestamp
+                                # print("Using cached artwork.")
+                                pass
+                            else:
+                                print(f"Processing new artwork for: {media_data['Artist']} - {media_data['Title']}")
+                                # Process and save with Pillow only when bytes actually change
+                                img = Image.open(io.BytesIO(buffer))
+                                if img.mode in ("RGBA", "P"):
+                                    img = img.convert("RGB")
+
+                                # Save at full resolution, but use faster compression settings
+                                img.save(
+                                    thumb_path, 
+                                    "JPEG", 
+                                    quality=80,          # Sweet spot for small size / high visual fidelity
+                                    subsampling=2,       # Faster compression algorithm for low-end CPUs
+                                    optimize=False       # Skips the extra CPU pass
+                                )
+                                
+                                # Update cache hash
+                                album_art_cache[safe_app_id] = img_hash
+                            
+                            # Use file modification time as the version token so browsers cache it aggressively
+                            thumb_version = int(os.path.getmtime(thumb_path) * 1000)
+                            
+                            # Construct the URL with the version query parameter
+                            thumb_url = f"http://{HOST}:{PORT}/artwork/{safe_app_id}?v={thumb_version}"
                     except Exception:
                         thumb_url = None
 
@@ -331,16 +371,71 @@ try:
                 })
             
             # Build the final payload
-            return {
+            payload = {
                 "app_version": APP_VERSION,
+                "os": f"{platform.system()} {platform.release()}",
                 "current_session_id": current_session_id, 
                 "sessions": sessions_list
             }
+            
+            # Save to global payload cache
+            last_payload = payload
+            return payload
+
         except Exception as e:
             return {"current_session_id": None, "sessions": [], "error": str(e)}
 
 
 
+    ########################
+    ### STARTUP SHORTCUT ###
+    ########################
+
+    def get_startup_shortcut_path():
+        # Gets the path to the current user's Startup folder
+        startup_dir = os.path.join(os.environ['APPDATA'], r'Microsoft\Windows\Start Menu\Programs\Startup')
+        return os.path.join(startup_dir, 'SMTCBridge.lnk')
+
+    def is_start_with_windows():
+        return os.path.exists(get_startup_shortcut_path())
+
+    def set_start_with_windows(enable: bool):
+        shortcut_path = get_startup_shortcut_path()
+        
+        if enable:
+            # Determine the correct path (handles both raw script and PyInstaller .exe)
+            if getattr(sys, 'frozen', False):
+                target_path = sys.executable
+                working_dir = os.path.dirname(sys.executable)
+            else:
+                target_path = sys.executable  # python.exe
+                working_dir = os.path.dirname(os.path.abspath(__file__))
+                # If running as script, you might want to point to the script instead, 
+                # but usually this feature is intended for the built .exe
+                
+            # Use a quick PowerShell command to create a proper Windows .lnk shortcut
+            # This avoids needing external libraries like winshell
+            script_args = f"""
+            $WshShell = New-Object -ComObject WScript.Shell
+            $Shortcut = $WshShell.CreateShortcut('{shortcut_path}')
+            $Shortcut.TargetPath = '{target_path}'
+            $Shortcut.WorkingDirectory = '{working_dir}'
+            $Shortcut.Save()
+            """
+            import subprocess
+            subprocess.run(["powershell", "-Command", script_args], capture_output=True, creationflags=subprocess.CREATE_NO_WINDOW)
+        else:
+            if os.path.exists(shortcut_path):
+                try:
+                    os.remove(shortcut_path)
+                except OSError:
+                    pass
+    
+    def toggle_startup(icon, item):
+        current_state = is_start_with_windows()
+        set_start_with_windows(not current_state)
+
+
     #################
     ### ENDPOINTS ###
     #################
@@ -420,6 +515,12 @@ try:
             pystray.MenuItem("★ Customize Overlay", lambda: webbrowser.open(f"https://widgets.nutty.gg/now-playing/settings/")),
             pystray.MenuItem("★ Try my stream widgets!", lambda: webbrowser.open(f"https://nutty.gg/collections/member-exclusive-widgets")),
             pystray.Menu.SEPARATOR,
+            pystray.MenuItem(
+                "Start with Windows", 
+                toggle_startup, 
+                checked=lambda item: is_start_with_windows()
+            ),
+            pystray.Menu.SEPARATOR,
             pystray.MenuItem("Quit", on_quit)
         )