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

1, add voice volume

Browse Source 
2, code optimization
This commit is contained in:
harry
2024-04-05 11:06:03 +08:00
parent 416a446b5e
commit ebe47d2507
6 changed files with 41 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/models/schema.py
View File

@@ -98,6 +98,7 @@ class VideoParams:
video_language: Optional[str] = "" # auto detect
voice_name: Optional[str] = ""
voice_volume: Optional[float] = 1.0
bgm_type: Optional[str] = "random"
bgm_file: Optional[str] = ""
bgm_volume: Optional[float] = 0.2
@@ -231,18 +232,18 @@ class VideoTermsResponse(BaseResponse):
}
class BgmListResponse(BaseResponse):
class BgmRetrieveResponse(BaseResponse):
class Config:
json_schema_extra = {
"example": {
"status": 200,
"message": "success",
"data": {
"bgm_list": [
"files": [
{
"filename": "output000.mp3",
"size": 2249517,
"filepath": "C:\\Users\\cathy\\Desktop\\MoneyPrinterTurbo\\resource\\songs\\output000.mp3"
"name": "output013.mp3",
"size": 1891269,
"file": "/MoneyPrinterTurbo/resource/songs/output013.mp3"
}
]
}
@@ -257,7 +258,7 @@ class BgmUploadResponse(BaseResponse):
"status": 200,
"message": "success",
"data": {
"uploaded_path": "/root/home/MoneyPrinterTurbo/resource/songs/example.mp3"
"file": "/MoneyPrinterTurbo/resource/songs/example.mp3"
}
},
}

7
app/services/video.py
View File

@@ -229,6 +229,11 @@ def generate_video(video_path: str,
result = CompositeVideoClip(clips)
audio = AudioFileClip(audio_path)
try:
audio = audio.volumex(params.voice_volume)
except Exception as e:
logger.warning(f"failed to set audio volume: {e}")
result = result.set_audio(audio)
temp_output_file = f"{output_file}.temp.mp4"
@@ -302,6 +307,8 @@ if __name__ == "__main__":
cfg.n_threads = 2
cfg.paragraph_number = 1
cfg.voice_volume = 3.0
generate_video(video_path=video_file,
audio_path=audio_file,
subtitle_path=subtitle_file,