Upload 5 files

cookie/19245776972.txt

@@ -0,0 +1 @@
+_ga=GA1.1.437012191.1745432605; d_ticket=7ef9a4bada1454fd46e0dc3fb5d3bbf7af1c9; flow_user_country=CN; _gcl_aw=GCL.1758716523.Cj0KCQjwrc7GBhCfARIsAHGcW5X3vlZenZizh4uxDAOeWTzQDdrT7JOqk-0_Toh0-nyTI98ilypZKRgaAjAREALw_wcB; _gcl_gs=2.1.k1$i1758716513$u91292815; gd_random=eyJtYXRjaCI6dHJ1ZSwicGVyY2VudCI6MC4yNTI2NzkwMDcwMDM2NzQyfQ==.RGhswbjyDU6o266pLNQGIlPw2Y8mDLztp4TGXAWMlcQ=; flow_ssr_sidebar_expand=1; s_v_web_id=verify_mind1jr9_VbD7t7qj_uikr_43bO_8n0b_zX2SCB6AdbOM; passport_csrf_token=32bfed5a75c5103dc3733669211dde21; passport_csrf_token_default=32bfed5a75c5103dc3733669211dde21; odin_tt=7f7bdbabac871052b33748b916fe0d9eeaee8805d718e032dc76eb3331e69248d07b38c4a514cd3ec554bfd56a88948f8d4623af808c742c18c899d4244d4d78; n_mh=8tLRwoSVKOVBgyNwo1ffSlHLPFA-RVJfi6Mvd4NuGM0; sid_guard=3a913278b150423698eda5269004a756%7C1764606680%7C2592000%7CWed%2C+31-Dec-2025+16%3A31%3A20+GMT; uid_tt=d6dad1722ee632bb79eabc2375f0ca93; uid_tt_ss=d6dad1722ee632bb79eabc2375f0ca93; sid_tt=3a913278b150423698eda5269004a756; sessionid=3a913278b150423698eda5269004a756; sessionid_ss=3a913278b150423698eda5269004a756; session_tlb_tag=sttt%7C16%7COpEyeLFQQjaY7aUmkASnVv________-1-1P7zRCjTPIPFv_pUUIHMZVYQsjlpXZD0w_8D__sD7Y%3D; session_tlb_tag_bk=sttt%7C16%7COpEyeLFQQjaY7aUmkASnVv________-1-1P7zRCjTPIPFv_pUUIHMZVYQsjlpXZD0w_8D__sD7Y%3D; is_staff_user=false; sid_ucp_v1=1.0.0-KGMzY2FkMTZkNzkyYjAxZDY3OTFlM2Q0MzYzY2UzNzE0OWRkMDhiZDUKIAiLzpCpts3aAxDYhbfJBhjCsR4gDDCqpdmtBjgHQPQHGgJobCIgM2E5MTMyNzhiMTUwNDIzNjk4ZWRhNTI2OTAwNGE3NTY; ssid_ucp_v1=1.0.0-KGMzY2FkMTZkNzkyYjAxZDY3OTFlM2Q0MzYzY2UzNzE0OWRkMDhiZDUKIAiLzpCpts3aAxDYhbfJBhjCsR4gDDCqpdmtBjgHQPQHGgJobCIgM2E5MTMyNzhiMTUwNDIzNjk4ZWRhNTI2OTAwNGE3NTY; user_language_code=zh; i18next=zh; ttwid=1%7CUWvZl8BhJG24HsPYG-AMSH425URWFERYswXeSG5o7JA%7C1764606681%7Cc06758733f55438250cd0bcd4655efd74e6c85bbaec7b29a4e0993cd7dac2c8f; passport_fe_beating_status=true; msToken=uMPTw6l1KqvcVnqkxi14Lxk9_77FWhrXrKQZEraq2s9cX4hnMNVW-JkviHvPlEgeh2qEDosTsKmlapgsIKH6SEC6PVrpWl1uzRR1hBBGvcSTL_bnIyoAE2fiD0SmV_QBIrHtxQhAeE8-; _ga_G8EP5CG8VZ=GS2.1.s1764606227$o73$g1$t1764606691$j51$l0$h0; tt_scid=t6ZrzYZXXDmFGF8bsZTLqjggfDFJ0m4ulDTSSZExH52G.8iucd7OjH.2oqhCdHPx0472

tts-server.py

@@ -8,6 +8,7 @@ import time
 import glob
 from typing import Optional, List, Dict, Any, AsyncGenerator
 
+import aiofiles
 import aiohttp
 import websockets
 from fastapi import FastAPI, HTTPException, Header, Request, BackgroundTasks
@@ -32,6 +33,7 @@ PORT = 1547
 HOST = "0.0.0.0"
 MODELS_FILE = "models.json"
 COOKIE_DIR = "cookie"  # Directory to store cookie txt files
+AUDIO_DIR = "saved_audio"
 
 # Initialize FastAPI
 @asynccontextmanager
@@ -110,6 +112,11 @@ class CookieManager:
         if self.failure_count >= 2:
             self._rotate()
 
+    def force_rotate(self):
+        """Force switch to the next cookie (e.g. when blocked)."""
+        logger.warning(f"Cookie index {self.current_index} blocked/rate-limited. Forcing rotation.")
+        self._rotate()
+
     def report_success(self):
         """Reset failure count on success."""
         if self.failure_count > 0:
@@ -239,6 +246,13 @@ class DoubaoTTS:
     async def stream_audio(self, text: str, voice: str, speed: float = 1.0, pitch: float = 1.0) -> AsyncGenerator[bytes, None]:
         """Connect to WebSocket and yield audio chunks with retry logic."""
         
+        # Ensure audio directory exists
+        if not os.path.exists(AUDIO_DIR):
+            try:
+                os.makedirs(AUDIO_DIR, exist_ok=True)
+            except Exception as e:
+                logger.error(f"Failed to create audio directory: {e}")
+
         # Map OpenAI speed (0.25 - 4.0) to Doubao rate (-100 to 100)
         doubao_rate = int((speed - 1) * 100)
         doubao_rate = max(-100, min(100, doubao_rate))
@@ -275,19 +289,56 @@ class DoubaoTTS:
                     
                     first_chunk_received = False
                     
-                    async for message in ws:
-                        if isinstance(message, bytes):
-                            if not first_chunk_received:
-                                first_chunk_received = True
-                                self.cookie_manager.report_success() # Mark cookie as good
-                            yield message
-                        elif isinstance(message, str):
-                            # Check for error messages in text frames
-                            # Sometimes errors come as text JSON
-                            pass
+                    # Prepare file for saving this attempt
+                    timestamp = int(time.time() * 1000)
+                    filename = f"{timestamp}_{voice}.aac"
+                    filepath = os.path.join(AUDIO_DIR, filename)
                     
-                    # If we finished the loop naturally, break retry loop
-                    return
+                    file_written = False
+                    try:
+                        async with aiofiles.open(filepath, "wb") as f_out:
+                            async for message in ws:
+                                if isinstance(message, bytes):
+                                    if not first_chunk_received:
+                                        first_chunk_received = True
+                                        self.cookie_manager.report_success() # Mark cookie as good
+                                        logger.info(f"Streaming and saving audio to: {filepath}")
+                                    
+                                    yield message
+                                    await f_out.write(message)
+                                    file_written = True
+                                    
+                                elif isinstance(message, str):
+                                    # Check for error messages in text frames
+                                    try:
+                                        msg_json = json.loads(message)
+                                        code = msg_json.get("code")
+                                        if code and code != 0:
+                                            error_msg = msg_json.get("message", "Unknown Error")
+                                            logger.error(f"Doubao API Error (Code {code}): {error_msg}")
+                                            
+                                            # If blocked or auth failed, report failure to rotate cookie
+                                            if "block" in str(error_msg).lower() or code == 710022002:
+                                                self.cookie_manager.force_rotate() # Force rotate immediately on block
+                                                raise Exception(f"Blocked by server (Risk Control): {error_msg}")
+                                    except json.JSONDecodeError:
+                                        pass
+                        
+                        # If we finished the loop naturally, break retry loop
+                        if file_written:
+                            return
+                        else:
+                            # If no bytes were written, something went wrong but no exception was raised
+                            logger.warning("Connection closed without receiving audio data.")
+                            
+                    finally:
+                        # Cleanup empty files if failed
+                        if not file_written and os.path.exists(filepath):
+                            try:
+                                os.remove(filepath)
+                                logger.info(f"Removed empty audio file: {filepath}")
+                            except Exception as e:
+                                logger.error(f"Failed to remove empty file: {e}")
 
             except websockets.exceptions.InvalidStatusCode as e:
                 logger.warning(f"WebSocket Handshake failed (Attempt {attempt+1}): {e.status_code}")