Upload tts-server.py

tts-server.py

@@ -327,9 +327,11 @@ class DoubaoTTS:
                                             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}")
+                                            # 710022002: Blocked
+                                            # 671000003: Rate Limited
+                                            if "block" in str(error_msg).lower() or code in [710022002, 671000003]:
+                                                self.cookie_manager.force_rotate() # Force rotate immediately on block/limit
+                                                raise Exception(f"Blocked/Limited by server ({code}): {error_msg}")
                                     except json.JSONDecodeError:
                                         pass
                         
@@ -357,10 +359,14 @@ class DoubaoTTS:
             except Exception as e:
                 logger.error(f"WebSocket error (Attempt {attempt+1}): {e}")
                 self.cookie_manager.report_failure()
+                # Wait a bit before retrying to avoid hammering
+                await asyncio.sleep(3)
                 # Continue to next attempt
 
         logger.error("All retry attempts failed.")
-        raise HTTPException(status_code=500, detail="Failed to generate audio after retries.")
+        # Do not raise HTTPException here as it causes RuntimeError if response started
+        # Just return to close the stream
+        return
 
 # Initialize Engine
 engine = DoubaoTTS()