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

Code optimization

Browse Source
This commit is contained in:
harry
2024-04-05 09:55:31 +08:00
parent ff6333092b
commit 416a446b5e
3 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/controllers/v1/llm.py
View File

@@ -3,14 +3,13 @@ from app.controllers.v1.base import new_router
from app.models.schema import VideoScriptResponse, VideoScriptRequest, VideoTermsResponse, VideoTermsRequest
from app.services import llm
from app.utils import utils
from app.controllers import base
# 认证依赖项
# router = new_router(dependencies=[Depends(base.verify_token)])
router = new_router()
@router.post("/generate_video_script", response_model=VideoScriptResponse, summary="Generate a video script")
@router.post("/scripts", response_model=VideoScriptResponse, summary="Create a script for the video")
def generate_video_script(request: Request, body: VideoScriptRequest):
video_script = llm.generate_script(video_subject=body.video_subject,
language=body.video_language,
@@ -21,7 +20,7 @@ def generate_video_script(request: Request, body: VideoScriptRequest):
return utils.get_response(200, response)
@router.post("/generate_video_terms", response_model=VideoTermsResponse, summary="Generate video terms by video script")
@router.post("/terms", response_model=VideoTermsResponse, summary="Generate video terms based on the video script")
def generate_video_terms(request: Request, body: VideoTermsRequest):
video_terms = llm.generate_terms(video_subject=body.video_subject,
video_script=body.video_script,