Create ACGNTTS.py (#200)

Type:ACGNTTS
2026-07-22 07:03:53 +08:00 · 2025-03-07 10:42:04 +08:00
parent 6464c81e72
commit e1ebcfa865
1 changed files with 74 additions and 0 deletions
@@ -0,0 +1,74 @@
+import os
+import uuid
+import json
+import requests
+import shutil
+from datetime import datetime
+from core.providers.tts.base import TTSProviderBase
+
+
+import http.client
+import urllib.parse
+
+class TTSProvider(TTSProviderBase):
+     def __init__(self, config, delete_audio_file):
+         super().__init__(config, delete_audio_file)   
+         self.url = config.get("url","https://u95167-bd74-2aef8085.westx.seetacloud.com:8443/flashsummary/tts?token=")  
+         self.voice_id = config.get("voice_id",1695)
+         self.token = config.get("token")
+         self.to_lang = config.get("to_lang")
+         self.volume_change_dB = config.get("volume_change_dB",0)
+         self.speed_factor = config.get("speed_factor",1)
+         self.stream = config.get("stream",False)
+         self.output_file = config.get("output_file")
+         self.pitch_factor = config.get("pitch_factor",0)
+         self.format = config.get("format","wav")
+         self.emotion = config.get("emotion",1)
+         self.header = {
+            "Content-Type": "application/json"
+         }
+     def generate_filename(self, extension=".mp3"):
+         return os.path.join(self.output_file, f"tts-{datetime.now().date()}@{uuid.uuid4().hex}{extension}")
+     async def text_to_speak(self, text, output_file):
+        url = f'{self.url}{self.token}'
+        result = "firefly"
+        print(url)
+        payload = json.dumps({
+            "to_lang": self.to_lang,
+            "text": text,
+            "emotion": self.emotion,
+            "format": self.format,
+            "volume_change_dB": self.volume_change_dB,
+            "voice_id": self.voice_id,
+            "pitch_factor": self.pitch_factor,
+            "speed_factor": self.speed_factor,
+            "token": self.token
+        })
+
+        
+        print("请求编成")
+        resp = requests.request("POST",url, data=payload)
+        print(resp.status_code)
+        print("请求开始")
+        print("完整请求参数：",payload) 
+        if resp.status_code != 200:
+           print(f"Error: {resp.status_code}")
+           return None
+        resp_json = resp.json() 
+        try:
+           result = resp_json['url'] + ':' + str(
+           resp_json['port']) + '/flashsummary/retrieveFileData?stream=True&token=' + self.token + '&voice_audio_path=' + resp_json['voice_path']
+           print("result已写入")
+        except Exception as e:
+           print("error:",e)
+       
+        audio_content = requests.get(result)
+        with open(output_file, "wb") as f:
+            f.write(audio_content.content)
+            print("音频写入成功")
+            return True
+        voice_path = resp_json.get("voice_path")
+        des_path = output_file
+        shutil.move(voice_path,des_path)
+        print(voice_path,"value")
+        print("API响应原始内容：",resp.text)