34 lines
1.2 KiB
Go
34 lines
1.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"video-factory/shortdrama/model/dto"
|
|
"video-factory/shortdrama/service"
|
|
)
|
|
|
|
// ==================== Generation ====================
|
|
|
|
func (c *drama) GenerateEpisode(ctx context.Context, req *dto.GenerateEpisodeReq) (res *struct{}, err error) {
|
|
return nil, service.DramaService.GenerateEpisode(ctx, req.DramaId, req.EpId)
|
|
}
|
|
|
|
func (c *drama) ContinueSegment(ctx context.Context, req *dto.ContinueSegmentReq) (res *struct{}, err error) {
|
|
return nil, service.DramaService.ContinueSegment(ctx, req.TaskId)
|
|
}
|
|
|
|
func (c *drama) FeedbackSegment(ctx context.Context, req *dto.FeedbackSegmentReq) (res *struct{}, err error) {
|
|
return nil, service.DramaService.FeedbackSegment(ctx, req.TaskId, req.Feedback)
|
|
}
|
|
|
|
func (c *drama) GetEpisodeTask(ctx context.Context, req *dto.GetEpisodeTaskReq) (res *dto.GetEpisodeTaskRes, err error) {
|
|
task, err := service.DramaService.GetEpisodeTask(ctx, req.EpId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &dto.GetEpisodeTaskRes{Task: task}, nil
|
|
}
|
|
|
|
func (c *drama) GetEpisodePollStatus(ctx context.Context, req *dto.EpisodePollReq) (res *dto.EpisodePollRes, err error) {
|
|
return service.DramaService.GetEpisodePollStatus(ctx, req.EpId)
|
|
}
|