Update tool2.py

tool2.py

@@ -102,4 +102,54 @@ try:
     clients.append(client_3)
     print("Loaded voice transformer TTS client (client_3) ✔")
 except ValueError as e:
-    print(f"Error loading voice transformer TTS client (
+    print(f"Error loading voice transformer TTS client (client_3): {e}")
+    print("Voice Transformer Text-to-Speech (client_3) will be unavailable.")
+except Exception as e: # Catch other potential exceptions during client initialization
+    print(f"An unexpected error occurred during voice transformer TTS client (client_3) initialization: {e}")
+    print("Voice Transformer Text-to-Speech (client_3) will be unavailable.")
+
+if not clients:
+    print("No Text-to-speech clients loaded due to errors. Audio functionality will be disabled.")
+else:
+    print(f"Loaded {len(clients)} TTS clients.")
+
+# Text-to-speech function with rate limiting, retry mechanism, and client rotation
+def text_to_speech(text, retries=3, delay=5):
+    global clients # Ensure we are using the global clients list
+    if not clients: # If no clients are loaded, return None immediately
+        print("Warning: No Text-to-speech clients available.")
+        return None
+
+    client_index = 0  # Start with the first available client
+    for attempt in range(retries):
+        try:
+            client = clients[client_index % len(clients)] # Use modulo to cycle through available clients
+            client_index += 1 # Increment client_index for the next attempt in case of rate limit
+            print(f"Attempt {attempt + 1} using client: {client_index % len(clients) + 1}") # Client index for logging (1-based)
+            if client == client_fast: # Check if using client_fast
+                result = client.predict(
+                    language="English",
+                    repo_id="csukuangfj/vits-piper-en_US-hfc_female-medium|1 speaker",
+                    text=text,
+                    sid="0",
+                    speed=0.8,
+                    api_name="/process"
+                )
+            else: # Assuming client_2 or client_3 or any other client in the future
+                result = client.predict(
+                    text=text,
+                    api_name="/predict"
+                )
+            return result
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 429:
+                print(f"Rate limit exceeded using client {client_index % len(clients) + 1}. Retrying in {delay} seconds...")
+                time.sleep(delay)
+            else:
+                raise e
+        except Exception as e: # Catch any other potential errors during prediction
+            print(f"Error during text-to-speech prediction using client {client_index % len(clients) + 1}: {e}")
+            # Consider rotating client on any error, not just rate limits, to try other clients if available
+            client_index += 1 # Rotate to the next client for the next attempt
+    print("Max retries exceeded for all TTS clients. Could not process the request.")
+    return None