Update scraper/voicevox.py

scraper/voicevox.py

@@ -1,21 +1,13 @@
 import requests
 import time
+from functools import lru_cache
+
 
 class StatusError(Exception):
     pass
 
-# Custom cache dictionary
-cache = {}
-
-def synthesis(text, speaker=None):
-    if speaker is None:
-        speaker = 1
-    
-    # Check if the result is already in the cache
-    cache_key = (text, speaker)
-    if cache_key in cache:
-        return cache[cache_key]
-    
+@lru_cache(maxsize=None)
+def synthesis(text, speaker):
     url = f"https://api.tts.quest/v3/voicevox/synthesis?text={text}&speaker={speaker}"
     resp = requests.get(url)
     resp.raise_for_status()
@@ -25,12 +17,9 @@ def synthesis(text, speaker=None):
         raise StatusError("Unexpected status")
 
     resp = requests.get(data["mp3DownloadUrl"])
-    audio_content = resp.content
+    return resp.content
     
-    # Cache the result
-    cache[cache_key] = audio_content
-    
-    return audio_content
+
 
 def check_audio_status(url):
     is_ready = False
@@ -39,6 +28,7 @@ def check_audio_status(url):
         resp.raise_for_status()
         data = resp.json()
         is_ready = data["isAudioReady"]
-        time.sleep(0.5)
+        time.sleep(1)
 
     return is_ready
+