File size: 2,253 Bytes
b6ac700 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import abc
class AbstractPromptStrategy:
"""
Represents a strategy for generating prompts for a given audio segment.
Note that the strategy must be picklable, as it will be serialized and sent to the workers.
"""
@abc.abstractmethod
def get_segment_prompt(self, segment_index: int, whisper_prompt: str, detected_language: str) -> str:
"""
Retrieves the prompt for a given segment.
Parameters
----------
segment_index: int
The index of the segment.
whisper_prompt: str
The prompt for the segment generated by Whisper. This is typically concatenated with the initial prompt.
detected_language: str
The language detected for the segment.
"""
pass
@abc.abstractmethod
def on_segment_finished(self, segment_index: int, whisper_prompt: str, detected_language: str, result: dict):
"""
Called when a segment has finished processing.
Parameters
----------
segment_index: int
The index of the segment.
whisper_prompt: str
The prompt for the segment generated by Whisper. This is typically concatenated with the initial prompt.
detected_language: str
The language detected for the segment.
result: dict
The result of the segment. It has the following format:
{
"text": str,
"segments": [
{
"text": str,
"start": float,
"end": float,
"words": [words],
}
],
"language": str,
}
"""
pass
def _concat_prompt(self, prompt1, prompt2):
"""
Concatenates two prompts.
Parameters
----------
prompt1: str
The first prompt.
prompt2: str
The second prompt.
"""
if (prompt1 is None):
return prompt2
elif (prompt2 is None):
return prompt1
else:
return prompt1 + " " + prompt2 |