cache songs

app.py

@@ -35,6 +35,7 @@ class BotConfig:
     HF_REPO: str = "not-lain/assets"
     HF_SPACE: str = "https://not-lain-ytdlp.hf.space/"
     MAX_RETRY_ATTEMPTS: int = 3
+    MAX_CACHE_SIZE: int = 100  # Maximum number of songs to cache
 
     @property
     def song_path(self) -> str:
@@ -64,6 +65,15 @@ class MusicBot:
         self.current_song: Optional[QueueItem] = None
         self.queue: list[QueueItem] = []
         self.hf_client = Client(config.HF_SPACE, hf_token=None)
+        self.song_cache: dict[str, tuple[str, str]] = {}  # url -> (title, file_path)
+
+    def _manage_cache(self) -> None:
+        """Remove oldest songs if cache exceeds maximum size"""
+        if len(self.song_cache) > self.config.MAX_CACHE_SIZE:
+            # Convert to list to get oldest entries (assuming insertion order is maintained)
+            oldest_urls = list(self.song_cache.keys())[: -self.config.MAX_CACHE_SIZE]
+            for url in oldest_urls:
+                del self.song_cache[url]
 
     async def ensure_voice_state(self, ctx: commands.Context) -> None:
         """Validate voice state and raise appropriate errors"""
@@ -74,15 +84,29 @@ class MusicBot:
             raise VoiceStateError("You must be in the same voice channel as the bot!")
 
     async def download_song(self, queue_item: QueueItem) -> None:
-        """Download song from URL and update queue item with file path"""
+        """Download song from URL and update queue item with file path, using cache when available"""
         try:
+            # Check if song is in cache
+            if queue_item.url in self.song_cache:
+                title, file_path = self.song_cache[queue_item.url]
+                queue_item.title = title
+                queue_item.file_path = file_path
+                logger.info(f"Found song in cache: {title}")
+                return
+
+            # Download if not in cache
             job = self.hf_client.submit(url=queue_item.url, api_name="/predict")
             while not job.done():
                 time.sleep(0.1)
-            print("job.outputs() = ", job.outputs())
             title, file_path = job.outputs()[0]
+
+            # Update cache
+            self.song_cache[queue_item.url] = (title, file_path)
+            self._manage_cache()
+
             queue_item.title = title
             queue_item.file_path = file_path
+            logger.info(f"Downloaded and cached new song: {title}")
         except Exception as e:
             logger.error(f"Error downloading song: {e}")
             raise
@@ -166,6 +190,7 @@ class MusicBot:
         self.loop_mode = LoopMode.NONE
         self.current_song = None
         self.queue.clear()
+        # Note: We don't clear the cache when resetting state
 
 
 async def handle_voice_command(