added summarization

Author: MrWong99

discord-bot/pkg/bot/campaign.go

@@ -185,6 +185,14 @@ func campaignHandler(s *discordgo.Session, i *discordgo.InteractionCreate) {
 				if err := campaign.Close(); err != nil {
 					slog.Warn("unexpected error while stopping campaign", "campaign", campaign.Name, "error", err)
 				}
+				summary, err := campaign.Summary()
+				if err != nil {
+					slog.Warn("could not summarize campaign transcript", "error", err)
+				} else {
+					s.InteractionResponseEdit(respI, &discordgo.WebhookEdit{
+						Content: &summary,
+					})
+				}
 				// Cleanup
 				close(componentButtons[i.GuildID]["stop_campaign"])
 				delete(componentButtons[i.GuildID], "stop_campaign")
@@ -193,7 +201,7 @@ func campaignHandler(s *discordgo.Session, i *discordgo.InteractionCreate) {
 			s.InteractionRespond(respI, &discordgo.InteractionResponse{
 				Type: discordgo.InteractionResponseUpdateMessage,
 				Data: &discordgo.InteractionResponseData{
-					Content: "Transcript finished",
+					Content: "Transcript finished.",
 					Files: []*discordgo.File{
 						{
 							Name:        "transcript.txt",

discord-bot/pkg/pnp/campaign.go

@@ -2,19 +2,25 @@ package pnp
 
 import (
 	"bytes"
+	"context"
 	"fmt"
 	"io"
 	"log/slog"
 	"math/rand"
 	"sync"
 	"text/template"
 
+	"github.com/MrWong99/TaileVoices/discord_bot/pkg/oai"
 	"github.com/MrWong99/TaileVoices/discord_bot/pkg/vecdb"
+	"github.com/sashabaranov/go-openai"
 	"gopkg.in/yaml.v3"
 
 	_ "embed"
 )
 
+//go:embed summarization_prompt.txt
+var summarizationPrompt string
+
 //go:embed stt_init_prompt.tpl
 var sttPromptText string
 
@@ -186,11 +192,9 @@ func (c *Campaign) HandleText(name string, segment string) {
 	}
 
 	go func() {
-		c.transcriptMu.Lock()
 		promptContext := PromptContext{
-			CurrentTranscript: c.CurrentSessionTranscript,
+			CurrentTranscript: c.CurrentTranscript(),
 		}
-		c.transcriptMu.Unlock() // don't defer as it will be a deadlock once c.HandleText() is called
 		oldTranscripts, err := c.dbClient.SearchTranscripts(c.Name, segment)
 		if err != nil {
 			slog.Warn("could not search old transcripts for reference", "error", err, "collection", c.Name, "concept", segment)
@@ -210,6 +214,37 @@ func (c *Campaign) HandleText(name string, segment string) {
 	}()
 }
 
+// Summary of the current campaign transcript. Excludes all non pen & paper related content.
+// Uses a GenAI to do the summary.
+//
+// Can be called after Close() has been called.
+func (c *Campaign) Summary() (string, error) {
+	resp, err := oai.Client.CreateChatCompletion(context.Background(), openai.ChatCompletionRequest{
+		Model: openai.GPT4o,
+		Messages: []openai.ChatCompletionMessage{
+			{
+				Role:    openai.ChatMessageRoleSystem,
+				Content: summarizationPrompt,
+			},
+			{
+				Role:    openai.ChatMessageRoleUser,
+				Content: c.CurrentTranscript(),
+			},
+		},
+	})
+	if err != nil {
+		return "", fmt.Errorf("failed to create chat completion: %w", err)
+	}
+	return resp.Choices[0].Message.Content, nil
+}
+
+// CurrentTranscript of this session.
+func (c *Campaign) CurrentTranscript() string {
+	c.transcriptMu.Lock()
+	defer c.transcriptMu.Unlock()
+	return c.CurrentSessionTranscript
+}
+
 // Close this campaigns session by closing C() and storing its transcript in the vector database.
 func (c *Campaign) Close() error {
 	close(c.actorResponses)

discord-bot/pkg/pnp/npc_system_prompt.tpl

@@ -29,6 +29,7 @@ Other name: text line
 
 Your answers should be responses in natural language that fit into the end of the current transcript.
 Omit your name at the beginning of the line so instead of "Name: My response" just respond "My response".
+Also never include lines of other speakers, just speak your next line and nothing more!
 Always answer in the same language as the current transcript!
 Keep your answers short unless the following script tells you otherwise.
 

discord-bot/pkg/pnp/summarization_prompt.txt

@@ -0,0 +1,20 @@
+You have the task of summarising a transcription of a role-play session. Your aim is to create a step-by-step summary of this transcription. List individual scenes in a bulleted list, e.g. like this:
+
+1. at the beginning, the money is in...
+2. a dragon appears and...
+3. the heroes travel back home...
+
+Concentrate only on what happened in the role-play and do not include any meta-information that has nothing to do with the role-play. Try to record the events objectively, but at the same time tell an exciting story. Always answer in the language that the transcription was provided in!
+
+Transcripts always have the following format:
+
+
+Character name 1: Role play text or meta question
+GameMaster: Narrative line or spoken line of the NPC
+Character name 2: Role-playing text or meta-question
+...
+
+
+The transcription therefore alternates between different characters, and each of them can either speak as the character they are currently playing or ask meta-questions about the current campaign.
+The gamemaster's job is to provide a narrative and context to the roleplay. The gamemaster can also impersonate an NPC.
+You will need to check each person's transcriptions and the context in which they are made to determine if they are impersonating a specific character or not roleplaying at all. Remember that the gamemaster sets most of the context for the roleplay