Upload 6 files

requirements.txt

@@ -2,4 +2,5 @@ fastapi
 uvicorn
 aiohttp
 websockets==10.4
-pydantic
+pydantic
+aiofiles

tts-server.py

@@ -6,6 +6,7 @@ import os
 import logging
 import time
 import glob
+import hashlib
 from typing import Optional, List, Dict, Any, AsyncGenerator
 
 import aiofiles
@@ -142,23 +143,22 @@ class DoubaoTTS:
         self.voice_url = "https://www.doubao.com/alice/user_voice/recommend"
         self.voices = []
         self.req_count = 0
-        self.current_id = self._generate_id()
+        # self.current_id = self._generate_id() # Removed global ID
         self.cookie_manager = CookieManager(COOKIE_DIR)
 
-    def _generate_id(self) -> str:
-        """Generate a random device/session ID."""
-        num1 = math.floor(1e8 + 9e8 * random.random())
-        num2 = math.floor(1e8 + 9e8 * random.random())
-        return f"{num1}{num2}"
-
-    def _get_common_params(self) -> str:
+    def _get_device_id(self, cookie: str) -> str:
+        """Generate a deterministic device ID based on the cookie."""
+        # Use MD5 of cookie to get a consistent hash
+        hash_object = hashlib.md5(cookie.encode())
+        hex_dig = hash_object.hexdigest()
+        # Convert first 15 chars of hex to int to get a large number
+        num_val = int(hex_dig[:15], 16)
+        # Ensure it's 19 digits long (similar to original random ID)
+        return str(num_val).zfill(19)[:19]
+
+    def _get_common_params(self, cookie: str) -> str:
         """Generate the common query parameters for requests."""
-        self.req_count += 1
-        if self.req_count > 5:
-            self.req_count = 0
-            self.current_id = self._generate_id()
-        
-        cid = self.current_id
+        cid = self._get_device_id(cookie)
         return (
             f"&mode=0&language=zh&browser_language=zh-CN&device_platform=web"
             f"&aid=586861&real_aid=586861&pkg_type=release_version"
@@ -192,7 +192,7 @@ class DoubaoTTS:
 
         async with aiohttp.ClientSession() as session:
             for rec_type, tab_key in tabs:
-                params = self._get_common_params().strip('&')
+                params = self._get_common_params(cookie).strip('&')
                 url = f"{self.voice_url}?{params}"
                 
                 payload = {
@@ -267,7 +267,7 @@ class DoubaoTTS:
                 yield b"" # End stream
                 return
 
-            params = self._get_common_params()
+            params = self._get_common_params(cookie)
             ws_url = f"{self.ws_url}?format=aac&speaker={voice}&speech_rate={doubao_rate}&pitch={doubao_pitch}{params}"
             
             headers = {
@@ -289,6 +289,15 @@ class DoubaoTTS:
                     
                     first_chunk_received = False
                     
+                    # Clean up old files (Keep only the latest)
+                    try:
+                        for f in os.listdir(AUDIO_DIR):
+                            f_path = os.path.join(AUDIO_DIR, f)
+                            if os.path.isfile(f_path):
+                                os.remove(f_path)
+                    except Exception as e:
+                        logger.warning(f"Failed to clean up old audio files: {e}")
+
                     # Prepare file for saving this attempt
                     timestamp = int(time.time() * 1000)
                     filename = f"{timestamp}_{voice}.aac"