46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
package promptcontext
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"gitea.maximumdirect.net/eric/audita/internal/core/schema"
|
|
)
|
|
|
|
type transcriptSectionSegment struct {
|
|
ID int `json:"id"`
|
|
Speaker string `json:"speaker"`
|
|
Start float64 `json:"start"`
|
|
End float64 `json:"end"`
|
|
Text string `json:"text"`
|
|
Categories []string `json:"categories,omitempty"`
|
|
}
|
|
|
|
type transcriptSectionPayload struct {
|
|
SectionIndex int `json:"section_index"`
|
|
Segments []transcriptSectionSegment `json:"segments"`
|
|
}
|
|
|
|
// MarshalTranscriptSectionJSON builds the standardized transcript-section JSON
|
|
// payload consumed by proposal prompt templates.
|
|
func MarshalTranscriptSectionJSON(transcript *schema.Transcript, sectionIndex int) ([]byte, error) {
|
|
payload := transcriptSectionPayload{
|
|
SectionIndex: sectionIndex,
|
|
Segments: make([]transcriptSectionSegment, 0),
|
|
}
|
|
|
|
if transcript != nil {
|
|
for _, s := range transcript.Segments {
|
|
payload.Segments = append(payload.Segments, transcriptSectionSegment{
|
|
ID: s.ID,
|
|
Speaker: s.Speaker,
|
|
Start: s.Start,
|
|
End: s.End,
|
|
Text: s.Text,
|
|
Categories: append([]string(nil), s.Categories...),
|
|
})
|
|
}
|
|
}
|
|
|
|
return json.MarshalIndent(payload, "", " ")
|
|
}
|