hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed to sync

Browse Source
This commit is contained in:
harry
2024-03-18 21:39:47 +08:00
parent 59ae82a47d
commit 6bfb04f755
5 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/services/llm.py
View File

@@ -37,7 +37,6 @@ def _generate_response(prompt: str) -> str:
api_key = config.app.get("oneapi_api_key")
model_name = config.app.get("oneapi_model_name")
base_url = config.app.get("oneapi_base_url", "")
model_name = config.app.get("g4f_model_name")
else:
raise ValueError("llm_provider is not set, please set it in the config.toml file.")

4
app/services/task.py
View File

@@ -21,7 +21,7 @@ def _parse_voice(name: str):
return _voice, _lang
async def start(task_id, params: VideoParams):
def start(task_id, params: VideoParams):
"""
{
"video_subject": "",
@@ -60,7 +60,7 @@ async def start(task_id, params: VideoParams):
subtitle_path = path.join(utils.task_dir(task_id), f"subtitle.srt")
logger.info("\n\n## generating audio")
sub_maker =await voice.tts(text=script, voice_name=voice_name, voice_file=audio_file)
sub_maker = voice.tts(text=script, voice_name=voice_name, voice_file=audio_file)
subtitle_provider = config.app.get("subtitle_provider", "").strip().lower()
logger.info(f"\n\n## generating subtitle, provider: {subtitle_provider}")

6
app/services/voice.py
View File

@@ -8,7 +8,7 @@ import edge_tts
from app.utils import utils
async def tts(text: str, voice_name: str, voice_file: str) -> SubMaker:
def tts(text: str, voice_name: str, voice_file: str) -> SubMaker:
logger.info(f"start, voice name: {voice_name}")
async def _do() -> SubMaker:
@@ -22,7 +22,7 @@ async def tts(text: str, voice_name: str, voice_file: str) -> SubMaker:
sub_maker.create_sub((chunk["offset"], chunk["duration"]), chunk["text"])
return sub_maker
sub_maker = await _do()
sub_maker = asyncio.run(_do())
logger.info(f"completed, output file: {voice_file}")
return sub_maker
@@ -100,7 +100,7 @@ if __name__ == "__main__":
for voice_name in voice_names:
voice_file = f"{temp_dir}/tts-{voice_name}.mp3"
subtitle_file = f"{temp_dir}/tts.mp3.srt"
sub_maker = await tts(text=text, voice_name=voice_name, voice_file=voice_file)
sub_maker = tts(text=text, voice_name=voice_name, voice_file=voice_file)
create_subtitle(sub_maker=sub_maker, text=text, subtitle_file=subtitle_file)