llama-index

1
"""Text to speech module."""
2
from abc import ABC, abstractmethod
3
from typing import Any
4

5

6
class BaseTTS(ABC):
7
    """Base class for text to speech modules."""
8

9
    def __init__(self) -> None:
10
        pass
11

12
    @abstractmethod
13
    def generate_audio(self, text: str) -> Any:
14
        """Generate audio from text.
15

16
        NOTE: return type is Any, but it should be any object that can be fed
17
        as `data` into IPython.display.Audio(). This includes numpy array, list,
18
        unicode, str or bytes
19

20
        """
21
        raise NotImplementedError(
22
            "generate_audio method should be implemented by subclasses"
23
        )
24