wuenlp_tools.models.scenes.summarisation

 1from wuenlp.impl.UIMANLPStructs import UIMADocument, UIMAAnnotation, UIMASystemScene, UIMASpan
 2
 3from wuenlp_tools.keys import OPENAI_API_KEY
 4from wuenlp_tools.pipeline import PipelineStep, PipelineCapability
 5from wuenlp_tools.utils.prompting import default_llm, gpt5nano
 6from wuenlp_tools.utils.summarize import Summarizer, Summary, SummarizeProcessor
 7
 8scene_summarizer = Summarizer(model=gpt5nano, num_sentences=3,
 9                              system_prompt="You are given a scene from a text. Please summarize it in about {} sentences and provide 1-3 keywords that will help a reader familiar with the text to recognise the scene.")
10
11SceneSummarizer = PipelineStep("Scene Summarizer", SummarizeProcessor(scene_summarizer), unit_type=UIMASystemScene,
12                               added_additional_features=["llama_summary", "llama_keywords"],
13                               requires_api_key=OPENAI_API_KEY, requires_paid_api_requests=True,
14                               provides=[PipelineCapability.SCENE_SUMMARIES], requires=[PipelineCapability.SCENES])