faster-whisper-webui / src /prompts /abstractPromptStrategy.py
aadnk's picture
Adding JSON initial prompt
74b7d77
raw
history blame
2.25 kB
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