add code (#137)

backend/fastrtc/__init__.py

@@ -6,7 +6,7 @@ from .credentials import (
 from .reply_on_pause import AlgoOptions, ReplyOnPause, SileroVadOptions
 from .reply_on_stopwords import ReplyOnStopWords
 from .speech_to_text import MoonshineSTT, get_stt_model
-from .stream import Stream
+from .stream import Stream, UIArgs
 from .text_to_speech import KokoroTTSOptions, get_tts_model
 from .tracks import (
     AsyncAudioVideoStreamHandler,
@@ -62,4 +62,5 @@ __all__ = [
     "get_tts_model",
     "KokoroTTSOptions",
     "wait_for_item",
+    "UIArgs",
 ]

backend/fastrtc/stream.py

@@ -36,6 +36,8 @@ class Body(BaseModel):
 class UIArgs(TypedDict):
     title: NotRequired[str]
     """Title of the demo"""
+    subtitle: NotRequired[str]
+    """Subtitle of the demo. Text will be centered and displayed below the title."""
     icon: NotRequired[str]
     """Icon to display on the button instead of the wave animation. The icon should be a path/url to a .svg/.png/.jpeg file."""
     icon_button_color: NotRequired[str]
@@ -190,6 +192,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Video Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Row():
@@ -229,6 +239,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Video Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Row():
@@ -269,6 +287,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Video Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Column(elem_classes=["my-column"]):
@@ -308,6 +334,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Audio Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Row():
@@ -352,6 +386,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Audio Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Row():
@@ -395,6 +437,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Audio Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                    {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Row():
@@ -439,6 +489,14 @@ class Stream(WebRTCConnectionMixin):
                 <h1 style='text-align: center'>
                 {ui_args.get("title", "Audio Video Streaming (Powered by FastRTC ⚡️)")}
                 </h1>
+                """
+                )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
                 """
                     )
                 with gr.Row():