|
from fastapi import APIRouter, HTTPException, Query
|
|
from loguru import logger
|
|
from app.services.video import generate_video, create_video_with_scenes, generate_voice
|
|
from app.schemas.video import VideoGenerateRequest, VideoGenerateResponse, StoryScene
|
|
import os
|
|
import json
|
|
from app.utils.utils import extract_id
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/generate")
|
|
async def generate_video_endpoint(
|
|
request: VideoGenerateRequest
|
|
):
|
|
"""生成视频"""
|
|
try:
|
|
video_file = await generate_video(request)
|
|
task_id = extract_id(video_file)
|
|
|
|
video_url = "http://127.0.0.1:8000/tasks/" + task_id + "/video.mp4"
|
|
return VideoGenerateResponse(
|
|
success=True,
|
|
data={"video_url": video_url}
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"Failed to generate video: {str(e)}")
|
|
return VideoGenerateResponse(
|
|
success=False,
|
|
message=str(e)
|
|
)
|
|
|
|
|
|
|
